Searched refs:weights_min (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
D | fully_connected_slicing_test.cc | 41 const float weights_min, const float weights_max, const int* bias_dims_data, in TestFullyConnectedQuantized() argument 59 weights_min, weights_max), in TestFullyConnectedQuantized() 128 const float weights_min = -128.0f; in TF_LITE_MICRO_TEST() local 149 weights_data, weights_min, weights_max, bias_dims_data, bias_data, in TF_LITE_MICRO_TEST() 157 const float weights_min = -128.0f; in TF_LITE_MICRO_TEST() local 184 weights_dims_data_local, weights_data_local, weights_min, weights_max, in TF_LITE_MICRO_TEST() 194 const float weights_min = -128.0f; in TF_LITE_MICRO_TEST() local 220 weights_dims_data_2, weights_data_2, weights_min, weights_max, in TF_LITE_MICRO_TEST() 229 const float weights_min = -128.0f; in TF_LITE_MICRO_TEST() local 256 weights_dims_data_local_2, weights_data_local_2, weights_min, weights_max, in TF_LITE_MICRO_TEST() [all …]
|
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
D | conv_2d_helpers.cc | 43 TfLiteContext* context, float* weights_min, float* weights_max, in ProcessPerChannelQuantizedWeights() argument 79 *weights_min = -128 * scale_max; in ProcessPerChannelQuantizedWeights() 230 float weights_min = 0; in InitializeWeightsNodes() local 233 ProcessPerChannelQuantizedWeights(inputs, outputs, context, &weights_min, in InitializeWeightsNodes() 238 weights_tensor, &weights_min, &weights_max)); in InitializeWeightsNodes() 241 quant_bound_shape.data(), reinterpret_cast<char*>(&weights_min), in InitializeWeightsNodes() 242 sizeof(weights_min)); in InitializeWeightsNodes()
|
D | transpose_conv_2d_builder.cc | 70 float weights_min = 0; in PopulateSubGraph() local 73 ProcessPerChannelQuantizedWeights(inputs, outputs, context, &weights_min, in PopulateSubGraph() 78 weights_tensor, &weights_min, &weights_max)); in PopulateSubGraph() 81 kScalarShape, reinterpret_cast<char*>(&weights_min), sizeof(weights_min)); in PopulateSubGraph()
|
D | conv_2d_builder.h | 88 TfLiteContext* context, float* weights_min, float* weights_max,
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes_test.cc | 3013 auto weights_min = ops::Placeholder(s.WithOpName("weights_min"), DT_FLOAT); in TEST_F() local 3016 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3034 auto weights_min = ops::Placeholder(s.WithOpName("weights_min"), DT_FLOAT); in TEST_F() local 3037 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3055 auto weights_min = ops::Placeholder(s.WithOpName("weights_min"), DT_FLOAT); in TEST_F() local 3058 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3073 auto weights_min = ops::Placeholder(s.WithOpName("weights_min"), DT_FLOAT); in TEST_F() local 3077 s.WithOpName("my_quantize"), input, weights_min, weights_max, num_bits); in TEST_F()
|