Home
last modified time | relevance | path

Searched refs:AddInstruction (Results 1 – 25 of 169) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc103 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
105 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
107 builder.AddInstruction(HloInstruction::CreateParameter(2, shape, "c")); in TEST_F()
108 HloInstruction* add0 = builder.AddInstruction( in TEST_F()
110 HloInstruction* add1 = builder.AddInstruction( in TEST_F()
112 HloInstruction* pred = builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
114 HloInstruction* sel = builder.AddInstruction( in TEST_F()
117 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
119 HloInstruction* dot = builder.AddInstruction( in TEST_F()
121 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
Dcopy_insertion_test.cc90 HloInstruction* x = builder.AddInstruction( in TEST_F()
93 builder.AddInstruction(HloInstruction::CreateTuple({x})); in TEST_F()
110 HloInstruction* constant = builder.AddInstruction( in TEST_F()
113 builder.AddInstruction(HloInstruction::CreateTuple({constant})); in TEST_F()
134 builder.AddInstruction(HloInstruction::CreateConstant( in TEST_F()
141 HloInstruction* copy_1 = builder.AddInstruction( in TEST_F()
143 HloInstruction* copy_2 = builder.AddInstruction( in TEST_F()
145 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
147 builder.AddInstruction( in TEST_F()
169 HloInstruction* constant1 = builder.AddInstruction( in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc120 auto constant1 = builder.AddInstruction( in TEST_F()
122 auto constant2 = builder.AddInstruction( in TEST_F()
124 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
149 auto param0 = builder.AddInstruction( in TEST_F()
151 auto param1 = builder.AddInstruction( in TEST_F()
154 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_F()
155 auto gte0 = builder.AddInstruction( in TEST_F()
157 auto gte1 = builder.AddInstruction( in TEST_F()
159 builder.AddInstruction( in TEST_F()
198 auto param0 = builder.AddInstruction( in TEST_F()
[all …]
Dtuple_simplifier_test.cc61 HloInstruction* param0 = builder.AddInstruction( in TEST_F()
63 HloInstruction* param1 = builder.AddInstruction( in TEST_F()
65 HloInstruction* param2 = builder.AddInstruction( in TEST_F()
67 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1, param2})); in TEST_F()
77 HloInstruction* param = builder.AddInstruction( in TEST_F()
79 builder.AddInstruction( in TEST_F()
90 HloInstruction* param0 = builder.AddInstruction( in TEST_F()
92 HloInstruction* param1 = builder.AddInstruction( in TEST_F()
94 HloInstruction* param2 = builder.AddInstruction( in TEST_F()
96 HloInstruction* tuple = builder.AddInstruction( in TEST_F()
[all …]
Dhlo_dce_test.cc53 auto constant1 = builder.AddInstruction( in TEST_F()
55 auto constant2 = builder.AddInstruction( in TEST_F()
57 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
74 auto constant = builder.AddInstruction( in TEST_F()
76 auto token = builder.AddInstruction(HloInstruction::CreateToken()); in TEST_F()
77 builder.AddInstruction( in TEST_F()
79 builder.AddInstruction(HloInstruction::CreateTuple({})); in TEST_F()
96 auto live_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
98 auto dead_param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
100 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc85 HloInstruction* param0 = builder.AddInstruction( in CreateR0F32UnaryOpComputation()
87 builder.AddInstruction( in CreateR0F32UnaryOpComputation()
104 auto constant1 = builder.AddInstruction( in TEST_P()
106 auto constant2 = builder.AddInstruction( in TEST_P()
108 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
147 auto param0 = builder.AddInstruction( in TEST_P()
149 auto param1 = builder.AddInstruction( in TEST_P()
152 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_P()
153 auto gte0 = builder.AddInstruction( in TEST_P()
155 auto gte1 = builder.AddInstruction( in TEST_P()
[all …]
Dwhile_loop_invariant_code_motion_test.cc53 builder.AddInstruction( in MakeAlwaysTrueComputation()
55 builder.AddInstruction( in MakeAlwaysTrueComputation()
68 HloInstruction* param = builder.AddInstruction( in TEST_F()
70 HloInstruction* gte_0 = builder.AddInstruction( in TEST_F()
72 HloInstruction* gte_1 = builder.AddInstruction( in TEST_F()
75 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
77 builder.AddInstruction( in TEST_F()
84 auto* init_value = builder.AddInstruction( in TEST_F()
86 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
110 HloInstruction* param = builder.AddInstruction( in TEST_F()
[all …]
Dbuffer_assignment_test.cc158 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildMapComputationPlus1()
159 auto value = builder.AddInstruction( in BuildMapComputationPlus1()
161 builder.AddInstruction( in BuildMapComputationPlus1()
169 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildReduceComputation()
171 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in BuildReduceComputation()
172 builder.AddInstruction( in BuildReduceComputation()
187 auto const4 = builder.AddInstruction( in BuildWhileConditionComputation()
189 auto param = builder.AddInstruction( in BuildWhileConditionComputation()
191 auto index = builder.AddInstruction( in BuildWhileConditionComputation()
193 builder.AddInstruction( in BuildWhileConditionComputation()
[all …]
Dhlo_rematerialization_test.cc61 auto param = builder.AddInstruction( in MakeRematerializableComputation()
63 auto reshape = builder.AddInstruction( in MakeRematerializableComputation()
65 auto bcast = builder.AddInstruction( in MakeRematerializableComputation()
67 auto negate = builder.AddInstruction( in MakeRematerializableComputation()
69 auto concat_1 = builder.AddInstruction(HloInstruction::CreateConcatenate( in MakeRematerializableComputation()
72 auto slice_1 = builder.AddInstruction(HloInstruction::CreateSlice( in MakeRematerializableComputation()
76 auto concat_2 = builder.AddInstruction(HloInstruction::CreateConcatenate( in MakeRematerializableComputation()
81 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat_2, in MakeRematerializableComputation()
106 auto param = builder.AddInstruction( in MakeRematerializableWhileComputation()
108 auto reshape = builder.AddInstruction( in MakeRematerializableWhileComputation()
[all …]
Dreshape_mover_test.cc43 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
45 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
48 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
50 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); in TEST_F()
51 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
81 auto rng0 = builder.AddInstruction(HloInstruction::CreateRng(
84 {builder.AddInstruction(
86 builder.AddInstruction(HloInstruction::CreateConstant(
89 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));
91 auto const1 = builder.AddInstruction(
[all …]
Dtuple_points_to_analysis_test.cc125 auto constant1 = builder.AddInstruction( in TEST_F()
127 auto constant2 = builder.AddInstruction( in TEST_F()
129 auto tuple = builder.AddInstruction( in TEST_F()
178 auto constant1 = builder.AddInstruction( in TEST_F()
180 auto constant2 = builder.AddInstruction( in TEST_F()
182 auto inner_tuple = builder.AddInstruction( in TEST_F()
185 auto constant3 = builder.AddInstruction( in TEST_F()
187 auto tuple = builder.AddInstruction( in TEST_F()
239 auto constant1 = builder.AddInstruction( in TEST_F()
241 auto constant2 = builder.AddInstruction( in TEST_F()
[all …]
Dbfloat16_conversion_folding_test.cc88 HloInstruction* a = builder.AddInstruction( in TEST_F()
90 HloInstruction* b = builder.AddInstruction( in TEST_F()
92 HloInstruction* c = builder.AddInstruction( in TEST_F()
95 HloInstruction* add0 = builder.AddInstruction( in TEST_F()
98 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add0)); in TEST_F()
99 HloInstruction* convert1 = builder.AddInstruction( in TEST_F()
102 HloInstruction* add1 = builder.AddInstruction( in TEST_F()
104 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add1)); in TEST_F()
122 HloInstruction* a = builder.AddInstruction( in TEST_F()
124 HloInstruction* b = builder.AddInstruction( in TEST_F()
[all …]
Dhlo_instruction_test.cc150 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
152 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
153 auto add = builder.AddInstruction( in TEST_F()
183 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
185 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
186 auto exp1 = builder.AddInstruction( in TEST_F()
188 auto exp2 = builder.AddInstruction( in TEST_F()
190 auto add = builder.AddInstruction( in TEST_F()
222 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
223 auto add = builder.AddInstruction( in TEST_F()
[all …]
Ddefuser_test.cc49 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
51 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); in TEST_F()
52 builder.AddInstruction( in TEST_F()
65 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
67 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); in TEST_F()
68 auto add = builder.AddInstruction( in TEST_F()
89 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
91 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); in TEST_F()
92 auto add = builder.AddInstruction( in TEST_F()
94 builder.AddInstruction( in TEST_F()
[all …]
Ddynamic_dimension_inference_test.cc55 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
57 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
59 embedded_builder.AddInstruction( in GetAdd()
66 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
68 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
70 embedded_builder.AddInstruction(HloInstruction::CreateCompare( in GetGe()
84 auto param = builder.AddInstruction( in TEST_F()
86 auto param2 = builder.AddInstruction( in TEST_F()
107 auto param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
130 auto param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
Dbuffer_liveness_test.cc98 builder.AddInstruction(HloInstruction::CreateParameter(0, vec_, "param")); in BuildDummyComputation()
112 builder.AddInstruction(HloInstruction::CreateParameter(0, vec_, "param")); in TEST_F()
113 auto negate = builder.AddInstruction( in TEST_F()
115 auto exp = builder.AddInstruction( in TEST_F()
117 auto log = builder.AddInstruction( in TEST_F()
156 auto param0 = builder.AddInstruction( in TEST_F()
158 auto param1 = builder.AddInstruction( in TEST_F()
160 auto negate = builder.AddInstruction( in TEST_F()
162 auto exp = builder.AddInstruction( in TEST_F()
164 auto add = builder.AddInstruction( in TEST_F()
[all …]
Dhlo_computation_test.cc53 auto param = builder.AddInstruction( in CreateNegateComputation()
55 builder.AddInstruction( in CreateNegateComputation()
64 auto param = builder.AddInstruction( in CreateMapComputation()
66 builder.AddInstruction( in CreateMapComputation()
104 auto param = builder.AddInstruction( in TEST_F()
106 auto map1 = builder.AddInstruction( in TEST_F()
108 auto map2 = builder.AddInstruction( in TEST_F()
110 builder.AddInstruction( in TEST_F()
127 auto constant = builder.AddInstruction( in TEST_F()
138 auto constant = builder.AddInstruction( in TEST_F()
[all …]
Dhlo_cse_test.cc55 auto constant1 = builder.AddInstruction( in TEST_F()
57 auto constant2 = builder.AddInstruction( in TEST_F()
59 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
83 auto constant1 = builder.AddInstruction( in TEST_F()
86 auto constant2 = builder.AddInstruction( in TEST_F()
89 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
115 auto constant1 = builder.AddInstruction( in TEST_F()
118 auto constant2 = builder.AddInstruction( in TEST_F()
121 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
146 constants.push_back(builder.AddInstruction( in TEST_F()
[all …]
Dalgebraic_simplifier_test.cc60 HloInstruction* param0 = builder.AddInstruction( in TEST_F()
62 HloInstruction* zero = builder.AddInstruction( in TEST_F()
64 builder.AddInstruction( in TEST_F()
283 HloInstruction* param0 = builder.AddInstruction( in TEST_F()
285 HloInstruction* zero = builder.AddInstruction( in TEST_F()
287 builder.AddInstruction( in TEST_F()
302 HloInstruction* param0 = builder.AddInstruction( in TEST_F()
304 HloInstruction* param1 = builder.AddInstruction( in TEST_F()
306 HloInstruction* one = builder.AddInstruction( in TEST_F()
308 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
[all …]
Dhlo_subcomputation_unification_test.cc34 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x")); in CreateR0S32IdentityComputation()
41 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x")); in CreateR0S32AdditionComputation()
43 builder.AddInstruction(HloInstruction::CreateParameter(1, r0s32_, "y")); in CreateR0S32AdditionComputation()
44 builder.AddInstruction( in CreateR0S32AdditionComputation()
53 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in CreateR1S32AdditionComputation()
55 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "y")); in CreateR1S32AdditionComputation()
56 builder.AddInstruction( in CreateR1S32AdditionComputation()
76 auto constant = builder.AddInstruction( in TEST_F()
78 auto x = builder.AddInstruction( in TEST_F()
80 auto y = builder.AddInstruction( in TEST_F()
[all …]
Dconditional_simplifier_test.cc55 true_computation_builder.AddInstruction(HloInstruction::CreateParameter( in MakeConditional()
57 auto one = true_computation_builder.AddInstruction( in MakeConditional()
60 true_computation_builder.AddInstruction(HloInstruction::CreateBinary( in MakeConditional()
72 auto param = false_computation_builder.AddInstruction( in MakeConditional()
75 auto forty_two = false_computation_builder.AddInstruction( in MakeConditional()
78 false_computation_builder.AddInstruction(HloInstruction::CreateBinary( in MakeConditional()
84 auto false_instrn = builder.AddInstruction( in MakeConditional()
86 auto false_param = builder.AddInstruction(HloInstruction::CreateParameter( in MakeConditional()
88 auto one = builder.AddInstruction( in MakeConditional()
91 builder.AddInstruction(HloInstruction::CreateConditional( in MakeConditional()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc57 auto input1 = builder.AddInstruction( in TEST_F()
59 auto input2 = builder.AddInstruction( in TEST_F()
62 auto add1 = builder.AddInstruction( in TEST_F()
64 builder.AddInstruction( in TEST_F()
96 auto input = builder.AddInstruction( in TEST_F()
98 auto negate = builder.AddInstruction( in TEST_F()
100 auto ceil = builder.AddInstruction( in TEST_F()
102 auto exp = builder.AddInstruction( in TEST_F()
104 auto floor = builder.AddInstruction( in TEST_F()
106 auto two = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc51 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
53 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
56 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
58 HloInstruction* dot = builder.AddInstruction( in TEST_F()
70 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
72 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
75 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
77 HloInstruction* dot = builder.AddInstruction( in TEST_F()
89 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
91 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc98 hlos[i + 1] = builder.AddInstruction(HloInstruction::CreateConstant( in TestElementwise2D()
124 hlos[0] = builder.AddInstruction(std::move(root_hlo)); in TestElementwise2D()
203 auto const0 = builder.AddInstruction(HloInstruction::CreateConstant( in XLA_TEST_F()
205 auto const1 = builder.AddInstruction(HloInstruction::CreateConstant( in XLA_TEST_F()
207 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
209 auto reshape3 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
211 auto const4 = builder.AddInstruction(HloInstruction::CreateConstant( in XLA_TEST_F()
213 auto concat5 = builder.AddInstruction(HloInstruction::CreateConcatenate( in XLA_TEST_F()
215 auto const6 = builder.AddInstruction(HloInstruction::CreateConstant( in XLA_TEST_F()
217 auto negate7 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment_test.cc47 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
49 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
51 HloInstruction* z = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
54 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y)); in TEST_F()
56 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, z)); in TEST_F()
68 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
70 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
73 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y)); in TEST_F()
75 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, y, x)); in TEST_F()
76 HloInstruction* add = builder.AddInstruction( in TEST_F()
[all …]

1234567