Searched refs:kCudnnConvForwardCallTarget (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores_test.cc | 50 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
97 kCudnnConvForwardCallTarget, op::Parameter(0), in TEST_F()
184 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
|
| D | cudnn_fused_conv_rewriter_test.cc | 48 Not(HasSubstr(kCudnnConvForwardCallTarget))); in TestMatchWithAllTypes()
61 EXPECT_THAT(optimized_hlo_string, HasSubstr(kCudnnConvForwardCallTarget)); in TestNotMatchWithAllTypes()
|
| D | ir_emission_utils.cc | 126 const char* const kCudnnConvForwardCallTarget = "__cudnn$convForward"; variable
139 return target == kCudnnConvForwardCallTarget || in IsCustomCallToDnnConvolution()
250 if (target == kCudnnConvForwardCallTarget) { in GetCudnnConvKind()
|
| D | cudnn_fused_conv_rewriter.cc | 148 if (conv->custom_call_target() != kCudnnConvForwardCallTarget) { in FindConvWithRelu()
257 if (call->custom_call_target() == kCudnnConvForwardCallTarget) { in Run()
|
| D | ir_emission_utils.h | 121 extern const char* const kCudnnConvForwardCallTarget;
|
| D | cudnn_conv_rewriter_test.cc | 389 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
493 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
602 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
|
| D | cudnn_conv_padding_legalization.cc | 34 CHECK(conv.custom_call_target() == kCudnnConvForwardCallTarget || in IsForwardConvolutionCanonical()
|
| D | cudnn_conv_rewriter.cc | 524 return CreateCudnnConv(kCudnnConvForwardCallTarget, conv->shape(), in RunOnInstruction()
|