Searched defs:scatter_op (Results 1 – 4 of 4) sorted by relevance
56 def _TestCase(self, shape, indices, scatter_op=state_ops.scatter_add): argument
442 auto scatter_op = func_builder->create<mlir::mhlo::ScatterOp>( in ImportInstructionImpl() local
1092 mhlo::ScatterOp scatter_op, ArrayRef<Value> args, in matchAndRewrite()
2502 auto scatter_op = mlir::cast<mlir::lmhlo::ScatterOp>(mlir_input.op); in EmitScatterFromMlir() local