| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 80 tensorflow::Tensor MakeTensor(const std::vector<int64_t>& shape, in MakeTensor() function
198 tensorflow::Tensor t1 = MakeTensor<float>( in TEST()
215 tensorflow::Tensor t1 = MakeTensor<float>( in TEST()
217 tensorflow::Tensor t2 = MakeTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}, in TEST()
228 tensorflow::Tensor t1 = MakeTensor<float>( in TEST()
243 tensorflow::Tensor t1 = MakeTensor<float>( in TEST()
|
| D | buffer_map_util.cc | 186 tensorflow::Tensor t = tensorflow::TensorCApi::MakeTensor( in SetTfTensorFromTfLite()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_wrapper_utils.cc | 35 void MakeTensor(const string& name, std::unique_ptr<TensorT>* tensor) { in MakeTensor() function
108 MakeTensor(name, &intermediate_tensor); in AddIntermediateTensorsToFusedOp()
|
| D | model_utils.cc | 85 void MakeTensor(const string& name, const std::vector<int32_t>& shape, in MakeTensor() function
106 MakeTensor(name, shape, shape_signature, type, tensor); in MakeTensorWithQuantParam()
|
| D | model_utils.h | 38 void MakeTensor(const string& name, const std::vector<int32_t>& shape,
|
| D | quantize_weights_portable.cc | 469 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature,
553 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| D | quantize_weights.cc | 483 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature,
567 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| D | quantize_model.cc | 428 utils::MakeTensor(leading_op_name, tensor->shape, tensor->shape_signature, in SetInputType()
490 utils::MakeTensor(tailing_op_name, tensor->shape, tensor->shape_signature, in SetOutputType()
1018 utils::MakeTensor(tensor->name + "_" + type_string, tensor->shape, in QuantizeOpInput()
1048 utils::MakeTensor(tensor->name + "_float", tensor->shape, in QuantizeOpInput()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared_test.cc | 226 static Node* MakeTensor(Graph* g, int B, int M, int N) { in MakeTensor() function
282 Node* dense = MakeTensor(g, 8, 4, N); \
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor_internal.h | 77 static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, in MakeTensor() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 307 static Tensor MakeTensor(DataType dtype, const TensorShape& shape, in MakeTensor() function
372 MakeTensor(output_dtype, output_shape, output_buffer, output_allocator); in GetOrCreateTensorForOutput()
|