| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 72 tensorflow::Tensor MakeTensor(const std::vector<int>& shape, in MakeTensor() function
165 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
182 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
183 tensorflow::Tensor t2 = MakeTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}); in TEST()
194 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
209 MakeTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
|
| D | buffer_map.cc | 173 tensorflow::Tensor t = tensorflow::TensorCApi::MakeTensor( in SetFromTfLite()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_wrapper_utils.cc | 33 void MakeTensor(const string& name, std::unique_ptr<TensorT>* tensor) { in MakeTensor() function
106 MakeTensor(name, &intermediate_tensor); in AddIntermediateTensorsToFusedOp()
|
| D | model_utils.cc | 84 void MakeTensor(const string& name, const std::vector<int32_t>& shape, in MakeTensor() function
105 MakeTensor(name, shape, shape_signature, type, tensor); in MakeTensorWithQuantParam()
|
| D | model_utils.h | 36 void MakeTensor(const string& name, const std::vector<int32_t>& shape,
|
| D | quantize_weights.cc | 327 void MakeTensor(const string& name, const std::vector<int32_t>& shape, in MakeTensor() function
509 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsInt8()
593 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| D | quantize_model.cc | 277 utils::MakeTensor(leading_op_name, tensor->shape, tensor->shape_signature, in SetInputType()
339 utils::MakeTensor(tailing_op_name, tensor->shape, tensor->shape_signature, in SetOutputType()
739 utils::MakeTensor(tensor->name + "_" + type_string, tensor->shape, in QuantizeOpInput()
769 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 | 76 static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, in MakeTensor() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 290 static Tensor MakeTensor(DataType dtype, const TensorShape& shape, in MakeTensor() function
326 return MakeTensor(output_dtype, output_shape, output_buffer, in GetOrCreateTensorForOutput()
|