Home
last modified time | relevance | path

Searched refs:kBackwardInput (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc160 case CudnnConvKind::kBackwardInput: in RunGpuConvInternalImpl()
287 case CudnnConvKind::kBackwardInput: in GetGpuConvConfig()
471 case CudnnConvKind::kBackwardInput: in GetGpuConvParams()
Dir_emission_utils.h55 kBackwardInput, // filter + output => input enumerator
Dgpu_layout_assignment.cc139 case CudnnConvKind::kBackwardInput: in AddBackendConstraintsToDnnConvCustomCall()
Dcudnn_pad_for_convolutions.cc214 case CudnnConvKind::kBackwardInput: in TryResolvePaddedShapesForTensorCore()
Dir_emission_utils.cc559 return CudnnConvKind::kBackwardInput; in GetCudnnConvKind()
576 case CudnnConvKind::kBackwardInput: in CudnnConvKindToString()
Dstream_executor_util.cc327 case CudnnConvKind::kBackwardInput: in GetDNNConvKindFromCudnnConvKind()
Dgpu_conv_padding_legalization.cc416 case CudnnConvKind::kBackwardInput: in RunOnComputation()
Dgpu_conv_algorithm_picker.cc111 case CudnnConvKind::kBackwardInput: in GetAlgorithms()
Dir_emitter_unnested.cc1198 descriptor.kind = CudnnConvKind::kBackwardInput; in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc810 case xla::gpu::CudnnConvKind::kBackwardInput: { in EmitDnnConvolution()