Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc204 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(0), in TEST_F()
205 match::ConstantScalar(1)))); in TEST_F()
207 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(1), in TEST_F()
208 match::ConstantScalar(0)))); in TEST_F()
210 Match(root, match::AnyOf<HloInstruction>(match::ConstantScalar(0), in TEST_F()
211 match::ConstantScalar(2)))); in TEST_F()
214 TEST_F(PatternMatcherTest, ConstantScalar) { in TEST_F() argument
216 using match::ConstantScalar; in TEST_F()
239 EXPECT_TRUE(Match(a, ConstantScalar())); in TEST_F()
240 EXPECT_TRUE(Match(a, ConstantScalar(1))); in TEST_F()
[all …]
Dlogistic_expander_test.cc70 m::MultiplyAnyOrder(m::Broadcast(m::ConstantScalar(0.5)), in TEST_F()
72 m::Broadcast(m::ConstantScalar(0.5)), in TEST_F()
74 m::Broadcast(m::ConstantScalar(0.5))))); in TEST_F()
97 m::Broadcast(m::ConstantScalar(1.0)), in TEST_F()
98 m::AddAnyOrder(m::Broadcast(m::ConstantScalar(1.0)), in TEST_F()
Dall_reduce_simplifier_test.cc89 m::Broadcast(m::Convert(m::ConstantScalar(8)))), in TEST_F()
92 m::Convert(m::ConstantScalar(8)))))); in TEST_F()
124 m::Broadcast(m::Convert(m::ConstantScalar(8)))))); in TEST_F()
166 m::Broadcast(m::Convert(m::ConstantScalar(4)))), in TEST_F()
Dtopk_rewriter.cc37 m::Lt(param_s32, m::ConstantScalar(0)), in IsNanSafeGt()
39 m::Subtract(m::ConstantScalar(std::numeric_limits<int32>::max()), in IsNanSafeGt()
53 m::Lt(param_s32, m::ConstantScalar(0)), in IsNanSafeGt()
55 m::Subtract(m::ConstantScalar(std::numeric_limits<int32>::max()), in IsNanSafeGt()
Dalgebraic_simplifier_test.cc117 m::ConstantScalar(0.125)))); in TEST_F()
156 m::MultiplyAnyOrder(m::ConstantScalar(2), m::ConstantScalar(4))))); in TEST_F()
175 m::Parameter(0), m::MultiplyAnyOrder(m::ConstantScalar(2), in TEST_F()
176 m::ConstantScalar(4))))); in TEST_F()
221 m::Broadcast(m::ConstantScalar(0.125))))); in TEST_F()
279 m::Broadcast(m::ConstantScalar(0.125))))); in TEST_F()
296 m::Parameter(0), m::Broadcast(m::ConstantScalar(3))))); in TEST_F()
312 m::Lt(m::Parameter(0), m::Broadcast(m::ConstantScalar(0))); in TEST_F()
316 m::ShiftRightLogical(match_abs, m::Broadcast(m::ConstantScalar(3))); in TEST_F()
336 m::Broadcast(m::ConstantScalar(7))))); in TEST_F()
[all …]
Dpattern_matcher.h2267 inline auto ConstantScalar() { return Op().IsConstantScalar(); }
2270 inline auto ConstantScalar(HloInstructionType** matched_inst) {
2275 inline auto ConstantScalar(ScalarTy val) {
2280 inline auto ConstantScalar(HloInstructionType** matched_inst, ScalarTy val) {
Dalgebraic_simplifier.cc682 m::Broadcast(m::ConstantScalar(&multiplier))))) { in ScalarMultiplyReduction()
713 m::Broadcast(m::ConstantScalar(&multiplier))))) { in ScalarMultiplyReduction()
836 m::Broadcast(m::ConstantScalar(&c1))), in HandleAdd()
837 m::Broadcast(m::ConstantScalar(&c2))))) { in HandleAdd()
2840 m::Multiply(m::Op(&a), m::Broadcast(m::ConstantScalar(&c1))), in HandleMultiply()
2841 m::Broadcast(m::ConstantScalar(&c2))))) { in HandleMultiply()
4458 if (!Match(slice_dim_start, m::ConstantScalar())) { in HandleDynamicUpdateSlice()
Dwhile_loop_simplifier.cc1007 m::ConstantScalar(&constant)) in TryMergeInductionVariables()
Dspace_to_batch_converter.cc1084 if (Match(op, m::ConstantScalar())) { in IsBroadcastTree()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc58 using match::ConstantScalar; in FindConvWithRelu()
67 auto zero_pattern = Broadcast(ConstantScalar(0)); in FindConvWithRelu()
86 auto alpha_pattern = Broadcast(ConstantScalar(&alpha_conv_instr)); in FindConvWithRelu()
93 auto alpha_pattern = Broadcast(ConstantScalar(&alpha_side_input_instr)); in FindConvWithRelu()
317 auto lower_pattern = Broadcast(match::ConstantScalar(-128)); in FindConvWithClampAndConvertToInt8()
318 auto upper_pattern = Broadcast(match::ConstantScalar(127)); in FindConvWithClampAndConvertToInt8()
Dgemm_rewriter.cc83 m::Broadcast(m::ConstantScalar(&alpha))))) { in HandleMultiply()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1119 m::Lt(param_s32, m::ConstantScalar(0)), in IsNanSafeGt()
1121 m::Subtract(m::ConstantScalar(std::numeric_limits<int32>::max()), in IsNanSafeGt()
1135 m::Lt(param_s32, m::ConstantScalar(0)), in IsNanSafeGt()
1137 m::Subtract(m::ConstantScalar(std::numeric_limits<int32>::max()), in IsNanSafeGt()