Searched refs:CudnnConvKind (Results 1 – 12 of 12) sorted by relevance
157 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 …]
194 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()
53 enum class CudnnConvKind { enum61 StatusOr<CudnnConvKind> GetCudnnConvKind(const HloCustomCallInstruction* instr);64 string CudnnConvKindToString(CudnnConvKind kind);
133 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()
552 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()
66 CudnnConvKind kind;126 CudnnConvKind kind;
323 CudnnConvKind kind) { in GetDNNConvKindFromCudnnConvKind()325 case CudnnConvKind::kBackwardFilter: in GetDNNConvKindFromCudnnConvKind()327 case CudnnConvKind::kBackwardInput: in GetDNNConvKindFromCudnnConvKind()329 case CudnnConvKind::kForward: in GetDNNConvKindFromCudnnConvKind()
413 case CudnnConvKind::kForward: in RunOnComputation()414 case CudnnConvKind::kForwardActivation: in RunOnComputation()416 case CudnnConvKind::kBackwardInput: in RunOnComputation()418 case CudnnConvKind::kBackwardFilter: in RunOnComputation()
102 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()
91 CudnnConvKind kind);
1195 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()
730 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()