Home
last modified time | relevance | path

Searched refs:MakeTensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc78 tensorflow::Tensor MakeTensor(const std::vector<int>& shape, in MakeTensor() function
171 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
188 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
189 tensorflow::Tensor t2 = MakeTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}); in TEST()
200 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
215 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
Dbuffer_map.cc163 tensorflow::Tensor t = tensorflow::TensorCApi::MakeTensor( in SetFromTfLite()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_dense_binary_op_shared_test.cc225 static Node* MakeTensor(Graph* g, int B, int M, int N) { in MakeTensor() function
280 Node* dense = MakeTensor(g, 8, 4, N); \
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc224 void MakeTensor(const string& name, const std::vector<int32_t>& shape, in MakeTensor() function
318 MakeTensor(dequant_name, tensor->shape, &dequantize_output); in QuantizeWeightsInternal()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.h221 static Tensor MakeTensor(DataType dtype, const TensorShape& shape, in MakeTensor() function
Dxla_launch_util.cc370 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs()
434 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs()
/external/tensorflow/tensorflow/c/
Dc_api_internal.h200 static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, in MakeTensor() function
Dc_api.cc508 *dst = TensorCApi::MakeTensor(src->dtype, src->shape, src->buffer); in TF_TensorToTensor()
/external/tensorflow/tensorflow/python/lib/core/
Dpy_func.cc323 Tensor MakeTensor(DataType dtype, const TensorShape& shape) { in MakeTensor() function in tensorflow::NumpyTensorBuffer
414 *ret = buf->MakeTensor(dtype, shape); in ConvertNdarrayToTensor()