Home
last modified time | relevance | path

Searched refs:MultiplyAnyOrder (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dall_reduce_simplifier_test.cc88 m::MultiplyAnyOrder(m::Parameter(0), in TEST_F()
91 m::MultiplyAnyOrder(m::Parameter(2), in TEST_F()
122 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
165 m::MultiplyAnyOrder(m::Parameter(0), in TEST_F()
Dlogistic_expander_test.cc70 m::MultiplyAnyOrder(m::Broadcast(m::ConstantScalar(0.5)), in TEST_F()
71 m::Tanh(m::MultiplyAnyOrder( in TEST_F()
Dpattern_matcher_test.cc286 TEST_F(PatternMatcherTest, MultiplyAnyOrder) { in TEST_F() argument
288 using match::MultiplyAnyOrder; in TEST_F()
303 root, MultiplyAnyOrder(&instr, ConstantScalar(42), ConstantScalar(52)))); in TEST_F()
305 root, MultiplyAnyOrder(&instr, ConstantScalar(52), ConstantScalar(42)))); in TEST_F()
310 root, MultiplyAnyOrder(&instr, ConstantScalar(42), ConstantScalar(52)) in TEST_F()
313 Match(root, MultiplyAnyOrder(ConstantScalar(42), ConstantScalar(52)) in TEST_F()
Dalgebraic_simplifier.cc680 if (Match(inst, m::MultiplyAnyOrder( in ScalarMultiplyReduction()
711 if (Match(inst, m::MultiplyAnyOrder( in ScalarMultiplyReduction()
913 Match(rhs, m::MultiplyAnyOrder(m::Op().Is(c), m::Op(&b)))) || in HandleAdd()
915 Match(rhs, m::MultiplyAnyOrder(m::Op().Is(c), m::Op(&b))))) && in HandleAdd()
2731 m::MultiplyAnyOrder( in HandleMultiply()
2751 m::MultiplyAnyOrder( in HandleMultiply()
2752 m::MultiplyAnyOrder(m::NonConstant(&a), m::Constant(&c1)), in HandleMultiply()
2753 m::MultiplyAnyOrder(m::NonConstant(&b), m::Constant(&c2))))) { in HandleMultiply()
2777 m::MultiplyAnyOrder( in HandleMultiply()
2778 m::MultiplyAnyOrder(m::NonConstant(&a), m::Constant(&c1)), in HandleMultiply()
[all …]
Dalgebraic_simplifier_test.cc95 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
115 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
154 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
155 m::MultiplyAnyOrder(m::Parameter(0), m::Parameter(1)), in TEST_F()
156 m::MultiplyAnyOrder(m::ConstantScalar(2), m::ConstantScalar(4))))); in TEST_F()
174 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
175 m::Parameter(0), m::MultiplyAnyOrder(m::ConstantScalar(2), in TEST_F()
197 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
198 m::Parameter(0), m::Broadcast(m::MultiplyAnyOrder( in TEST_F()
219 GmockMatch(m::MultiplyAnyOrder( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc61 using match::MultiplyAnyOrder; in FindConvWithRelu()
90 MultiplyAnyOrder(&mul1, alpha_pattern, conv_pattern), conv_pattern); in FindConvWithRelu()
102 MultiplyAnyOrder(&mul2, alpha_pattern, side_input_pattern), in FindConvWithRelu()
Dgemm_rewriter.cc81 m::MultiplyAnyOrder( in HandleMultiply()