Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk_emitter.cc49 context_->GetThunkInfo(inst), std::move(config), in BuildGemmThunk()
54 return absl::make_unique<SequentialThunk>(context_->GetThunkInfo(inst), in BuildGemmThunk()
60 context_->GetThunkInfo(inst), std::move(config), in BuildGemmThunk()
67 Thunk::ThunkInfo ThunkEmitter::EmissionContext::GetThunkInfo( in GetThunkInfo() function in xla::gpu::ThunkEmitter::EmissionContext
Dthunk_emitter.h39 virtual Thunk::ThunkInfo GetThunkInfo(const HloInstruction* hlo) const;
Dir_emitter_unnested.h736 Thunk::ThunkInfo GetThunkInfo(const HloInstruction* hlo) const override;
Dir_emitter_unnested.cc1746 input.thunk_info = GetThunkInfo(hlo); in GetMlirEmitterInput()
3468 implements_whole_instruction ? GetThunkInfo(inst) : Thunk::ThunkInfo(), in BuildKernelThunk()
3862 GetThunkInfo(hlo), in BuildWhileThunk()
3889 GetThunkInfo(hlo), loop_limit, ir_emitter_body->ConsumeThunkSequence(), in BuildForThunk()
3930 GetThunkInfo(hlo), std::move(config), in BuildConditionalThunk()
5829 Thunk::ThunkInfo IrEmitterUnnested::GetThunkInfo( in GetThunkInfo() function in xla::gpu::IrEmitterUnnested
5831 auto info = ThunkEmitter::EmissionContext::GetThunkInfo(hlo); in GetThunkInfo()