Searched refs:GpuElementalIrEmitter (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 72 GpuElementalIrEmitter::GpuElementalIrEmitter( in GpuElementalIrEmitter() function in xla::gpu::GpuElementalIrEmitter
79 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitDeviceMathCall( in EmitDeviceMathCall()
121 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLlvmIntrinsicMathCall( in EmitLlvmIntrinsicMathCall()
144 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitMathCall( in EmitMathCall()
162 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitFloatBinaryOp( in EmitFloatBinaryOp()
191 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitPowerOp( in EmitPowerOp()
202 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLog(PrimitiveType prim_type, in EmitLog()
208 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLog1p(PrimitiveType prim_type, in EmitLog1p()
214 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitSin(PrimitiveType prim_type, in EmitSin()
220 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos()
[all …]
|
| D | elemental_ir_emitter.h | 39 class GpuElementalIrEmitter : public ElementalIrEmitter {
46 GpuElementalIrEmitter(const HloModuleConfig& hlo_module_config,
|
| D | ir_emitter.h | 164 GpuElementalIrEmitter::NestedComputer GetNestedComputer() { in GetNestedComputer()
|
| D | ir_emitter.cc | 99 *hlo, GpuElementalIrEmitter(hlo_module_config_, module_, &b_, in DefaultAction()
580 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion()
|
| D | ir_emitter_unnested.cc | 1794 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusionFromMlir()
1898 GpuElementalIrEmitter operand_elemental_emitter( in HandleFusion()
1942 GpuElementalIrEmitter scatter_elemental_emitter( in HandleFusion()
2031 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion()
3727 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildFusedInitializerThunkForMlir()
4153 GpuElementalIrEmitter elem_emitter(hlo_module_config_, module_, &b_, in EmitTileElementForFusion()
5615 absl::optional<GpuElementalIrEmitter> elemental_emitter; in EmitReductionFromOrToContiguousDimensions()
5738 GpuElementalIrEmitter elem_emitter(hlo_module_config_, module_, &b_, in EmitElementForInputFusibleSlices()
|