Searched refs:kCudnnConvBackwardInputCallTarget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 311 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
360 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
450 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
556 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
639 op::GetTupleElement(op::CustomCall(kCudnnConvBackwardInputCallTarget, _, in TEST_F()
|
| D | cudnn_pad_for_convolutions_test.cc | 73 EXPECT_THAT(root, op::CustomCall(kCudnnConvBackwardInputCallTarget, in TEST_F()
119 kCudnnConvBackwardInputCallTarget, op::Parameter(0), in TEST_F()
|
| D | ir_emission_utils.h | 134 extern const char* const kCudnnConvBackwardInputCallTarget;
|
| D | ir_emission_utils.cc | 183 const char* const kCudnnConvBackwardInputCallTarget = variable
196 target == kCudnnConvBackwardInputCallTarget || in IsCustomCallToDnnConvolution()
558 if (target == kCudnnConvBackwardInputCallTarget) { in GetCudnnConvKind()
|
| D | gpu_layout_assignment.cc | 106 instr->custom_call_target() == kCudnnConvBackwardInputCallTarget && in HeuristicLayoutAssignment()
|
| D | gpu_conv_rewriter.cc | 625 return CreateGpuConv(kCudnnConvBackwardInputCallTarget, conv->shape(), in CreateCustomCallHelper()
|