Searched refs:select_and_scatter_op (Results 1 – 1 of 1) sorted by relevance
2201 auto select_and_scatter_op = in EmitSelectAndScatterFromMlir() local2204 std::string name = mlir::GetNameFromLoc(select_and_scatter_op.getLoc()); in EmitSelectAndScatterFromMlir()2210 mlir_input.op, select_and_scatter_op.init_value(), in EmitSelectAndScatterFromMlir()2211 select_and_scatter_op.out())); in EmitSelectAndScatterFromMlir()2219 select_and_scatter_op, in EmitSelectAndScatterFromMlir()2220 {select_and_scatter_op.operand(), select_and_scatter_op.source(), in EmitSelectAndScatterFromMlir()2221 select_and_scatter_op.out()}, in EmitSelectAndScatterFromMlir()2233 TypeToShape(select_and_scatter_op.source().getType()); in EmitSelectAndScatterFromMlir()2235 TypeToShape(select_and_scatter_op.operand().getType()); in EmitSelectAndScatterFromMlir()2241 select_and_scatter_op, launch_dimensions.launch_bound(), &b_); in EmitSelectAndScatterFromMlir()[all …]