Searched refs:axis_i32 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 216 IntegerAttr axis_i32; in matchAndRewrite() local
217 if (failed(ConvertToI32Attr(axis_int, &axis_i32))) return failure(); in matchAndRewrite()
222 op, output_type, values, axis_i32, fused_activation_function); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1503 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite() local
1506 convertPackOp(rewriter, op, tf_pack_op.getResult(), inputs, axis_i32); in matchAndRewrite()
1525 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite() local
1528 convertUnpackOp(rewriter, op, tf_unpack_op.value(), axis_i32); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2116 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite() local
2119 convertPackOp(rewriter, op, tfl_pack_op.getResult(), inputs, axis_i32); in matchAndRewrite()
2138 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite() local
2141 convertUnpackOp(rewriter, op, tfl_unpack_op.input(), axis_i32); in matchAndRewrite()
|