Searched defs:max_num_elements (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | list_ops_test.py | 54 def _testPushPop(self, max_num_elements): argument
68 def testPushPop(self, max_num_elements): argument
74 def testPushPopGPU(self, max_num_elements): argument
91 def testPopFromEmptyTensorListFails(self, max_num_elements): argument
156 def _testStack(self, max_num_elements): argument
170 def testStack(self, max_num_elements): argument
176 def testStackGPU(self, max_num_elements): argument
183 def testStackWithUnknownElementShape(self, max_num_elements): argument
205 def testStackWithPartiallyDefinedElementShape(self, max_num_elements): argument
228 def testStackEmptyList(self, max_num_elements): argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | parse_tensor_op_fuzz.cc | 58 const int64 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/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 53 max_num_elements=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 216 int64 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 | 552 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 | 1374 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample() local
|