Searched refs:kCudnnConvBackwardFilterCallTarget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 130 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
197 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
227 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
256 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
675 op::CustomCall(kCudnnConvBackwardFilterCallTarget, _, _), 0)); in TEST_F()
|
| D | cudnn_pad_for_convolutions_test.cc | 141 kCudnnConvBackwardFilterCallTarget, in TEST_F()
163 kCudnnConvBackwardFilterCallTarget, in TEST_F()
|
| D | ir_emission_utils.h | 135 extern const char* const kCudnnConvBackwardFilterCallTarget;
|
| D | ir_emission_utils.cc | 185 const char* const kCudnnConvBackwardFilterCallTarget = variable
197 target == kCudnnConvBackwardFilterCallTarget || in IsCustomCallToDnnConvolution()
561 if (target == kCudnnConvBackwardFilterCallTarget) { in GetCudnnConvKind()
|
| D | gpu_conv_padding_legalization.cc | 205 kCudnnConvBackwardFilterCallTarget); in CanonicalizeBackwardFilterConvolution()
|
| D | gpu_conv_rewriter.cc | 632 return CreateGpuConv(kCudnnConvBackwardFilterCallTarget, conv->shape(), lhs, in CreateCustomCallHelper()
|