Searched refs:GetGenerator (Results 1 – 4 of 4) sorted by relevance
172 ElementGenerator update_array_generator = fused_emitter.GetGenerator(update); in EmitFusedDynamicUpdateSliceInPlaceImpl()176 fused_emitter.GetGenerator(dynamic_update_slice->operand(2 + index)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
88 IndexedGenerator GetGenerator(const HloInstruction* instruction) const;
201 FusedIrEmitter::IndexedGenerator FusedIrEmitter::GetGenerator( in GetGenerator() function in xla::FusedIrEmitter
599 *fusion, operand_fused_emitter.GetGenerator(root->operand(0)), in HandleFusion()620 scatter_fused_emitter.GetGenerator(root->operand(1)), in HandleFusion()622 scatter_fused_emitter.GetGenerator(root->operand(2)))); in HandleFusion()2007 ParallelLoopEmitter(fused_emitter.GetGenerator(init_value_operand), in BuildInitializerThunk()2683 .GetGenerator(init_value_operand)(IrArray::Index(b_.getInt32Ty())) in EmitPrologueForOneReduction()2902 input_gens.push_back(fused_emitter.GetGenerator(inst->operand(0))); in EmitTileElementForReduction()2904 extra_output_gens.emplace_back(fused_emitter.GetGenerator(inst), in EmitTileElementForReduction()