Home
last modified time | relevance | path

Searched refs:CudnnConvKind (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc157 case CudnnConvKind::kForward: in RunGpuConvInternalImpl()
160 case CudnnConvKind::kBackwardInput: in RunGpuConvInternalImpl()
172 case CudnnConvKind::kBackwardFilter: in RunGpuConvInternalImpl()
184 case CudnnConvKind::kForwardActivation: { in RunGpuConvInternalImpl()
205 case CudnnConvKind::kForward: in RunGpuConvInternalImpl()
208 case CudnnConvKind::kForwardActivation: in RunGpuConvInternalImpl()
281 case CudnnConvKind::kForward: in GetGpuConvConfig()
282 case CudnnConvKind::kForwardActivation: in GetGpuConvConfig()
287 case CudnnConvKind::kBackwardInput: in GetGpuConvConfig()
292 case CudnnConvKind::kBackwardFilter: in GetGpuConvConfig()
[all …]
Dcudnn_pad_for_convolutions.cc194 if (kind == CudnnConvKind::kForwardActivation) { in TryResolvePaddedShapesForTensorCore()
210 case CudnnConvKind::kForward: in TryResolvePaddedShapesForTensorCore()
211 case CudnnConvKind::kForwardActivation: in TryResolvePaddedShapesForTensorCore()
214 case CudnnConvKind::kBackwardInput: in TryResolvePaddedShapesForTensorCore()
217 case CudnnConvKind::kBackwardFilter: in TryResolvePaddedShapesForTensorCore()
294 if (kind != CudnnConvKind::kForward && in TryResolvePaddedShapesForIntegerConvolution()
295 kind != CudnnConvKind::kForwardActivation) { in TryResolvePaddedShapesForIntegerConvolution()
315 case CudnnConvKind::kForward: in TryResolvePaddedShapesForIntegerConvolution()
329 case CudnnConvKind::kForwardActivation: in TryResolvePaddedShapesForIntegerConvolution()
Dir_emission_utils.h53 enum class CudnnConvKind { enum
61 StatusOr<CudnnConvKind> GetCudnnConvKind(const HloCustomCallInstruction* instr);
64 string CudnnConvKindToString(CudnnConvKind kind);
Dgpu_layout_assignment.cc133 case CudnnConvKind::kForward: in AddBackendConstraintsToDnnConvCustomCall()
134 case CudnnConvKind::kForwardActivation: in AddBackendConstraintsToDnnConvCustomCall()
139 case CudnnConvKind::kBackwardInput: in AddBackendConstraintsToDnnConvCustomCall()
144 case CudnnConvKind::kBackwardFilter: in AddBackendConstraintsToDnnConvCustomCall()
183 if (kind != CudnnConvKind::kForwardActivation) { in AddBackendConstraintsToDnnConvCustomCall()
Dir_emission_utils.cc552 StatusOr<CudnnConvKind> GetCudnnConvKind( in GetCudnnConvKind()
556 return CudnnConvKind::kForward; in GetCudnnConvKind()
559 return CudnnConvKind::kBackwardInput; in GetCudnnConvKind()
562 return CudnnConvKind::kBackwardFilter; in GetCudnnConvKind()
565 return CudnnConvKind::kForwardActivation; in GetCudnnConvKind()
570 string CudnnConvKindToString(CudnnConvKind kind) { in CudnnConvKindToString()
572 case CudnnConvKind::kForward: in CudnnConvKindToString()
574 case CudnnConvKind::kBackwardFilter: in CudnnConvKindToString()
576 case CudnnConvKind::kBackwardInput: in CudnnConvKindToString()
578 case CudnnConvKind::kForwardActivation: in CudnnConvKindToString()
Dgpu_conv_runner.h66 CudnnConvKind kind;
126 CudnnConvKind kind;
Dstream_executor_util.cc323 CudnnConvKind kind) { in GetDNNConvKindFromCudnnConvKind()
325 case CudnnConvKind::kBackwardFilter: in GetDNNConvKindFromCudnnConvKind()
327 case CudnnConvKind::kBackwardInput: in GetDNNConvKindFromCudnnConvKind()
329 case CudnnConvKind::kForward: in GetDNNConvKindFromCudnnConvKind()
Dgpu_conv_padding_legalization.cc413 case CudnnConvKind::kForward: in RunOnComputation()
414 case CudnnConvKind::kForwardActivation: in RunOnComputation()
416 case CudnnConvKind::kBackwardInput: in RunOnComputation()
418 case CudnnConvKind::kBackwardFilter: in RunOnComputation()
Dgpu_conv_algorithm_picker.cc102 std::vector<AlgorithmDesc> GetAlgorithms(CudnnConvKind kind, in GetAlgorithms()
107 case CudnnConvKind::kBackwardFilter: in GetAlgorithms()
111 case CudnnConvKind::kBackwardInput: in GetAlgorithms()
114 case CudnnConvKind::kForward: in GetAlgorithms()
115 case CudnnConvKind::kForwardActivation: in GetAlgorithms()
428 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(instr)); in PickBestAlgorithmNoCacheCuda()
Dstream_executor_util.h91 CudnnConvKind kind);
Dir_emitter_unnested.cc1195 descriptor.kind = CudnnConvKind::kForward; in EmitConvolutionThunkFromMlir()
1198 descriptor.kind = CudnnConvKind::kBackwardInput; in EmitConvolutionThunkFromMlir()
1201 descriptor.kind = CudnnConvKind::kBackwardFilter; in EmitConvolutionThunkFromMlir()
1204 descriptor.kind = CudnnConvKind::kForwardActivation; in EmitConvolutionThunkFromMlir()
1208 descriptor.kind = CudnnConvKind::kForwardActivation; in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc730 TF_ASSIGN_OR_RETURN(const xla::gpu::CudnnConvKind kind, in EmitDnnConvolution()
804 case xla::gpu::CudnnConvKind::kForward: { in EmitDnnConvolution()
810 case xla::gpu::CudnnConvKind::kBackwardInput: { in EmitDnnConvolution()
816 case xla::gpu::CudnnConvKind::kBackwardFilter: { in EmitDnnConvolution()
822 case xla::gpu::CudnnConvKind::kForwardActivation: { in EmitDnnConvolution()