Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h110 Status RunGpuConv(const GpuConvConfig& conv_config,
116 Status RunGpuConv(const GpuConvConfig& conv_config,
Dconvolution_thunk.cc59 TF_RETURN_IF_ERROR(RunGpuConv(config_, absl::MakeSpan(operand_se_buffers), in ExecuteOnStream()
Dgpu_conv_runner.cc495 Status RunGpuConv(const gpu::GpuConvConfig& config, in RunGpuConv() function
501 return RunGpuConv(config, operand_buffers, result_buffer, &scratch_allocator, in RunGpuConv()
505 Status RunGpuConv(const gpu::GpuConvConfig& config, in RunGpuConv() function
Dgpu_conv_algorithm_picker.cc474 RunGpuConv(config, absl::MakeSpan(operand_buffers), result_buffer, in PickBestAlgorithmNoCacheCuda()
728 RunGpuConv(config, absl::MakeSpan(operand_buffers), result_buffer, in PickBestAlgorithmNoCacheRocm()