Home
last modified time | relevance | path

Searched defs:axis_val (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2337 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceOpCommon() local
2532 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceMeanOp() local
Dlegalize_tf.cc1959 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local
Dlegalize_tfl.cc2773 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc148 int64_t axis_val = axis.getInt(); in InferOneHotOpType() local