HomeSort by relevance Sort by last modified time
    Searched defs:IrEmitterUnnested (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.h 36 // within the root computation, but it's emitted using IrEmitterUnnested! Don't
46 class IrEmitterUnnested : public IrEmitter {
48 IrEmitterUnnested(const HloModuleConfig& hlo_module_config,
51 IrEmitterUnnested(const IrEmitterUnnested&) = delete;
52 IrEmitterUnnested& operator=(const IrEmitterUnnested&) = delete;
61 // IrEmitterUnnested handles the following instructions differently from
ir_emitter_unnested.cc 178 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config,
187 Status IrEmitterUnnested::Postprocess(HloInstruction* hlo) {
226 llvm::Function* IrEmitterUnnested::BuildKernelPrototype(
288 Status IrEmitterUnnested::DefaultAction(HloInstruction* hlo) {
293 Status IrEmitterUnnested::HandleDot(HloInstruction* dot) {
307 Status IrEmitterUnnested::HandleConditional(HloInstruction* conditional) {
312 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) {
317 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) {
484 Status IrEmitterUnnested::HandleFft(HloInstruction* fft)
    [all...]

Completed in 67 milliseconds