Searched refs:operand_to_generator (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | elemental_ir_emitter.h | 50 const HloToElementGeneratorMap& operand_to_generator); 205 const HloToElementGeneratorMap& operand_to_generator, 210 const HloToElementGeneratorMap& operand_to_generator, 215 const HloToElementGeneratorMap& operand_to_generator, 220 const HloToElementGeneratorMap& operand_to_generator, 225 const HloToElementGeneratorMap& operand_to_generator, 230 const HloToElementGeneratorMap& operand_to_generator, 235 const HloToElementGeneratorMap& operand_to_generator, 240 const HloToElementGeneratorMap& operand_to_generator, 265 const HloToElementGeneratorMap& operand_to_generator,
|
D | elemental_ir_emitter.cc | 1717 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalSelect() argument 1720 operand_to_generator.at(hlo->operand(0))(index)); in EmitElementalSelect() 1722 operand_to_generator.at(hlo->operand(1))(index)); in EmitElementalSelect() 1724 operand_to_generator.at(hlo->operand(2))(index)); in EmitElementalSelect() 1731 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalClamp() argument 1734 operand_to_generator.at(hlo->operand(0))(index)); in EmitElementalClamp() 1736 operand_to_generator.at(hlo->operand(1))(index)); in EmitElementalClamp() 1738 operand_to_generator.at(hlo->operand(2))(index)); in EmitElementalClamp() 1754 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalConcatenate() argument 1829 operand_to_generator.at(operand)(operand_index)); in EmitElementalConcatenate() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.cc | 91 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local 93 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in DefaultAction() 101 .MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 3212 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local 3214 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in DefaultAction() 3220 hlo, elemental_emitter.MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
|