mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
b82f9e9ea7
Summary: Simple FindOp for CPU and GPU which searches a list of unordered needles from an unordered index. CPU version might be faster if first sorting the index / needles, but we can get back to that later. CUDA op is also kind of brutish, but pretty parallel. Since the index and the queries are smallish at least in the use case currently in mind (Machine Translation's team word candidate search), I think this is a sufficient start. Note that this is much simpler than the Index-class of ops which allow modifying the index etc. Since CUDA ops are more complex to implement for the full Index functionality, I decided to make a separate op with this very simple functionality. Differential Revision: D4910131 fbshipit-source-id: 6df35c9e3c71d5392a500d5b98fd708ab0c8e587 |
||
|---|---|---|
| .. | ||
| activation_ops_test.py | ||
| adagrad_test.py | ||
| adam_test.py | ||
| atomic_ops_test.py | ||
| boolean_unmask_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_linear_op_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 | ||
| find_op_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 | ||
| rnn_cell_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 | ||