Searched refs:kCudnnConvBiasActivationForwardCallTarget (Results 1 – 5 of 5) sorted by relevance
50 HasSubstr(kCudnnConvBiasActivationForwardCallTarget)); in TestMatchWithAllTypes()63 Not(HasSubstr(kCudnnConvBiasActivationForwardCallTarget))); in TestNotMatchWithAllTypes()
131 const char* const kCudnnConvBiasActivationForwardCallTarget = variable142 target == kCudnnConvBiasActivationForwardCallTarget; in IsCustomCallToDnnConvolution()259 if (target == kCudnnConvBiasActivationForwardCallTarget) { in GetCudnnConvKind()
124 extern const char* const kCudnnConvBiasActivationForwardCallTarget;
225 conv->shape(), args, kCudnnConvBiasActivationForwardCallTarget)); in TryRewriteToCudnnForwardRelu()
35 conv.custom_call_target() == kCudnnConvBiasActivationForwardCallTarget); in IsForwardConvolutionCanonical()