Searched refs:IntConstWithMatchingType (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | slice_spmd_expander.cc | 435 mlir::Value IntConstWithMatchingType(mlir::OpBuilder& builder, in IntConstWithMatchingType() function
607 IntConstWithMatchingType(builder, strided_slice_op.getLoc(), end, in ExpandOp()
755 IntConstWithMatchingType(builder, strided_slice_op.getLoc(), end, in ExpandOp()
896 IntConstWithMatchingType(builder, strided_slice_grad_op.getLoc(), end, in ExpandOp()
905 mlir::Value new_shape = IntConstWithMatchingType( in ExpandOp()
|