Searched refs:add2_pattern (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_fused_conv_rewriter.cc | 117 auto add2_pattern = Add(&add1, Op(&addends[0]), Op(&addends[1])); in FindConvWithRelu() local 119 AddAnyOrder(&add2, add2_pattern, Op(&addends[2])), add2_pattern, in FindConvWithRelu()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | pattern_matcher_test.cc | 445 auto add2_pattern = Add(Op(&addend0), Op(&addend1)); in TEST_F() local 447 AddAnyOrder(add2_pattern, Op(&addend2)), add2_pattern, Op(&addend0)); in TEST_F()
|