Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h138 StatusOr<GpuConvConfig> GetGpuConvConfig(
144 StatusOr<GpuConvConfig> GetGpuConvConfig(const GpuConvDescriptor& desc,
Dgpu_conv_runner.cc258 StatusOr<GpuConvConfig> GetGpuConvConfig( in GetGpuConvConfig() function
440 StatusOr<GpuConvConfig> GetGpuConvConfig( in GetGpuConvConfig() function
454 return GetGpuConvConfig(descriptor, cudnn_call->ToString()); in GetGpuConvConfig()
Dgpu_conv_algorithm_picker.cc131 TF_ASSIGN_OR_RETURN(GpuConvConfig config, GetGpuConvConfig(instr)); in GetMIOpenAlgorithms()
449 TF_ASSIGN_OR_RETURN(GpuConvConfig config, GetGpuConvConfig(instr)); in PickBestAlgorithmNoCacheCuda()
710 TF_ASSIGN_OR_RETURN(GpuConvConfig config, GetGpuConvConfig(instr)); in PickBestAlgorithmNoCacheRocm()
Dir_emitter_unnested.cc1216 TF_ASSIGN_OR_RETURN(GpuConvConfig config, GetGpuConvConfig(descriptor, "")); in EmitConvolutionThunkFromMlir()