Searched defs:strided_slice_op (Results 1 – 3 of 3) sorted by relevance
582 auto strided_slice_op = mlir::cast<mlir::TF::StridedSliceOp>(op); in ExpandOp() local650 mlir::TF::StridedSliceOp strided_slice_op = in ComputeLayoutForward() local674 mlir::TF::StridedSliceOp strided_slice_op = in ComputeLayoutBackward() local711 mlir::TF::TensorStridedSliceUpdateOp strided_slice_op = in ExpandOp() local785 mlir::TF::TensorStridedSliceUpdateOp strided_slice_op = in ComputeLayoutForward() local819 mlir::TF::TensorStridedSliceUpdateOp strided_slice_op = in ComputeLayoutBackward() local
424 TF::StridedSliceOp strided_slice_op = llvm::cast<TF::StridedSliceOp>(op); in RewriteNewAxisMask() local503 TF::StridedSliceOp strided_slice_op = llvm::cast<TF::StridedSliceOp>(op); in RewriteEllipsisMask() local668 TF::StridedSliceOp strided_slice_op = llvm::cast<TF::StridedSliceOp>(op); in matchAndRewrite() local
3555 auto strided_slice_op = ops::StridedSlice( in TEST_F() local3646 auto strided_slice_op = ops::StridedSlice( in TEST_F() local3707 auto strided_slice_op = ops::StridedSlice( in TEST_F() local