Searched refs:kForwardActivation (Results 1 – 9 of 9) sorted by relevance
184 case CudnnConvKind::kForwardActivation: { in RunGpuConvInternalImpl()208 case CudnnConvKind::kForwardActivation: in RunGpuConvInternalImpl()282 case CudnnConvKind::kForwardActivation: in GetGpuConvConfig()301 if (config.kind == CudnnConvKind::kForwardActivation) { in GetGpuConvConfig()466 case CudnnConvKind::kForwardActivation: in GetGpuConvParams()483 if (config.kind == CudnnConvKind::kForwardActivation) { in GetGpuConvParams()
194 if (kind == CudnnConvKind::kForwardActivation) { in TryResolvePaddedShapesForTensorCore()211 case CudnnConvKind::kForwardActivation: in TryResolvePaddedShapesForTensorCore()295 kind != CudnnConvKind::kForwardActivation) { in TryResolvePaddedShapesForIntegerConvolution()329 case CudnnConvKind::kForwardActivation: in TryResolvePaddedShapesForIntegerConvolution()
134 case CudnnConvKind::kForwardActivation: in AddBackendConstraintsToDnnConvCustomCall()183 if (kind != CudnnConvKind::kForwardActivation) { in AddBackendConstraintsToDnnConvCustomCall()
57 kForwardActivation, // activation(conv(input, filter) + broadcast(bias) + enumerator
565 return CudnnConvKind::kForwardActivation; in GetCudnnConvKind()578 case CudnnConvKind::kForwardActivation: in CudnnConvKindToString()
414 case CudnnConvKind::kForwardActivation: in RunOnComputation()
115 case CudnnConvKind::kForwardActivation: in GetAlgorithms()
1204 descriptor.kind = CudnnConvKind::kForwardActivation; in EmitConvolutionThunkFromMlir()1208 descriptor.kind = CudnnConvKind::kForwardActivation; in EmitConvolutionThunkFromMlir()
822 case xla::gpu::CudnnConvKind::kForwardActivation: { in EmitDnnConvolution()