Searched refs:kCudnnConvBackwardFilterCallTarget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_cudnn.cc | 37 const char* const kCudnnConvBackwardFilterCallTarget = variable
49 target == kCudnnConvBackwardFilterCallTarget || in IsCustomCallToDnnConvolution()
62 if (target == kCudnnConvBackwardFilterCallTarget) { in GetCudnnConvKind()
|
| D | cublas_cudnn.h | 99 extern const char* const kCudnnConvBackwardFilterCallTarget;
|
| D | gpu_conv_rewriter_test.cc | 131 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
198 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
228 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
257 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
660 op::CustomCall(kCudnnConvBackwardFilterCallTarget, _, _), 0)); in TEST_F()
|
| D | cudnn_pad_for_convolutions_test.cc | 142 kCudnnConvBackwardFilterCallTarget, in TEST_F()
164 kCudnnConvBackwardFilterCallTarget, m::Parameter(0), in TEST_F()
|
| D | gpu_conv_rewriter.cc | 592 } else if (call_target == kCudnnConvBackwardFilterCallTarget) { in CreateGpuConv()
682 return CreateGpuConv(kCudnnConvBackwardFilterCallTarget, conv->shape(), lhs, in CreateCustomCallHelper()
|
| D | gpu_conv_padding_legalization.cc | 208 kCudnnConvBackwardFilterCallTarget); in CanonicalizeBackwardFilterConvolution()
|