pytorch/caffe2/python/operator_test
Tilak Sharma 1e8a16224f PackSegments: return value presence.
Summary:
Optionally return a blob of shape [batch size, max length] that is
false only in locations where the output tensor was padded.
One can separately convert lengths to segment ids and cast, but
this is more convenient, and possibly more efficient.

Differential Revision: D6006073

fbshipit-source-id: af6c4ea31972566e7d059dcd3fdd8afba97a88e9
2017-10-12 11:17:34 -07:00
..
activation_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
adagrad_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
adam_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
apmeter_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
atomic_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
batch_box_cox_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
blobs_queue_db_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
boolean_mask_test.py adding batch support to SequenceMaskOps 2017-10-04 15:18:09 -07:00
boolean_unmask_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
channel_shuffle_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
checkpoint_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
clip_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
concat_split_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
conditional_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
conv_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
conv_transpose_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
copy_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
cosine_embedding_criterion_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
counter_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
crf_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
cross_entropy_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
cudnn_recurrent_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
dataset_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
deform_conv_test.py Deformable convolution operator in Caffe2 2017-09-26 16:20:31 -07:00
distance_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
dropout_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
duplicate_operands_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
elementwise_linear_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
elementwise_logical_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
elementwise_op_broadcast_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
elementwise_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
emptysample_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
extend_tensor_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
fc_operator_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
filler_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
find_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
gather_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
gather_ranges_op_test.py add GatherRangesToDense operator 2017-10-04 15:18:10 -07:00
given_tensor_fill_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
group_conv_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
gru_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
hsm_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
im2col_col2im_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
image_input_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
index_hash_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
index_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
instance_norm_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
layer_norm_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
leaky_relu_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
lengths_tile_op_test.py Add GPU support for LengthsTile 2017-10-12 11:17:34 -07:00
lengths_top_k_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
loss_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
lpnorm_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
map_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
margin_ranking_criterion_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
math_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
matmul_op_test.py Enable axis argument for MatmulOp 2017-10-10 16:47:37 -07:00
merge_id_lists_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
mkl_conv_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
mkl_packed_fc_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
mkl_speed_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
momentum_sgd_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
mpi_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
normalize_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
one_hot_ops_test.py Add BatchBucketOneHotOp 2017-10-06 13:25:12 -07:00
pack_ops_test.py PackSegments: return value presence. 2017-10-12 11:17:34 -07:00
pack_rnn_sequence_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
pad_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
partition_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
piecewise_linear_transform_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
pooling_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
prepend_dim_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
python_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
rank_loss_operator_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
rebatching_queue_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
record_queue_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
recurrent_net_executor_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
recurrent_network_test.py Switch RNNOp to use NetDef argument for step represenetation. 2017-10-10 22:01:51 -07:00
reduce_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
reduction_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
relu_op_test.py Kick fbsync 2017-09-29 16:31:11 -07:00
reshape_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
resize_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
rmac_regions_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
rnn_cell_test.py Switch RNNOp to use NetDef argument for step represenetation. 2017-10-10 22:01:51 -07:00
segment_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sequence_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
shape_inference_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sinusoid_position_encoding_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
softmax_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
softplus_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sparse_gradient_checker_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sparse_lengths_sum_benchmark.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sparse_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
sparse_to_dense_mask_op_test.py Add int64 support to sparse_to_dense_mask_op 2017-10-10 17:19:44 -07:00
spatial_bn_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
specialized_segment_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
square_root_divide_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
stats_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
string_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
text_file_reader_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
tile_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
top_k_test.py add cuda support for Topk Gradient 2017-10-04 19:31:56 -07:00
unique_uniform_fill_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
utility_ops_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
video_input_op_test.py Re-license to Apache 2017-09-28 16:22:00 -07:00
weighted_sample_test.py Add weighted_sampling operator to Caffe2 2017-10-05 20:33:59 -07:00