Searched refs:BuildTensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_resize_bilinear_op_test.cc | 39 Tensor BuildTensor(const int batch_size, const int height, const int width, in BuildTensor() function
52 Tensor BuildTensor<float>(const int batch_size, const int height, in BuildTensor() function
307 const Tensor image_quantized_tensor = BuildTensor<T>( in RunTestResizeBilinearTwoDims()
331 const Tensor image_quantized_tensor = BuildTensor<T>( in RunBenchmarkResizeBilinearTwoDims()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | list_ops_test.py | 1309 def BuildTensor(element_shape): function
1314 self.assertIsNone(BuildTensor(None).shape.rank)
1315 self.assertAllEqual(BuildTensor([None, 2, 3]).shape.as_list(), [None, 2, 3])
1317 BuildTensor([None, 2, None]).shape.as_list(), [None, 2, None])
1318 self.assertAllEqual(BuildTensor([1, 2, 3]).shape.as_list(), [None, 2, 3])
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 540 Optional<BufferOffset<tflite::Tensor>> BuildTensor(
791 Optional<BufferOffset<tflite::Tensor>> Translator::BuildTensor( in BuildTensor() function in __anonaf2241e10111::Translator
1382 BuildTensor(value, tensor_name, buffers_.size(), quant_parameters); in BuildSubGraph()
|