Searched refs:per_axis_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.h | 61 int per_axis_index = 0; member
|
| D | quantization_utils.h | 141 int per_axis_index, ErrorReporter* error_reporter);
|
| D | quantize_model.cc | 636 tensor_property.per_axis_index, in QuantizeOpInput()
1130 weight_property.per_axis, weight_property.per_axis_index, in QuantizeBiases()
1214 int32_t channel_dim_index = input.second.per_axis_index; in FillQuantizationParams()
1236 input.second.per_axis_index) { in FillQuantizationParams()
1241 input.second.per_axis_index, in FillQuantizationParams()
1362 weight_tensor->shape[weight_property.per_axis_index]) { in EnsureBiasScaleCompatibility()
1367 weight_tensor->shape[weight_property.per_axis_index]); in EnsureBiasScaleCompatibility()
|
| D | operator_property.cc | 149 tensor_property.per_axis_index = 0; in GetOperatorProperty()
160 tensor_property.per_axis_index = 0; in GetOperatorProperty()
171 tensor_property.per_axis_index = 3; in GetOperatorProperty()
|
| D | quantization_utils.cc | 698 int per_axis_index, ErrorReporter* error_reporter) { in QuantizeWeight() argument
703 return SymmetricQuantizeTensorPerChannel(model, tensor, per_axis_index, in QuantizeWeight()
|
| D | quantization_utils_test.cc | 493 int per_axis_index = 0; in TEST_F() local
495 EXPECT_EQ(QuantizeWeight(&model, weights_tensor, per_channel, per_axis_index, in TEST_F()
|