Home
last modified time | relevance | path

Searched refs:scales_ (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/native/annotator/
Dmodel-executor.cc100 scales_(scales), in TFLiteEmbeddingExecutor()
177 if (!DequantizeAdd(scales_->data.f, embeddings_->data.uint8, in AddEmbedding()
Dmodel-executor.h110 const TfLiteTensor* scales_ = nullptr; variable
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.cc1530 bool scales_are_valid = (depth == scales_.size()); in GetBiasHandle()
1531 scales_.resize(depth); in GetBiasHandle()
1537 if (scales_are_valid && std::abs(tmp_scale - scales_[i]) > 1e-6) { in GetBiasHandle()
1540 scales_[i] = tmp_scale; in GetBiasHandle()
1545 bias_attr.set_output_scales(0, scales_); in GetBiasHandle()
1547 bias_attr.set_output_scales(1, scales_); in GetBiasHandle()
1594 std::vector<float> scales_; member in tensorflow::MklQuantizedConv2DOp