Searched defs:num_repeats (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | from_generator_test.py | 38 def _testFromGenerator(self, generator, elem_sequence, num_repeats, argument
53 def testFromGeneratorUsingFn(self, num_repeats, requires_initialization): argument
71 def testFromGeneratorUsingList(self, num_repeats, requires_initialization): argument
85 def testFromGeneratorUsingNdarray(self, num_repeats, requires_initialization): argument
99 def testFromGeneratorUsingGeneratorExpression(self, num_repeats, argument
|
| D | shuffle_test.py | 428 num_repeats=5, argument
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | RepeatInterleave.cpp | 39 const ValueRef num_repeats, in add_repeat_interleave_node()
80 const ValueRef num_repeats = args[args_i++]; in repeat_interleave() local
|
| D | SDPA.cpp | 262 const ValueRef num_repeats = in sdpa_with_kv_cache_impl() local
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | sql_dataset_test.py | 42 num_repeats=1): argument
595 def _build_dataset(self, num_repeats): argument
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_ops_test.h | 59 int num_repeats) { in RepeatElements()
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_test_blit_perf.c | 524 unsigned num_warmup_repeats = 1, num_repeats = 4; in si_test_blit_perf() local
|
| /external/eigen/test/ |
| D | array_cwise.cpp | 46 const int num_repeats = 32; in pow_test() local
|
| /external/executorch/backends/vulkan/test/op_tests/ |
| D | sdpa_test.cpp | 187 const int num_repeats = q_projected.size(2) / key_cache.size(2); in sdpa_reference_impl() local
|
| /external/pytorch/test/ |
| D | test_mps.py | 3890 def helper(shape, dtype=torch.float32, num_repeats=torch.Tensor(), dim=None): argument
|