Searched refs:DynamicUpdateSliceOp (Results 1 – 8 of 8) sorted by relevance
40 class DynamicUpdateSliceOp : public XlaOpKernel { class42 explicit DynamicUpdateSliceOp(OpKernelConstruction* context) in DynamicUpdateSliceOp() function in tensorflow::__anon743243e20111::DynamicUpdateSliceOp74 REGISTER_XLA_OP(Name("XlaDynamicUpdateSlice"), DynamicUpdateSliceOp);
434 } else if (isa<mlir::mhlo::DynamicUpdateSliceOp, in MhloToHloOpcode()435 mlir::lmhlo::DynamicUpdateSliceOp>(op)) { in MhloToHloOpcode()
376 ->create<mlir::mhlo::DynamicUpdateSliceOp>( in ImportInstructionImpl()
795 auto dus = mlir::dyn_cast<mlir::mhlo::DynamicUpdateSliceOp>( in CanEmitFusedDynamicUpdateSliceInPlaceForGpu()
534 return MakeXlaOp(builder_.create<mlir::mhlo::DynamicUpdateSliceOp>( in DynamicUpdateSliceInternal()
353 return builder->create<DynamicUpdateSliceOp>(loc, type, v, update, in DynamicUpdateSliceInMinorDims()5107 indices = builder->create<mhlo::DynamicUpdateSliceOp>( in matchAndRewrite()5110 indices = builder->create<mhlo::DynamicUpdateSliceOp>( in matchAndRewrite()5223 input = rewriter.create<mhlo::DynamicUpdateSliceOp>( in matchAndRewrite()5251 rewriter.replaceOpWithNewOp<mhlo::DynamicUpdateSliceOp>( in matchAndRewrite()
297 return CreateOpWithoutAttrs<lmhlo::DynamicUpdateSliceOp>(instr); in EmitOp()
431 static LogicalResult Verify(DynamicUpdateSliceOp op) { in Verify()