Searched refs:kCudnnConvForwardCallTarget (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions_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()
209 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
234 kCudnnConvForwardCallTarget, op::Parameter(0), in TEST_F()
|
| D | cudnn_fused_conv_rewriter.cc | 153 if (conv->custom_call_target() != kCudnnConvForwardCallTarget) { in FindConvWithRelu()
272 if (call->custom_call_target() == kCudnnConvForwardCallTarget) { in RunFuseBiasSideActivation()
412 if ((call->custom_call_target() == kCudnnConvForwardCallTarget || in RunFuseClamp()
444 .WithCustomCallTarget(kCudnnConvForwardCallTarget), in FindConvWithConvertToFloat()
|
| D | cudnn_fused_conv_rewriter_test.cc | 52 Not(HasSubstr(kCudnnConvForwardCallTarget))); in TestMatchWithAllTypes()
78 EXPECT_THAT(optimized_hlo_string, HasSubstr(kCudnnConvForwardCallTarget)); in TestNotMatchWithAllTypes()
|
| D | gpu_conv_rewriter_test.cc | 167 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
395 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
501 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
612 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
|
| D | ir_emission_utils.h | 133 extern const char* const kCudnnConvForwardCallTarget;
|
| D | ir_emission_utils.cc | 182 const char* const kCudnnConvForwardCallTarget = "__cudnn$convForward"; variable
195 return target == kCudnnConvForwardCallTarget || in IsCustomCallToDnnConvolution()
555 if (target == kCudnnConvForwardCallTarget) { in GetCudnnConvKind()
|
| D | gpu_conv_padding_legalization.cc | 34 CHECK(conv.custom_call_target() == kCudnnConvForwardCallTarget || in IsForwardConvolutionCanonical()
|
| D | gpu_conv_rewriter.cc | 682 return CreateGpuConv(kCudnnConvForwardCallTarget, conv->shape(), in CreateCustomCallHelper()
|