Home
last modified time | relevance | path

Searched refs:input_params (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc280 auto* input_params = reinterpret_cast<TfLiteAffineQuantization*>( in TEST_F() local
282 input_params->scale = TfLiteFloatArrayCreate(1); in TEST_F()
283 input_params->scale->data[0] = 0.5; in TEST_F()
284 input_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
285 input_params->zero_point->data[0] = 5; in TEST_F()
286 input.quantization.params = reinterpret_cast<void*>(input_params); in TEST_F()
386 auto* input_params = reinterpret_cast<TfLiteAffineQuantization*>( in TEST_F() local
388 input_params->scale = TfLiteFloatArrayCreate(1); in TEST_F()
389 input_params->scale->data[0] = 0.5; in TEST_F()
390 input_params->zero_point = TfLiteIntArrayCreate(1); in TEST_F()
[all …]
Delementwise.cc102 const auto* input_params = in GenericPrepare() local
106 TF_LITE_ENSURE(context, input_params != nullptr); in GenericPrepare()
107 TF_LITE_ENSURE(context, input_params->scale != nullptr); in GenericPrepare()
108 TF_LITE_ENSURE(context, input_params->scale->size > 0); in GenericPrepare()
109 TF_LITE_ENSURE(context, input_params->zero_point->size > 0); in GenericPrepare()
114 op_data->input_offset = input_params->zero_point->data[0]; in GenericPrepare()
120 const float input_scale = input_params->scale->data[0]; in GenericPrepare()
Dsvdf.cc258 auto* input_params = in Prepare() local
268 const double effective_scale_1 = input_params->scale->data[0] * in Prepare()
374 auto* input_params = reinterpret_cast<TfLiteAffineQuantization*>( in Eval() local
396 op_data->effective_scale_2_b, input_params->zero_point->data[0], in Eval()
/external/tensorflow/tensorflow/lite/c/
Dc_api_test.cc68 TfLiteQuantizationParams input_params = in TEST() local
70 EXPECT_EQ(input_params.scale, 0.f); in TEST()
71 EXPECT_EQ(input_params.zero_point, 0); in TEST()
127 TfLiteQuantizationParams input_params = in TEST() local
129 EXPECT_EQ(input_params.scale, 0.003922f); in TEST()
130 EXPECT_EQ(input_params.zero_point, 0); in TEST()
Dc_test.c98 TfLiteQuantizationParams input_params = in TestSmokeTest() local
100 ASSERT_EQ(input_params.scale, 0.f); in TestSmokeTest()
101 ASSERT_EQ(input_params.zero_point, 0); in TestSmokeTest()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc193 for (const GraphTransferNodeInputInfo& input_params : in SetupGraph() local
195 inputs_count += input_params.node_input_size(); in SetupGraph()
208 for (const GraphTransferNodeInputInfo& input_params : in SetupGraph() local
210 const int count = input_params.node_input_size(); in SetupGraph()
215 const GraphTransferNodeInput& node_input = input_params.node_input(i); in SetupGraph()
220 const int node_id = input_params.node_id(); in SetupGraph()
Dgraph_transferer_test.cc242 const GraphTransferNodeInputInfo* input_params = in SanityCheckNodes() local
244 ASSERT_NE(nullptr, input_params); in SanityCheckNodes()
245 EXPECT_EQ(params.input_count(), input_params->node_input_size()); in SanityCheckNodes()
246 EXPECT_EQ(params.node_id(), input_params->node_id()); in SanityCheckNodes()
248 input_params->node_input()) { in SanityCheckNodes()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc284 auto input_params = graph_properties.GetInputProperties(node->name()); in GetInputProperties() local
285 auto in_shape = input_params.at(in_port); in GetInputProperties()