Searched refs:kCudnnConvBackwardInputCallTarget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 127 const char* const kCudnnConvBackwardInputCallTarget = variable
140 target == kCudnnConvBackwardInputCallTarget || in IsCustomCallToDnnConvolution()
253 if (target == kCudnnConvBackwardInputCallTarget) { in GetCudnnConvKind()
|
| D | cudnn_conv_rewriter_test.cc | 308 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
355 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
443 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
547 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
629 op::GetTupleElement(op::CustomCall(kCudnnConvBackwardInputCallTarget, _, in TEST_F()
|
| D | cudnn_conv_pad_for_tensor_cores_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 | 122 extern const char* const kCudnnConvBackwardInputCallTarget;
|
| D | gpu_layout_assignment.cc | 84 instr->custom_call_target() == kCudnnConvBackwardInputCallTarget && in HeuristicLayoutAssignment()
|
| D | cudnn_conv_rewriter.cc | 517 return CreateCudnnConv(kCudnnConvBackwardInputCallTarget, conv->shape(), in RunOnInstruction()
|