Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc247 StatusOr<CudnnConvKind> GetCudnnConvKind( in GetCudnnConvKind()
251 return CudnnConvKind::kForward; in GetCudnnConvKind()
254 return CudnnConvKind::kBackwardInput; in GetCudnnConvKind()
257 return CudnnConvKind::kBackwardFilter; in GetCudnnConvKind()
260 return CudnnConvKind::kForwardActivation; in GetCudnnConvKind()
265 string CudnnConvKindToString(CudnnConvKind kind) { in CudnnConvKindToString()
267 case CudnnConvKind::kForward: in CudnnConvKindToString()
269 case CudnnConvKind::kBackwardFilter: in CudnnConvKindToString()
271 case CudnnConvKind::kBackwardInput: in CudnnConvKindToString()
273 case CudnnConvKind::kForwardActivation: in CudnnConvKindToString()
Dcudnn_conv_runner.cc62 CudnnConvKind kind;
117 CudnnConvKind kind = params.kind; in RunCudnnConvImpl()
239 case CudnnConvKind::kForward: in RunCudnnConvImpl()
250 case CudnnConvKind::kBackwardInput: in RunCudnnConvImpl()
261 case CudnnConvKind::kBackwardFilter: in RunCudnnConvImpl()
272 case CudnnConvKind::kForwardActivation: { in RunCudnnConvImpl()
328 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(conv)); in GetCudnnConvParams()
342 case CudnnConvKind::kForward: in GetCudnnConvParams()
350 case CudnnConvKind::kBackwardInput: in GetCudnnConvParams()
358 case CudnnConvKind::kBackwardFilter: in GetCudnnConvParams()
[all …]
Dir_emission_utils.h46 enum class CudnnConvKind { enum
54 StatusOr<CudnnConvKind> GetCudnnConvKind(const HloCustomCallInstruction* instr);
57 string CudnnConvKindToString(CudnnConvKind kind);
Dcudnn_conv_pad_for_tensor_cores.cc135 if (kind == CudnnConvKind::kForwardActivation) { in PadForTensorCores()
150 case CudnnConvKind::kForward: in PadForTensorCores()
151 case CudnnConvKind::kForwardActivation: in PadForTensorCores()
154 case CudnnConvKind::kBackwardInput: in PadForTensorCores()
157 case CudnnConvKind::kBackwardFilter: in PadForTensorCores()
Dgpu_layout_assignment.cc111 case CudnnConvKind::kForward: in AddBackendConstraintsToDnnConvCustomCall()
112 case CudnnConvKind::kForwardActivation: in AddBackendConstraintsToDnnConvCustomCall()
117 case CudnnConvKind::kBackwardInput: in AddBackendConstraintsToDnnConvCustomCall()
122 case CudnnConvKind::kBackwardFilter: in AddBackendConstraintsToDnnConvCustomCall()
161 if (kind != CudnnConvKind::kForwardActivation) { in AddBackendConstraintsToDnnConvCustomCall()
Dcudnn_conv_algorithm_picker.cc44 std::vector<AlgorithmDesc> GetAlgorithms(CudnnConvKind kind, in GetAlgorithms()
49 case CudnnConvKind::kBackwardFilter: in GetAlgorithms()
53 case CudnnConvKind::kBackwardInput: in GetAlgorithms()
56 case CudnnConvKind::kForward: in GetAlgorithms()
57 case CudnnConvKind::kForwardActivation: in GetAlgorithms()
233 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(instr)); in PickBestAlgorithm()
Dcudnn_conv_padding_legalization.cc405 case CudnnConvKind::kForward: in RunOnComputation()
406 case CudnnConvKind::kForwardActivation: in RunOnComputation()
408 case CudnnConvKind::kBackwardInput: in RunOnComputation()
410 case CudnnConvKind::kBackwardFilter: in RunOnComputation()