Searched refs:weights_max (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() 129 const float weights_max = 127.0f; in TF_LITE_MICRO_TEST() local 149 weights_data, weights_min, weights_max, bias_dims_data, bias_data, in TF_LITE_MICRO_TEST() 158 const float weights_max = 127.0f; in TF_LITE_MICRO_TEST() local 184 weights_dims_data_local, weights_data_local, weights_min, weights_max, in TF_LITE_MICRO_TEST() 195 const float weights_max = 127.0f; in TF_LITE_MICRO_TEST() local 220 weights_dims_data_2, weights_data_2, weights_min, weights_max, in TF_LITE_MICRO_TEST() 230 const float weights_max = 127.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 80 *weights_max = 127 * scale_max; in ProcessPerChannelQuantizedWeights() 231 float weights_max = 0; in InitializeWeightsNodes() local 234 &weights_max, graph_builder_, in InitializeWeightsNodes() 238 weights_tensor, &weights_min, &weights_max)); in InitializeWeightsNodes() 244 quant_bound_shape.data(), reinterpret_cast<char*>(&weights_max), in InitializeWeightsNodes() 245 sizeof(weights_max)); in InitializeWeightsNodes()
|
D | transpose_conv_2d_builder.cc | 71 float weights_max = 0; in PopulateSubGraph() local 74 &weights_max, graph_builder_, in PopulateSubGraph() 78 weights_tensor, &weights_min, &weights_max)); in PopulateSubGraph() 83 kScalarShape, reinterpret_cast<char*>(&weights_max), sizeof(weights_max)); 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 | 3014 auto weights_max = ops::Placeholder(s.WithOpName("weights_max"), DT_FLOAT); in TEST_F() local 3016 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3035 auto weights_max = ops::Placeholder(s.WithOpName("weights_max"), DT_FLOAT); in TEST_F() local 3037 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3056 auto weights_max = ops::Placeholder(s.WithOpName("weights_max"), DT_FLOAT); in TEST_F() local 3058 s.WithOpName("my_quantize"), input, weights_min, weights_max); in TEST_F() 3074 auto weights_max = ops::Placeholder(s.WithOpName("weights_max"), DT_FLOAT); in TEST_F() local 3077 s.WithOpName("my_quantize"), input, weights_min, weights_max, num_bits); in TEST_F()
|