Searched refs:CreateFloatTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | eval_pipeline_test.cc | 25 Tensor CreateFloatTensor(float value) { in CreateFloatTensor() function
70 TF_CHECK_OK(eval_pipeline.Run(CreateFloatTensor(5), CreateFloatTensor(27))); in TEST()
98 auto status = eval_pipeline.Run(string_tensor, CreateFloatTensor(27)); in TEST()
118 auto status = eval_pipeline.Run(CreateFloatTensor(5), CreateFloatTensor(27)); in TEST()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/testing/ |
| D | test_utils.h | 107 inline TfLiteTensor CreateFloatTensor(const float* data, TfLiteIntArray* dims, in CreateFloatTensor() function
117 inline TfLiteTensor CreateFloatTensor(std::initializer_list<float> data, in CreateFloatTensor() function
119 return CreateFloatTensor(data.begin(), dims, name); in CreateFloatTensor()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | depthwise_conv_test.cc | 47 CreateFloatTensor(input_data, input_dims, "input_tensor"), in TestDepthwiseConvFloat()
48 CreateFloatTensor(filter_data, filter_dims, "filter_tensor"), in TestDepthwiseConvFloat()
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestDepthwiseConvFloat()
50 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestDepthwiseConvFloat()
|
| D | softmax_test.cc | 40 CreateFloatTensor(input_data, input_dims, "input_tensor"), in TestSoftmaxFloat()
41 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestSoftmaxFloat()
|
| D | fully_connected_test.cc | 47 CreateFloatTensor(input_data, input_dims, "input_tensor"), in TestFullyConnectedFloat()
48 CreateFloatTensor(weights_data, weights_dims, "weights_tensor"), in TestFullyConnectedFloat()
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestFullyConnectedFloat()
50 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestFullyConnectedFloat()
|