Home
last modified time | relevance | path

Searched refs:QuantizeWeight (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils.h140 TfLiteStatus QuantizeWeight(ModelT* model, TensorT* tensor, bool per_channel,
Dquantization_utils.cc697 TfLiteStatus QuantizeWeight(ModelT* model, TensorT* tensor, bool per_channel, in QuantizeWeight() function
Dquantization_utils_test.cc495 EXPECT_EQ(QuantizeWeight(&model, weights_tensor, per_channel, per_axis_index, in TEST_F()
Dquantize_model.cc635 if (utils::QuantizeWeight(model, tensor, tensor_property.per_axis, in QuantizeOpInput()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dprepare-quantize.mlir687 // CHECK-LABEL: QuantizeWeight
688 func @QuantizeWeight(%arg0: tensor<1x224x224x3xf32>) -> tensor<1x112x112x32xf32> {