Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc76 Status RunGpuConvForward(GpuConvParams params, in RunGpuConvForward()
96 Status RunGpuConvForwardActivation(GpuConvParams params, in RunGpuConvForwardActivation()
149 Status RunGpuConvInternalImpl(GpuConvParams params, in RunGpuConvInternalImpl()
197 Status RunGpuConvInternalImpl(GpuConvParams params, in RunGpuConvInternalImpl()
221 Status RunGpuConvImpl(const GpuConvParams& params, in RunGpuConvImpl()
457 StatusOr<GpuConvParams> GetGpuConvParams( in GetGpuConvParams()
461 GpuConvParams params; in GetGpuConvParams()
485 GpuConvParams::FusionParams& fusion = *params.fusion; in GetGpuConvParams()
510 TF_ASSIGN_OR_RETURN(GpuConvParams params, in RunGpuConv()
Dgpu_conv_runner.h82 struct GpuConvParams { struct
148 StatusOr<GpuConvParams> GetGpuConvParams(
Dbackend_configs.proto34 // GpuConvParams for their meanings.
Dgpu_conv_algorithm_picker.cc139 TF_ASSIGN_OR_RETURN(GpuConvParams params, in GetMIOpenAlgorithms()