Home
last modified time | relevance | path

Searched defs:axis_op (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc121 auto axis_op = CreateI32SplatConst(builder, {1}, axis, location); in Reverse() local
625 auto axis_op = CreateScalarConstantOp(axis, loc, builder); in CreateEqualSizeSplitVOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2782 auto getAxis = [&](Operation *axis_op) { in matchAndRewrite()
2790 Operation *axis_op = non_control_operands.back().getDefiningOp(); in matchAndRewrite() local