Searched refs:GetCudnnConvKind (Results 1 – 8 of 8) sorted by relevance
181 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(conv)); in TryResolvePaddedShapesForTensorCore()284 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(conv)); in TryResolvePaddedShapesForIntegerConvolution()
61 StatusOr<CudnnConvKind> GetCudnnConvKind(const HloCustomCallInstruction* instr);
131 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(instr)); in AddBackendConstraintsToDnnConvCustomCall()
410 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(instruction)); in RunOnComputation()
444 TF_ASSIGN_OR_RETURN(descriptor.kind, GetCudnnConvKind(cudnn_call)); in GetGpuConvConfig()
428 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(instr)); in PickBestAlgorithmNoCacheCuda()
552 StatusOr<CudnnConvKind> GetCudnnConvKind( in GetCudnnConvKind() function
731 xla::gpu::GetCudnnConvKind(custom_call)); in EmitDnnConvolution()