mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Summary: new SumReduceLikeOp CPU/GPU implementation and doc. Unit tests and NMT team tests passed. Some benchmark results here: shape(A) = [100, 1000, 100] shape(B) = [1000] 0.36684 ms/iter (0.00122679 ms/iter) SumReduceLike 0.246593 ms/iter (0.00151116 ms/iter) ReduceBackSum 0.202563 ms/iter (0.00511932 ms/iter) ReduceFrontSum // This means that we are faster than back+front sum now shape(A) = [32, 32, 100] shape(B) = [32, 100] 0.0253826 ms/iter (0.00257504 ms/iter) ReduceFrontSum 0.0233368 ms/iter (0.00118283 ms/iter) SumReduceLike shape(A) = [32, 32, 100] shape(B) = [32, 32] 0.0276206 ms/iter (0.00691918 ms/iter) ReduceBackSum 0.0254768 ms/iter (0.00325529 ms/iter) SumReduceLike Reviewed By: Yangqing Differential Revision: D4873222 fbshipit-source-id: 736b1537998f4289876bc53d38607b8052e89c70 |
||
|---|---|---|
| .. | ||
| activation_ops_test.py | ||
| atomic_ops_test.py | ||
| checkpoint_test.py | ||
| conv_test.py | ||
| conv_transpose_test.py | ||
| copy_ops_test.py | ||
| cosine_embedding_criterion_op_test.py | ||
| counter_ops_test.py | ||
| crf_test.py | ||
| cross_entropy_ops_test.py | ||
| dataset_ops_test.py | ||
| duplicate_operands_test.py | ||
| elementwise_op_broadcast_test.py | ||
| elementwise_ops_test.py | ||
| emptysample_ops_test.py | ||
| extend_tensor_op_test.py | ||
| fc_operator_test.py | ||
| filler_ops_test.py | ||
| gather_ops_test.py | ||
| gather_ranges_op_test.py | ||
| given_tensor_fill_op_test.py | ||
| group_conv_test.py | ||
| hsm_test.py | ||
| index_ops_test.py | ||
| instance_norm_test.py | ||
| lengths_tile_op_test.py | ||
| loss_ops_test.py | ||
| margin_ranking_criterion_op_test.py | ||
| matmul_op_test.py | ||
| mkl_conv_op_test.py | ||
| mkl_packed_fc_op_test.py | ||
| mkl_speed_test.py | ||
| momentum_sgd_test.py | ||
| mpi_test.py | ||
| one_hot_ops_test.py | ||
| pack_ops_test.py | ||
| pad_test.py | ||
| partition_ops_test.py | ||
| piecewise_linear_transform_test.py | ||
| pooling_test.py | ||
| pow_op_test.py | ||
| python_op_test.py | ||
| rank_loss_operator_test.py | ||
| record_queue_test.py | ||
| recurrent_network_test.py | ||
| reduce_ops_test.py | ||
| reduction_ops_test.py | ||
| relu_op_test.py | ||
| reshape_ops_test.py | ||
| resize_op_test.py | ||
| segment_ops_test.py | ||
| sequence_ops_test.py | ||
| shape_inference_test.py | ||
| softmax_ops_test.py | ||
| sparse_gradient_checker_test.py | ||
| sparse_ops_test.py | ||
| spatial_bn_op_test.py | ||
| square_root_divide_op_test.py | ||
| stats_ops_test.py | ||
| string_ops_test.py | ||
| text_file_reader_test.py | ||
| tile_op_test.py | ||
| top_k_test.py | ||
| unique_uniform_fill_op_test.py | ||
| utility_ops_test.py | ||