Home
last modified time | relevance | path

Searched refs:axis_val (Results 1 – 5 of 5) 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
2338 if (axis_val < 0) axis_val += input_rank; in convertReduceOpCommon()
2339 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon()
2341 shape_vec[axis_val] = 1; in convertReduceOpCommon()
2532 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceMeanOp() local
2533 if (axis_val < 0) axis_val += input_rank; in convertReduceMeanOp()
2534 num_elems_on_reduced_axis *= input_type.getShape()[axis_val]; in convertReduceMeanOp()
Dlegalize_tf.cc1959 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local
1960 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()
1961 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
Dlegalize_tfl.cc2773 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local
2774 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()
2775 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc148 int64_t axis_val = axis.getInt(); in InferOneHotOpType() local
151 if (axis_val < -1) return unranked_ty; in InferOneHotOpType()
157 if (axis_val == -1) axis_val = shape.size(); in InferOneHotOpType()
164 shape.insert(shape.begin() + axis_val, depth_val); in InferOneHotOpType()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md651 size_t axis_val = positive_axis(%axis.as_constant()[i])
652 … %output = tosa.REVERSE(%output) {axis=%axis_val} : (tensor<%tensor.type>) -> tensor<%tensor.type>