Home
last modified time | relevance | path

Searched refs:flatten_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc297 int64_t flatten_index = 0; in ExtractMinMaxFromAttr() local
299 ++it, ++flatten_index) { in ExtractMinMaxFromAttr()
301 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc684 int64_t flatten_index = 0; in matchAndRewrite() local
688 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite()