Searched refs:ElementalIrEmitter (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 208 StatusOr<llvm::Value*> ElementalIrEmitter::EmitUnaryOp( in EmitUnaryOp()
220 StatusOr<llvm::Value*> ElementalIrEmitter::EmitIntegerUnaryOp( in EmitIntegerUnaryOp()
335 StatusOr<llvm::Value*> ElementalIrEmitter::EmitFloatUnaryOp( in EmitFloatUnaryOp()
476 StatusOr<llvm::Value*> ElementalIrEmitter::EmitComplexUnaryOp( in EmitComplexUnaryOp()
684 StatusOr<llvm::Value*> ElementalIrEmitter::EmitBinaryOp( in EmitBinaryOp()
699 StatusOr<llvm::Value*> ElementalIrEmitter::EmitFloatBinaryOp( in EmitFloatBinaryOp()
760 StatusOr<llvm::Value*> ElementalIrEmitter::EmitComplexPower( in EmitComplexPower()
794 StatusOr<llvm::Value*> ElementalIrEmitter::EmitComplexBinaryOp( in EmitComplexBinaryOp()
879 llvm::Value* ElementalIrEmitter::EmitFloatMax(llvm::Value* lhs_value, in EmitFloatMax()
884 llvm::Value* ElementalIrEmitter::EmitFloatMin(llvm::Value* lhs_value, in EmitFloatMin()
[all …]
|
| D | elemental_ir_emitter.h | 32 class ElementalIrEmitter : public IrBuilderMixin<ElementalIrEmitter> {
37 ElementalIrEmitter(const HloModuleConfig& hlo_module_config, in ElementalIrEmitter() function
41 virtual ~ElementalIrEmitter() = default;
231 const ElementalIrEmitter::HloToElementGeneratorMap& operand_to_generator,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.h | 30 class CpuElementalIrEmitter : public ElementalIrEmitter {
34 : ElementalIrEmitter(module_config, module, ir_emitter->b()), in CpuElementalIrEmitter()
|
| D | elemental_ir_emitter.cc | 144 return ElementalIrEmitter::MakeElementGenerator(hlo, in MakeElementGenerator()
|
| D | ir_emitter.cc | 2653 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng()
3028 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.h | 81 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
89 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
|
| D | fused_ir_emitter.h | 63 ElementalIrEmitter* elemental_emitter) in FusedIrEmitter()
123 ElementalIrEmitter* elemental_emitter_;
|
| D | dynamic_update_slice_util.cc | 140 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl()
188 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace()
199 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 73 : ElementalIrEmitter(hlo_module_config, module, b), in GpuElementalIrEmitter()
185 return ElementalIrEmitter::EmitFloatBinaryOp(op, lhs_value, rhs_value); in EmitFloatBinaryOp()
486 return ElementalIrEmitter::MakeElementGenerator(hlo, in MakeElementGenerator()
|
| D | elemental_ir_emitter.h | 38 class GpuElementalIrEmitter : public ElementalIrEmitter {
|
| D | ir_emitter.cc | 69 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction()
|
| D | ir_emitter_unnested.cc | 1000 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in HandleRng()
|