Home
last modified time | relevance | path

Searched refs:kCudnnConvBackwardFilterCallTarget (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc129 const char* const kCudnnConvBackwardFilterCallTarget = variable
141 target == kCudnnConvBackwardFilterCallTarget || in IsCustomCallToDnnConvolution()
256 if (target == kCudnnConvBackwardFilterCallTarget) { in GetCudnnConvKind()
Dcudnn_conv_rewriter_test.cc129 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
165 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
195 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
225 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
254 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
Dcudnn_conv_pad_for_tensor_cores_test.cc141 kCudnnConvBackwardFilterCallTarget, in TEST_F()
163 kCudnnConvBackwardFilterCallTarget, in TEST_F()
Dir_emission_utils.h123 extern const char* const kCudnnConvBackwardFilterCallTarget;
Dcudnn_conv_padding_legalization.cc205 kCudnnConvBackwardFilterCallTarget); in CanonicalizeBackwardFilterConvolution()
Dcudnn_conv_rewriter.cc509 return CreateCudnnConv(kCudnnConvBackwardFilterCallTarget, conv->shape(), in RunOnInstruction()