| /external/pytorch/benchmarks/operator_benchmark/pt/ |
| D | qembedding_pack_test.py | 43 def init(self, num_embeddings, embedding_dim, op_func): argument
54 def init(self, num_embeddings, embedding_dim, batch_size, op_func): argument
68 def init(self, num_embeddings, embedding_dim, op_func): argument
78 def init(self, num_embeddings, embedding_dim, batch_size, op_func): argument
|
| D | pool_test.py | 43 def init(self, kernel, stride, N, C, L, device, op_func): argument
99 def init(self, kernel, stride, N, C, H, W, device, op_func): argument
157 def init(self, kernel, stride, N, C, D, H, W, device, op_func): argument
|
| D | softmax_test.py | 80 def init(self, N, C, H, W, device, op_func): argument
89 def init(self, M, N, dim, device, op_func): argument
|
| D | matrix_mult_test.py | 45 def init(self, B, M, N, K, device, op_func): argument
95 def init(self, B, M, N, device, op_func): argument
|
| D | qarithmetic_test.py | 51 def init(self, N, dtype, contig, op_func): argument
71 def init(self, N, dtype, contig, op_func): argument
|
| D | binary_test.py | 32 def init(self, in_one, in_two, dtype, device, op_func): argument
88 def init(self, M, N, K, device, dtype_one, dtype_two, op_func): argument
|
| D | qobserver_test.py | 107 def init(self, C, M, N, dtype, qscheme, op_func, device): argument
117 def init(self, C, M, N, dtype, qscheme, op_func, device): argument
|
| D | qembedding_bag_lookups_test.py | 100 op_func, argument
211 op_func, argument
|
| D | nan_to_num_test.py | 39 def init(self, M, N, dtype, replace_inf, op_func): argument
|
| D | remainder_test.py | 43 def init(self, M, N, K, device, dtype, op_func): argument
|
| D | hardswish_test.py | 40 def init(self, N, C, H, W, device, op_func): argument
|
| D | hardsigmoid_test.py | 40 def init(self, N, C, H, W, device, op_func): argument
|
| D | qcomparators_test.py | 29 def init(self, N, dtype, contig, other_scalar, out_variant, op_func): argument
|
| D | qunary_test.py | 30 def init(self, M, N, dtype, op_func): argument
|
| D | qactivation_test.py | 78 def init(self, dims, contig, inplace, dtype, op_func): argument
|
| D | unary_test.py | 27 def init(self, M, N, device, op_func): argument
|
| D | quantization_test.py | 235 def init(self, N, C, H, W, zero_point_dtype, nbits, device, op_func): argument
315 def init(self, N, C, H, W, zero_point_dtype, nbits, device, op_func): argument
|
| /external/pytorch/benchmarks/operator_benchmark/common/tests/ |
| D | add_ops_list_test.py | 26 def init(self, M, N, op_func): argument
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2152 def _convert_softmax(pfor_input, op_type, op_func): argument
2165 def _convert_identity(pfor_input, op_type, op_func): argument
2231 def _convert_searchsorted(pfor_input, _, op_func): argument
2473 def _convert_transpose(pfor_input, _, op_func): argument
2804 def _convert_reduction(pfor_input, _, op_func): argument
2815 def _convert_argmax_argmin(pfor_input, _, op_func): argument
2841 def _convert_cumfoo(pfor_input, _, op_func): argument
2883 def _convert_unsortedsegmentsum(pfor_input, _, op_func): argument
2930 def _convert_sparse_segment(pfor_input, _, op_func): argument
2969 def _convert_sparse_segment_grad(pfor_input, _, op_func): argument
[all …]
|
| D | math_test.py | 526 def test_sparse_segment(self, op_func, with_num_segments): argument
563 def test_sparse_segment_grad(self, op_func): argument
|
| /external/pytorch/torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/ |
| D | _common.py | 84 op_func, argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter_test.py | 54 def makeApiInfoForGenOp(self, op_name, op_func): argument
|
| /external/pytorch/test/torch_np/numpy_tests/core/ |
| D | test_scalarmath.py | 918 def op_func(self, other): function
941 def op_func(self, other): function
|
| /external/autotest/server/hosts/ |
| D | remote.py | 260 def op_func(): function
|
| /external/autotest/client/common_lib/hosts/ |
| D | base_classes.py | 552 def log_op(self, op, op_func): argument
|