| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| stream_assignment_test.cc | 44 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, x, y));
46 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, dot1, z));
63 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, x, y));
65 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, y, x));
67 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, dot1, dot2));
94 HloInstruction* d00 = builder.AddInstruction(HloInstruction::CreateBinary(
97 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, params[1], d00));
99 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, d00, params[4]));
101 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, params[0], d10));
103 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, d10, d11))
[all...] |
| fusion_merger_test.cc | 65 auto out0 = builder.AddInstruction(HloInstruction::CreateBinary(
71 HloInstruction::CreateBinary(data_shape_, HloOpcode::kAdd, gte1, gte2));
76 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary(
78 auto out1 = builder.AddInstruction(HloInstruction::CreateBinary(
82 auto mul0 = builder.AddInstruction(HloInstruction::CreateBinary(
84 auto out2 = builder.AddInstruction(HloInstruction::CreateBinary(
132 auto mul0 = builder.AddInstruction(HloInstruction::CreateBinary(
135 mul0 = builder.AddInstruction(HloInstruction::CreateBinary(
144 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary(
146 auto out0 = builder.AddInstruction(HloInstruction::CreateBinary(
[all...] |
| gpu_hlo_support_checker_test.cc | 43 builder.AddInstruction(HloInstruction::CreateBinary(
58 builder.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_schedule_test.cc | 68 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, x, y));
70 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, dot1, z));
126 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, x, y));
128 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, y, z));
130 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, add1, add2));
196 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, x, y));
198 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, y, x));
200 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, dot1, dot2));
262 HloInstruction* d00 = builder.AddInstruction(HloInstruction::CreateBinary(
265 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kDot, params[1], d00))
[all...] |
| cudnn_batchnorm_rewriter.cc | 127 computation_->AddInstruction(HloInstruction::CreateBinary(
132 computation_->AddInstruction(HloInstruction::CreateBinary(
176 computation_->AddInstruction(HloInstruction::CreateBinary(
180 computation_->AddInstruction(HloInstruction::CreateBinary(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander.cc | 92 HloInstruction::CreateBinary(ShapeUtil::MakeShape(primitive_type, {}),
205 auto operand_squared = add(HloInstruction::CreateBinary(
232 auto mean = add(HloInstruction::CreateBinary(
239 auto square_mean = add(HloInstruction::CreateBinary(
243 auto mean_square = add(HloInstruction::CreateBinary(
247 auto var = add(HloInstruction::CreateBinary(
254 auto var_add_epsilon = add(HloInstruction::CreateBinary(
263 auto rsqrt_var_add_epsilon = add(HloInstruction::CreateBinary(
267 auto operand_minus_mean = add(HloInstruction::CreateBinary(
272 HloInstruction::CreateBinary(operand_shape, HloOpcode::kMultiply
[all...] |
| defuser_test.cc | 52 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
67 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
90 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
117 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
121 HloInstruction::CreateBinary(shape_, HloOpcode::kSubtract, add, negate));
123 HloInstruction::CreateBinary(shape_, HloOpcode::kMultiply, sub, param3));
125 HloInstruction::CreateBinary(shape_, HloOpcode::kDivide, mul, param3));
129 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, constant, div));
155 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1));
159 HloInstruction::CreateBinary(shape_, HloOpcode::kSubtract, add, negate))
[all...] |
| bfloat16_conversion_folding_test.cc | 89 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, a, b));
96 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, convert1, c));
123 HloInstruction::CreateBinary(f32_shape, HloOpcode::kMultiply, a, b));
129 HloInstruction* mul1 = builder.AddInstruction(HloInstruction::CreateBinary(
158 HloInstruction::CreateBinary(f32_shape, HloOpcode::kSubtract, a, b));
164 HloInstruction* sub1 = builder.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_instruction_test.cc | 151 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar));
188 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar));
221 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, foo));
252 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, c0));
254 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, c0, param1));
256 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright));
299 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, neg1));
301 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, neg1, param1));
303 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright));
337 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value))
[all...] |
| bfloat16_normalization_test.cc | 90 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kAdd, a, b));
93 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, add0, c));
118 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, a, b));
121 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, mul0, c));
148 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, a, b));
151 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, sub0, c));
177 HloInstruction::CreateBinary(bf16_scalar_shape, HloOpcode::kAdd,
|
| hlo_subcomputation_unification_test.cc | 46 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y));
58 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, x, y));
84 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y));
123 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y));
195 HloInstruction::CreateBinary(r0f32_, HloOpcode::kPower, x, y));
|
| hlo_dce_test.cc | 59 builder.AddInstruction(HloInstruction::CreateBinary(
136 auto dead_add = builder.AddInstruction(HloInstruction::CreateBinary(
145 builder.AddInstruction(HloInstruction::CreateBinary(
149 builder.AddInstruction(HloInstruction::CreateBinary(
227 cond_builder.AddInstruction(HloInstruction::CreateBinary(
241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed));
334 subcomp_builder.AddInstruction(HloInstruction::CreateBinary(
375 subcomp_builder.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_execution_profile_test.cc | 58 builder.AddInstruction(HloInstruction::CreateBinary(
61 builder.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_graph_dumper_test.cc | 61 sums.push_back(b.AddInstruction(HloInstruction::CreateBinary(
64 sums.push_back(b.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_cse_test.cc | 58 builder.AddInstruction(HloInstruction::CreateBinary(
88 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
120 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
311 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1));
313 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1));
315 HloInstruction::CreateBinary(shape_r0, HloOpcode::kMultiply, add1, add2));
355 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary(
362 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary(
397 builder.AddInstruction(HloInstruction::CreateBinary(
437 builder.AddInstruction(HloInstruction::CreateBinary(
[all...] |
| hlo_rematerialization_test.cc | 354 HloInstruction::CreateBinary(vec1024_shape_, HloOpcode::kAdd, rng, tanh));
355 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary(
357 builder.AddInstruction(HloInstruction::CreateBinary(
359 builder.AddInstruction(HloInstruction::CreateBinary(
361 builder.AddInstruction(HloInstruction::CreateBinary(
439 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary(
443 auto add_2 = builder.AddInstruction(HloInstruction::CreateBinary(
447 auto add_3 = builder.AddInstruction(HloInstruction::CreateBinary(
451 auto add_4 = builder.AddInstruction(HloInstruction::CreateBinary(
555 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary(
[all...] |
| reshape_mover_test.cc | 50 builder.AddInstruction(HloInstruction::CreateBinary(
88 builder.AddInstruction(HloInstruction::CreateBinary(
113 builder.AddInstruction(HloInstruction::CreateBinary(
139 builder.AddInstruction(HloInstruction::CreateBinary(
227 builder.AddInstruction(HloInstruction::CreateBinary(
315 builder.AddInstruction(HloInstruction::CreateBinary(
357 builder.AddInstruction(HloInstruction::CreateBinary(
384 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
498 auto multiply = builder.AddInstruction(HloInstruction::CreateBinary(
540 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary(
[all...] |
| algebraic_simplifier_test.cc | 62 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, zero));
83 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, constant, param0));
107 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, constant1));
109 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, add1, constant2));
131 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0));
153 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0));
174 HloInstruction::CreateBinary(r0f32, HloOpcode::kSubtract, param0, zero));
194 builder.AddInstruction(HloInstruction::CreateBinary(
218 HloInstruction::CreateBinary(r0f32, HloOpcode::kDivide, param0, param1));
220 HloInstruction::CreateBinary(r0f32, HloOpcode::kDivide, div, param2))
[all...] |
| while_loop_invariant_code_motion_test.cc | 73 builder.AddInstruction(HloInstruction::CreateBinary(
118 builder.AddInstruction(HloInstruction::CreateBinary(
121 builder.AddInstruction(HloInstruction::CreateBinary(
129 builder.AddInstruction(HloInstruction::CreateBinary(
132 builder.AddInstruction(HloInstruction::CreateBinary(
187 builder.AddInstruction(HloInstruction::CreateBinary(
225 builder.AddInstruction(HloInstruction::CreateBinary(
345 builder.AddInstruction(HloInstruction::CreateBinary(
393 builder.AddInstruction(HloInstruction::CreateBinary(
|
| hlo_matchers_test.cc | 43 auto mul = HloInstruction::CreateBinary(shape, HloOpcode::kMultiply,
45 auto add = HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param.get(),
|
| hlo_verifier_test.cc | 110 HloInstruction::CreateBinary(s2, HloOpcode::kAdd, param, param));
115 HloInstruction::CreateBinary(s2, HloOpcode::kMultiply, add, add));
|
| inliner_test.cc | 49 max_builder.AddInstruction(HloInstruction::CreateBinary(
121 max_builder.AddInstruction(HloInstruction::CreateBinary(
|
| liveness_util_test.cc | 65 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1));
157 HloInstruction::CreateBinary(out_shape, HloOpcode::kEq, param0, param1));
290 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
328 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
364 HloInstruction::CreateBinary(data_shape, HloOpcode::kAdd, reverse, two));
386 builder.AddInstruction(HloInstruction::CreateBinary(
396 HloInstruction::CreateBinary(data_shape, HloOpcode::kAdd, data, data));
436 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, sub_param, ones));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker_test.cc | 43 builder.AddInstruction(HloInstruction::CreateBinary(
58 builder.AddInstruction(HloInstruction::CreateBinary(
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| multioutput_fusion_test.cc | 68 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary(
77 HloInstruction* add2 = builder.AddInstruction(HloInstruction::CreateBinary(
79 HloInstruction* sub = builder.AddInstruction(HloInstruction::CreateBinary(
129 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary(
132 builder.AddInstruction(HloInstruction::CreateBinary(
|
