Home
last modified time | relevance | path

Searched refs:operand_to_generator (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h50 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,
Delemental_ir_emitter.cc1717 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/
Dir_emitter.cc91 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/
Dir_emitter.cc3212 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()