Home
last modified time | relevance | path

Searched defs:op_func (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dqembedding_pack_test.py43 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
H A Dpool_test.py43 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
H A Dsoftmax_test.py80 def init(self, N, C, H, W, device, op_func): argument
89 def init(self, M, N, dim, device, op_func): argument
H A Dmatrix_mult_test.py45 def init(self, B, M, N, K, device, op_func): argument
95 def init(self, B, M, N, device, op_func): argument
H A Dqarithmetic_test.py51 def init(self, N, dtype, contig, op_func): argument
71 def init(self, N, dtype, contig, op_func): argument
H A Dbinary_test.py32 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
H A Dqobserver_test.py107 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
H A Dqembedding_bag_lookups_test.py100 op_func, argument
211 op_func, argument
H A Dquantization_test.py235 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
H A Dnan_to_num_test.py39 def init(self, M, N, dtype, replace_inf, op_func): argument
H A Dremainder_test.py43 def init(self, M, N, K, device, dtype, op_func): argument
H A Dhardswish_test.py40 def init(self, N, C, H, W, device, op_func): argument
H A Dhardsigmoid_test.py40 def init(self, N, C, H, W, device, op_func): argument
H A Dqcomparators_test.py29 def init(self, N, dtype, contig, other_scalar, out_variant, op_func): argument
H A Dqactivation_test.py78 def init(self, dims, contig, inplace, dtype, op_func): argument
H A Dqunary_test.py30 def init(self, M, N, dtype, op_func): argument
H A Dunary_test.py27 def init(self, M, N, device, op_func): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py2152 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 …]
H A Dmath_test.py526 def test_sparse_segment(self, op_func, with_num_segments): argument
563 def test_sparse_segment_grad(self, op_func): argument
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/
H A Dtest_scalarmath.py918 def op_func(self, other): function
941 def op_func(self, other): function
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/common/tests/
H A Dadd_ops_list_test.py26 def init(self, M, N, op_func): argument
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/
H A D_common.py84 op_func, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dpython_api_parameter_converter_test.py54 def makeApiInfoForGenOp(self, op_name, op_func): argument
/aosp_15_r20/external/autotest/server/hosts/
H A Dremote.py260 def op_func(): function
/aosp_15_r20/external/autotest/client/common_lib/hosts/
H A Dbase_classes.py552 def log_op(self, op, op_func): argument

12