Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2530 int64_t num_elems_on_reduced_axis = 1; in convertReduceMeanOp() local
2534 num_elems_on_reduced_axis *= input_type.getShape()[axis_val]; in convertReduceMeanOp()
2536 double div_scale = 1.0 / static_cast<double>(num_elems_on_reduced_axis); in convertReduceMeanOp()