Home
last modified time | relevance | path

Searched refs:GpuConvConfig (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h46 struct GpuConvConfig { struct
83 GpuConvConfig config; argument
110 Status RunGpuConv(const GpuConvConfig& conv_config,
116 Status RunGpuConv(const GpuConvConfig& conv_config,
138 StatusOr<GpuConvConfig> GetGpuConvConfig(
144 StatusOr<GpuConvConfig> GetGpuConvConfig(const GpuConvDescriptor& desc,
149 const GpuConvConfig& conv_config,
Dconvolution_thunk.h45 ConvolutionThunk(ThunkInfo thunk_info, GpuConvConfig config,
61 const GpuConvConfig config_;
Dgpu_conv_runner.cc258 StatusOr<GpuConvConfig> GetGpuConvConfig( in GetGpuConvConfig()
260 GpuConvConfig config; in GetGpuConvConfig()
303 GpuConvConfig::FusionConfig& fusion = *config.fusion; in GetGpuConvConfig()
440 StatusOr<GpuConvConfig> GetGpuConvConfig( in GetGpuConvConfig()
458 const GpuConvConfig& config, in GetGpuConvParams()
495 Status RunGpuConv(const gpu::GpuConvConfig& config, in RunGpuConv()
505 Status RunGpuConv(const gpu::GpuConvConfig& config, in RunGpuConv()
Dconvolution_thunk.cc34 ThunkInfo thunk_info, GpuConvConfig config, in ConvolutionThunk()
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()