Searched refs:QuantizeWeight (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/tools/optimize/ |
D | quantization_utils.h | 140 TfLiteStatus QuantizeWeight(ModelT* model, TensorT* tensor, bool per_channel,
|
D | quantization_utils.cc | 697 TfLiteStatus QuantizeWeight(ModelT* model, TensorT* tensor, bool per_channel, in QuantizeWeight() function
|
D | quantization_utils_test.cc | 495 EXPECT_EQ(QuantizeWeight(&model, weights_tensor, per_channel, per_axis_index, in TEST_F()
|
D | quantize_model.cc | 635 if (utils::QuantizeWeight(model, tensor, tensor_property.per_axis, in QuantizeOpInput()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
D | prepare-quantize.mlir | 687 // CHECK-LABEL: QuantizeWeight 688 func @QuantizeWeight(%arg0: tensor<1x224x224x3xf32>) -> tensor<1x112x112x32xf32> {
|