Home
last modified time | relevance | path

Searched refs:kCudnnConvBiasActivationForwardCallTarget (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter_test.cc54 HasSubstr(kCudnnConvBiasActivationForwardCallTarget)); in TestMatchWithAllTypes()
80 Not(HasSubstr(kCudnnConvBiasActivationForwardCallTarget))); in TestNotMatchWithAllTypes()
Dir_emission_utils.h136 extern const char* const kCudnnConvBiasActivationForwardCallTarget;
Dcudnn_fused_conv_rewriter.cc239 conv->shape(), args, kCudnnConvBiasActivationForwardCallTarget)); in TryRewriteToCudnnForwardRelu()
414 kCudnnConvBiasActivationForwardCallTarget) && in RunFuseClamp()
Dir_emission_utils.cc187 const char* const kCudnnConvBiasActivationForwardCallTarget = variable
198 target == kCudnnConvBiasActivationForwardCallTarget; in IsCustomCallToDnnConvolution()
564 if (target == kCudnnConvBiasActivationForwardCallTarget) { in GetCudnnConvKind()
Dgpu_conv_padding_legalization.cc35 conv.custom_call_target() == kCudnnConvBiasActivationForwardCallTarget); in IsForwardConvolutionCanonical()