Searched refs:BuildInitializerThunkForMlir (Results 1 – 2 of 2) sorted by relevance
679 StatusOr<std::unique_ptr<Thunk>> BuildInitializerThunkForMlir(
2209 BuildInitializerThunkForMlir( in EmitSelectAndScatterFromMlir()3649 IrEmitterUnnested::BuildInitializerThunkForMlir(mlir::Operation* op, in BuildInitializerThunkForMlir() function in xla::gpu::IrEmitterUnnested5562 BuildInitializerThunkForMlir(reduce, reduce.init_values()[0], in EmitReductionFromOrToContiguousDimensions()