Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.cc109 const HloToElementGeneratorMap& operand_to_generator) { in MakeElementGenerator() argument
112 return [this, hlo, &operand_to_generator]( in MakeElementGenerator()
117 operand_to_generator.at(hlo->operand(i))(index)); in MakeElementGenerator()
124 return [this, hlo, &operand_to_generator](const IrArray::Index& index) { in MakeElementGenerator()
127 operand_to_generator.at(hlo->operand(0)), index); in MakeElementGenerator()
130 return [this, hlo, &operand_to_generator](const IrArray::Index& index) { in MakeElementGenerator()
133 operand_to_generator.at(hlo->operand(0)), in MakeElementGenerator()
134 operand_to_generator.at(hlo->operand(1)), index); in MakeElementGenerator()
137 return [this, hlo, &operand_to_generator](const IrArray::Index& index) { in MakeElementGenerator()
140 operand_to_generator.at(hlo->operand(0)), in MakeElementGenerator()
[all …]
Delemental_ir_emitter.h39 const HloToElementGeneratorMap& operand_to_generator) override;
Dir_emitter.cc2653 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng() local
2655 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in HandleRng()
2662 rng, elemental_emitter.MakeElementGenerator(rng, operand_to_generator))); in HandleRng()
3028 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local
3030 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in DefaultAction()
3036 hlo, elemental_emitter.MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h54 const HloToElementGeneratorMap& operand_to_generator);
167 const HloToElementGeneratorMap& operand_to_generator,
172 const HloToElementGeneratorMap& operand_to_generator,
177 const HloToElementGeneratorMap& operand_to_generator,
182 const HloToElementGeneratorMap& operand_to_generator,
187 const HloToElementGeneratorMap& operand_to_generator,
192 const HloToElementGeneratorMap& operand_to_generator,
197 const HloToElementGeneratorMap& operand_to_generator,
202 const HloToElementGeneratorMap& operand_to_generator,
223 const HloToElementGeneratorMap& operand_to_generator);
[all …]
Delemental_ir_emitter.cc1370 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in ConvertValueForDistribution() argument
1373 operand_to_generator.at(hlo->operand(0))(index)); in ConvertValueForDistribution()
1375 operand_to_generator.at(hlo->operand(1))(index)); in ConvertValueForDistribution()
1592 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator) { in MakePhiloxRngElementGenerator() argument
1663 return ConvertValueForDistribution(hlo, operand_to_generator, index, in MakePhiloxRngElementGenerator()
1670 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalSelect() argument
1673 operand_to_generator.at(hlo->operand(0))(index)); in EmitElementalSelect()
1675 operand_to_generator.at(hlo->operand(1))(index)); in EmitElementalSelect()
1677 operand_to_generator.at(hlo->operand(2))(index)); in EmitElementalSelect()
1684 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator, in EmitElementalClamp() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc330 const HloToElementGeneratorMap& operand_to_generator) { in MakeElementGenerator() argument
333 return [=, &operand_to_generator]( in MakeElementGenerator()
341 operand_to_generator.at(operand)(index)); in MakeElementGenerator()
356 return [=, &operand_to_generator]( in MakeElementGenerator()
367 operand_to_generator.at(hlo->operand(1))( in MakeElementGenerator()
431 operand_to_generator.at(operand)(input_index)); in MakeElementGenerator()
443 return [=, &operand_to_generator]( in MakeElementGenerator()
451 operand_to_generator.at(hlo->operand(1))( in MakeElementGenerator()
476 operand_to_generator.at(hlo->operand(0))(input_index)); in MakeElementGenerator()
487 operand_to_generator); in MakeElementGenerator()
Delemental_ir_emitter.h51 const HloToElementGeneratorMap& operand_to_generator) override;
Dir_emitter.cc69 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction() local
71 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in DefaultAction()
78 .MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
Dir_emitter_unnested.cc1000 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng() local
1002 operand_to_generator[operand] = [=](const llvm_ir::IrArray::Index& index) { in HandleRng()
1010 .MakeElementGenerator(rng, operand_to_generator), in HandleRng()