Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc172 case CudnnConvKind::kBackwardFilter: in RunGpuConvInternalImpl()
292 case CudnnConvKind::kBackwardFilter: in GetGpuConvConfig()
476 case CudnnConvKind::kBackwardFilter: in GetGpuConvParams()
Dir_emission_utils.h56 kBackwardFilter, // input + output => filter enumerator
Dgpu_layout_assignment.cc144 case CudnnConvKind::kBackwardFilter: in AddBackendConstraintsToDnnConvCustomCall()
Dcudnn_pad_for_convolutions.cc217 case CudnnConvKind::kBackwardFilter: in TryResolvePaddedShapesForTensorCore()
Dir_emission_utils.cc562 return CudnnConvKind::kBackwardFilter; in GetCudnnConvKind()
574 case CudnnConvKind::kBackwardFilter: in CudnnConvKindToString()
Dstream_executor_util.cc325 case CudnnConvKind::kBackwardFilter: in GetDNNConvKindFromCudnnConvKind()
Dgpu_conv_padding_legalization.cc418 case CudnnConvKind::kBackwardFilter: in RunOnComputation()
Dgpu_conv_algorithm_picker.cc107 case CudnnConvKind::kBackwardFilter: in GetAlgorithms()
Dir_emitter_unnested.cc1201 descriptor.kind = CudnnConvKind::kBackwardFilter; in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc816 case xla::gpu::CudnnConvKind::kBackwardFilter: { in EmitDnnConvolution()