| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | stream_assignment_test.cc | 37 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter(
39 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter(
41 HloInstruction* z = builder.AddInstruction(HloInstruction::CreateParameter(
43 HloInstruction* dot1 = builder.AddInstruction(
45 HloInstruction* dot2 = builder.AddInstruction(
58 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter(
60 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter(
62 HloInstruction* dot1 = builder.AddInstruction(
64 HloInstruction* dot2 = builder.AddInstruction(
66 HloInstruction* add = builder.AddInstruction(
[all...] |
| H A D | fusion_merger_test.cc | 49 auto param = builder.AddInstruction(
53 auto gte0 = builder.AddInstruction(
55 auto gte1 = builder.AddInstruction(
57 auto gte2 = builder.AddInstruction(
61 auto one_vec = builder.AddInstruction(HloInstruction::CreateConstant(
65 auto out0 = builder.AddInstruction(HloInstruction::CreateBinary(
70 auto add1 = builder.AddInstruction(
76 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary(
78 auto out1 = builder.AddInstruction(HloInstruction::CreateBinary(
82 auto mul0 = builder.AddInstruction(HloInstructio
[all...] |
| H A D | gpu_hlo_support_checker_test.cc | 39 HloInstruction* param0 = builder.AddInstruction(
41 HloInstruction* param1 = builder.AddInstruction(
43 builder.AddInstruction(HloInstruction::CreateBinary(
54 HloInstruction* param0 = builder.AddInstruction(
56 HloInstruction* param1 = builder.AddInstruction(
58 builder.AddInstruction(HloInstruction::CreateBinary(
|
| H A D | instruction_fusion_test.cc | 31 HloInstruction* const0 = builder.AddInstruction(
33 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary(
36 builder.AddInstruction(HloInstruction::CreateBroadcast(
51 HloInstruction* const0 = builder.AddInstruction(
53 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary(
56 builder.AddInstruction(HloInstruction::CreateBroadcast(
71 HloInstruction* const0 = builder.AddInstruction(
73 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary(
75 HloInstruction* reshape2 = builder.AddInstruction(
90 HloInstruction* const0 = builder.AddInstruction(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_dce_test.cc | 55 auto constant1 = builder.AddInstruction(
57 auto constant2 = builder.AddInstruction(
59 builder.AddInstruction(HloInstruction::CreateBinary(
76 auto constant = builder.AddInstruction(
78 builder.AddInstruction(
80 builder.AddInstruction(HloInstruction::CreateTuple({}));
97 auto live_param = builder.AddInstruction(HloInstruction::CreateParameter(
99 auto dead_param1 = builder.AddInstruction(HloInstruction::CreateParameter(
101 builder.AddInstruction(HloInstruction::CreateParameter(
105 builder.AddInstruction(HloInstructio
[all...] |
| H A D | while_loop_simplifier_test.cc | 43 auto loop_iter_init = builder.AddInstruction(
45 auto loop_data_init = builder.AddInstruction(
47 auto loop_init = builder.AddInstruction(
53 auto loop_var = cond_builder.AddInstruction(
56 cond_builder.AddInstruction(HloInstruction::CreateGetTupleElement(
58 auto limit = cond_builder.AddInstruction(HloInstruction::CreateConstant(
60 cond_builder.AddInstruction(HloInstruction::CreateBinary(
69 auto loop_var = body_builder.AddInstruction(
72 body_builder.AddInstruction(HloInstruction::CreateGetTupleElement(
75 body_builder.AddInstruction(HloInstructio
[all...] |
| H A D | bfloat16_conversion_folding_test.cc | 81 HloInstruction* a = builder.AddInstruction(
83 HloInstruction* b = builder.AddInstruction(
85 HloInstruction* c = builder.AddInstruction(
88 HloInstruction* add0 = builder.AddInstruction(
91 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add0));
92 HloInstruction* convert1 = builder.AddInstruction(
95 HloInstruction* add1 = builder.AddInstruction(
97 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add1));
115 HloInstruction* a = builder.AddInstruction(
117 HloInstruction* b = builder.AddInstruction(
[all...] |
| H A D | hlo_alias_analysis_test.cc | 118 auto constant1 = builder.AddInstruction(
120 auto constant2 = builder.AddInstruction(
122 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
146 auto param0 = builder.AddInstruction(
148 auto param1 = builder.AddInstruction(
151 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1}));
152 auto gte0 = builder.AddInstruction(
154 auto gte1 = builder.AddInstruction(
156 builder.AddInstruction(
194 auto param0 = builder.AddInstruction(
[all...] |
| H A D | reshape_mover_test.cc | 42 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
44 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
47 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0));
49 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
50 builder.AddInstruction(HloInstruction::CreateBinary(
79 auto rng0 = builder.AddInstruction(
83 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));
85 auto const1 = builder.AddInstruction(
88 builder.AddInstruction(HloInstruction::CreateBinary(
105 auto param0 = builder.AddInstruction(HloInstructio
[all...] |
| H A D | hlo_scheduling_test.cc | 43 HloInstruction* cond_param = cond_builder.AddInstruction(
45 HloInstruction* cond_iter = cond_builder.AddInstruction(
47 HloInstruction* cond_data = cond_builder.AddInstruction(
50 HloInstruction* cond_lt = cond_builder.AddInstruction(
58 HloInstruction* body_param = body_builder.AddInstruction(
65 HloInstruction* iter = builder.AddInstruction(
67 HloInstruction* data = builder.AddInstruction(
71 builder.AddInstruction(HloInstruction::CreateTuple({iter, data}));
74 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
|
| H A D | while_loop_invariant_code_motion_test.cc | 52 builder.AddInstruction(
54 builder.AddInstruction(
66 HloInstruction* param = builder.AddInstruction(
68 HloInstruction* gte_0 = builder.AddInstruction(
70 HloInstruction* gte_1 = builder.AddInstruction(
73 builder.AddInstruction(HloInstruction::CreateBinary(
75 builder.AddInstruction(
82 auto* init_value = builder.AddInstruction(
84 builder.AddInstruction(HloInstruction::CreateWhile(
108 HloInstruction* param = builder.AddInstruction(
[all...] |
| H A D | graphviz_example.cc | 47 auto x_value = builder.AddInstruction(HloInstruction::CreateParameter(
49 auto half = builder.AddInstruction(
51 builder.AddInstruction(HloInstruction::CreateBinary(
60 auto lhs = builder.AddInstruction(
62 auto rhs = builder.AddInstruction(
64 builder.AddInstruction(
74 auto arg = builder.AddInstruction(
76 builder.AddInstruction(
96 builder.AddInstruction(HloInstruction::CreateParameter(0, vshape, "foo"));
98 builder.AddInstruction(HloInstructio
[all...] |
| H A D | tuple_simplifier_test.cc | 55 HloInstruction* param0 = builder.AddInstruction(
57 HloInstruction* param1 = builder.AddInstruction(
59 HloInstruction* param2 = builder.AddInstruction(
61 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1, param2}));
71 HloInstruction* param = builder.AddInstruction(
73 builder.AddInstruction(
84 HloInstruction* param0 = builder.AddInstruction(
86 HloInstruction* param1 = builder.AddInstruction(
88 HloInstruction* param2 = builder.AddInstruction(
90 HloInstruction* tuple = builder.AddInstruction(
[all...] |
| H A D | defuser_test.cc | 48 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
50 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
51 builder.AddInstruction(
63 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
65 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
66 auto add = builder.AddInstruction(
86 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
88 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
89 auto add = builder.AddInstruction(
91 builder.AddInstruction(
[all...] |
| H A D | inliner_test.cc | 45 auto param1 = max_builder.AddInstruction(
47 auto param2 = max_builder.AddInstruction(
49 max_builder.AddInstruction(HloInstruction::CreateBinary(
54 auto lhs = builder.AddInstruction(
56 auto rhs = builder.AddInstruction(
58 builder.AddInstruction(
82 auto param1 = const2_builder.AddInstruction(
85 const2_builder.AddInstruction(
90 auto lhs = builder.AddInstruction(HloInstruction::CreateConstant(
92 builder.AddInstruction(
[all...] |
| H A D | liveness_util_test.cc | 58 auto tuple = builder.AddInstruction(HloInstruction::CreateParameter(
60 auto gte0 = builder.AddInstruction(
62 auto gte1 = builder.AddInstruction(
64 builder.AddInstruction(
86 auto tuple = builder.AddInstruction(HloInstruction::CreateParameter(
88 auto gte0 = builder.AddInstruction(
90 auto gte1 = builder.AddInstruction(
94 auto starts = builder.AddInstruction(
96 auto update = builder.AddInstruction(HloInstruction::CreateConstant(
99 builder.AddInstruction(HloInstructio
[all...] |
| H A D | hlo_subcomputation_unification_test.cc | 35 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x"));
42 builder.AddInstruction(HloInstruction::CreateParameter(0, r0s32_, "x"));
44 builder.AddInstruction(HloInstruction::CreateParameter(1, r0s32_, "y"));
45 builder.AddInstruction(
54 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x"));
56 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "y"));
57 builder.AddInstruction(
77 auto constant = builder.AddInstruction(
79 auto x = builder.AddInstruction(
81 auto y = builder.AddInstruction(
[all...] |
| H A D | hlo_cse_test.cc | 54 auto constant1 = builder.AddInstruction(
56 auto constant2 = builder.AddInstruction(
58 builder.AddInstruction(HloInstruction::CreateBinary(
82 auto constant1 = builder.AddInstruction(
85 auto constant2 = builder.AddInstruction(
88 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
114 auto constant1 = builder.AddInstruction(
117 auto constant2 = builder.AddInstruction(
120 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
144 builder.AddInstruction(
[all...] |
| H A D | copy_insertion_test.cc | 90 HloInstruction* x = builder.AddInstruction(
93 builder.AddInstruction(HloInstruction::CreateTuple({x}));
110 HloInstruction* constant = builder.AddInstruction(
113 builder.AddInstruction(HloInstruction::CreateTuple({constant}));
133 HloInstruction* constant = builder.AddInstruction(
135 HloInstruction* copy_1 = builder.AddInstruction(HloInstruction::CreateUnary(
137 HloInstruction* copy_2 = builder.AddInstruction(HloInstruction::CreateUnary(
139 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary(
141 HloInstruction* add_copy = builder.AddInstruction(
164 HloInstruction* constant1 = builder.AddInstruction(
[all...] |
| H A D | batchnorm_expander_test.cc | 48 HloInstruction* param0 = builder.AddInstruction(
51 HloInstruction* param1 = builder.AddInstruction(
54 HloInstruction* param2 = builder.AddInstruction(
57 builder.AddInstruction(HloInstruction::CreateBatchNormTraining(
84 HloInstruction* param0 = builder.AddInstruction(
87 HloInstruction* param1 = builder.AddInstruction(
90 HloInstruction* param2 = builder.AddInstruction(
93 HloInstruction* param3 = builder.AddInstruction(
96 HloInstruction* param4 = builder.AddInstruction(
99 builder.AddInstruction(HloInstructio
[all...] |
| H A D | hlo_dataflow_analysis_test.cc | 84 HloInstruction* param0 = builder.AddInstruction(
86 builder.AddInstruction(
103 auto constant1 = builder.AddInstruction(
105 auto constant2 = builder.AddInstruction(
107 auto add = builder.AddInstruction(HloInstruction::CreateBinary(
145 auto param0 = builder.AddInstruction(
147 auto param1 = builder.AddInstruction(
150 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1}));
151 auto gte0 = builder.AddInstruction(
153 auto gte1 = builder.AddInstruction(
[all...] |
| H A D | buffer_assignment_test.cc | 108 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x"));
109 auto value = builder.AddInstruction(
111 builder.AddInstruction(
126 auto const4 = builder.AddInstruction(
128 auto param = builder.AddInstruction(
130 auto index = builder.AddInstruction(
132 builder.AddInstruction(
151 auto const1 = builder.AddInstruction(
153 auto constv = builder.AddInstruction(HloInstruction::CreateConstant(
155 auto param = builder.AddInstruction(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 44 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
46 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
49 HloInstruction* exp0 = builder.AddInstruction(HloInstruction::CreateUnary(
51 HloInstruction* dot = builder.AddInstruction(
63 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
65 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
68 HloInstruction* exp1 = builder.AddInstruction(HloInstruction::CreateUnary(
70 HloInstruction* dot = builder.AddInstruction(
82 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
84 HloInstruction* arg1 = builder.AddInstruction(HloInstructio
[all...] |
| H A D | cpu_hlo_support_checker_test.cc | 39 HloInstruction* param0 = builder.AddInstruction(
41 HloInstruction* param1 = builder.AddInstruction(
43 builder.AddInstruction(HloInstruction::CreateBinary(
54 HloInstruction* param0 = builder.AddInstruction(
56 HloInstruction* param1 = builder.AddInstruction(
58 builder.AddInstruction(HloInstruction::CreateBinary(
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | multioutput_fusion_test.cc | 63 auto const0 = builder.AddInstruction(
65 auto param0 = builder.AddInstruction(
68 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary(
71 HloInstruction* broadcast = builder.AddInstruction(
74 auto param1 = builder.AddInstruction(
77 HloInstruction* add2 = builder.AddInstruction(HloInstruction::CreateBinary(
79 HloInstruction* sub = builder.AddInstruction(HloInstruction::CreateBinary(
84 HloInstruction* dot = builder.AddInstruction(
89 auto tuple = computation->AddInstruction(HloInstruction::CreateTuple(
91 auto gte0 = computation->AddInstruction(
[all...] |