Home
last modified time | relevance | path

Searched defs:max_num_elements (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlist_ops_test.py50 def _testPushPop(self, max_num_elements): argument
64 def testPushPop(self, max_num_elements): argument
70 def testPushPopGPU(self, max_num_elements): argument
87 def testPopFromEmptyTensorListFails(self, max_num_elements): argument
162 def _testStack(self, max_num_elements): argument
176 def testStack(self, max_num_elements): argument
182 def testStackGPU(self, max_num_elements): argument
189 def testStackWithUnknownElementShape(self, max_num_elements): argument
211 def testStackWithPartiallyDefinedElementShape(self, max_num_elements): argument
234 def testStackEmptyList(self, max_num_elements): argument
[all …]
/aosp_15_r20/external/icing/icing/file/
H A Dfile-backed-vector_test.cc719 int32_t max_num_elements = in TEST_F() local
811 int32_t max_num_elements = in TEST_F() local
938 int32_t max_num_elements = (max_file_size - header_size) / element_type_size; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Dparse_tensor_op_fuzz.cc58 const int64_t max_num_elements = 1 << 18; in FuzzImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_list.h123 int max_num_elements = -1; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlist_ops.py49 max_num_elements=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util_test.cc470 auto max_num_elements = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtensor_list_ops.cc220 int64_t max_num_elements; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/
H A Dparse_example.cc554 const size_t max_num_elements = max_num_features / stride_size; in FastParseExampleLite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_fast_parsing.cc1390 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample() local