Home
last modified time | relevance | path

Searched refs:ConstantScalar (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc225 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(0), in TEST()
226 match::ConstantScalar(1)))); in TEST()
228 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(1), in TEST()
229 match::ConstantScalar(0)))); in TEST()
231 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(0), in TEST()
232 match::ConstantScalar(2)))); in TEST()
235 TEST(PatternMatcherTest, ConstantScalar) { in TEST() argument
237 using match::ConstantScalar; in TEST()
259 EXPECT_TRUE(Match(a, ConstantScalar())); in TEST()
260 EXPECT_TRUE(Match(a, ConstantScalar(1))); in TEST()
[all …]
Dalgebraic_simplifier_test.cc115 m::ConstantScalar(0.125)))); in TEST_F()
137 m::Broadcast(m::ConstantScalar(0.125))))); in TEST_F()
195 m::Broadcast(m::ConstantScalar(0.125))))); in TEST_F()
212 m::Parameter(0), m::Broadcast(m::ConstantScalar(3))))); in TEST_F()
228 m::Lt(m::Parameter(0), m::Broadcast(m::ConstantScalar(0))); in TEST_F()
232 m::ShiftRightLogical(match_abs, m::Broadcast(m::ConstantScalar(3))); in TEST_F()
252 m::Broadcast(m::ConstantScalar(7))))); in TEST_F()
268 m::Lt(m::Parameter(0), m::Broadcast(m::ConstantScalar(0))); in TEST_F()
272 m::AndAnyOrder(match_abs, m::Broadcast(m::ConstantScalar(7))); in TEST_F()
4087 EXPECT_THAT(root, GmockMatch(m::Reshape(m::ConstantScalar(-7.0)))); in TEST_F()
[all …]
Dpattern_matcher.h2403 inline auto ConstantScalar() -> decltype(Op().IsConstantScalar()) {
2408 inline auto ConstantScalar(HloInstructionType** matched_inst)
2414 inline auto ConstantScalar(ScalarTy val)
2420 inline auto ConstantScalar(HloInstructionType** matched_inst, ScalarTy val)
Dwhile_loop_simplifier.cc795 m::ConstantScalar(&constant)) in TryMergeInductionVariables()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc62 auto zero_pattern = Broadcast(match::ConstantScalar(0)); in FindConvWithRelu()