Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc2676 auto axis_val = axis.getInt(); in convertReduceOpCommon() local
2698 for (auto axis_val : axes) { in convertReduceOpCommon() local
2902 int64_t axis_val = axes_elems.getValues<IntegerAttr>()[i].getInt(); in convertReduceMeanOp() local
H A Dlegalize_tf.cc2058 int64_t axis_val = axis_elems.getValues<IntegerAttr>()[i].getInt(); in matchAndRewrite() local
H A Dlegalize_tfl.cc2960 int64_t axis_val = axis_elems.getValues<APInt>()[i].getSExtValue(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc165 int64_t axis_val = axis.getInt(); in InferOneHotOpType() local