Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h54 StatusOr<CudnnConvKind> GetCudnnConvKind(const HloCustomCallInstruction* instr);
Dcudnn_conv_pad_for_tensor_cores.cc122 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(conv)); in PadForTensorCores()
Dir_emission_utils.cc247 StatusOr<CudnnConvKind> GetCudnnConvKind( in GetCudnnConvKind() function
Dgpu_layout_assignment.cc109 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(instr)); in AddBackendConstraintsToDnnConvCustomCall()
Dcudnn_conv_algorithm_picker.cc233 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(instr)); in PickBestAlgorithm()
Dcudnn_conv_padding_legalization.cc402 TF_ASSIGN_OR_RETURN(auto kind, GetCudnnConvKind(instruction)); in RunOnComputation()
Dcudnn_conv_runner.cc328 TF_ASSIGN_OR_RETURN(CudnnConvKind kind, GetCudnnConvKind(conv)); in GetCudnnConvParams()