Home
last modified time | relevance | path

Searched refs:BuildTensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_resize_bilinear_op_test.cc39 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlist_ops_test.py1350 def BuildTensor(element_shape): function
1355 self.assertIsNone(BuildTensor(None).shape.rank)
1356 self.assertAllEqual(BuildTensor([None, 2, 3]).shape.as_list(), [None, 2, 3])
1358 BuildTensor([None, 2, None]).shape.as_list(), [None, 2, None])
1359 self.assertAllEqual(BuildTensor([1, 2, 3]).shape.as_list(), [None, 2, 3])
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_reader.cc257 Status s = tensorflow::Tensor::BuildTensor(type, shape, t.get()); in GetTensor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc556 Optional<BufferOffset<tflite::Tensor>> BuildTensor(
871 Optional<BufferOffset<tflite::Tensor>> Translator::BuildTensor( in BuildTensor() function in __anon46e5f16b0111::Translator
1482 BuildTensor(value, tensor_name, buffers_.size(), quant_parameters); in BuildSubGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor.h185 static Status BuildTensor(DataType type, const TensorShape& shape,
H A Dtensor.cc845 Status Tensor::BuildTensor(DataType type, const TensorShape& shape, in BuildTensor() function in tensorflow::Tensor