Searched defs:operand_to_generator (Results 1 – 6 of 6) sorted by relevance
109 const HloToElementGeneratorMap& operand_to_generator) { in MakeElementGenerator()
2653 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng() local3028 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local
1370 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in ConvertValueForDistribution()1592 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator) { in MakePhiloxRngElementGenerator()1670 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalSelect()1684 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalClamp()1707 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalConcatenate()1814 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalDynamicSlice()1861 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalGather()1966 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalDynamicUpdateSlice()2052 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalPad()2113 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalDot()[all …]
330 const HloToElementGeneratorMap& operand_to_generator) { in MakeElementGenerator()
69 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local
1000 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng() local