Searched defs:max_num_elements (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | list_ops_test.py | 50 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | parse_tensor_op_fuzz.cc | 58 const int64_t max_num_elements = 1 << 18; in FuzzImpl() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_list.h | 123 int max_num_elements = -1; variable
|
| /external/icing/icing/file/ |
| D | file-backed-vector_test.cc | 603 int32_t max_num_elements = in TEST_F() local
695 int32_t max_num_elements = in TEST_F() local
816 int32_t max_num_elements = (max_file_size - header_size) / element_type_size; in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 49 max_num_elements=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 220 int64_t max_num_elements; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 470 auto max_num_elements = in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 554 const size_t max_num_elements = max_num_features / stride_size; in FastParseExampleLite() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1390 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample() local
|