| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 58 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
60 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
69 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAddTuple()
72 embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAddTuple()
74 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAddTuple()
77 embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAddTuple()
89 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
91 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
108 HloInstruction::CreateParameter(0, input_shape, "param")); in TEST_F()
110 HloInstruction::CreateParameter(1, scalar_shape_, "param")); in TEST_F()
[all …]
|
| D | hlo_live_range_test.cc | 90 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
92 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
115 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
117 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
121 HloInstruction::CreateParameter(2, f32vec4_, "paramY")); in TEST_F()
151 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
153 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
157 HloInstruction::CreateParameter(2, f32vec4_, "paramY")); in TEST_F()
188 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
190 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
[all …]
|
| D | bfloat16_normalization_test.cc | 100 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
102 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
104 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
128 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
130 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
132 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
158 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
160 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
162 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
190 HloInstruction::CreateParameter(0, bf16_scalar_shape, "param0")); in TEST_F()
[all …]
|
| D | defuser_test.cc | 49 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
51 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); 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()
89 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
91 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); in TEST_F()
115 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
117 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1")); in TEST_F()
119 builder.AddInstruction(HloInstruction::CreateParameter(2, shape_, "p2")); in TEST_F()
154 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0")); in TEST_F()
[all …]
|
| D | tuple_simplifier_test.cc | 62 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
64 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_F()
66 HloInstruction::CreateParameter(2, scalar_shape_, "param2")); in TEST_F()
78 HloInstruction::CreateParameter(0, tuple_shape_, "param")); in TEST_F()
91 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
93 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_F()
95 HloInstruction::CreateParameter(2, scalar_shape_, "param2")); in TEST_F()
115 HloInstruction::CreateParameter(0, scalar_shape_, "param")); in TEST_F()
145 HloInstruction::CreateParameter(0, scalar_shape_, "param")); in TEST_F()
175 HloInstruction::CreateParameter(0, tuple_shape_, "param")); in TEST_F()
[all …]
|
| D | bfloat16_conversion_folding_test.cc | 89 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
91 HloInstruction::CreateParameter(1, f32_shape, "b")); in TEST_F()
93 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
123 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
125 HloInstruction::CreateParameter(1, f32_shape, "b")); in TEST_F()
127 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
158 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
160 HloInstruction::CreateParameter(1, f32_shape, "b")); in TEST_F()
162 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
193 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
[all …]
|
| D | bfloat16_propagation_test.cc | 103 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()
144 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "a")); in TEST_F()
146 HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {}), "b")); in TEST_F()
155 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
157 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
159 builder.AddInstruction(HloInstruction::CreateParameter(2, shape, "c")); in TEST_F()
202 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
204 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
[all …]
|
| D | batchnorm_expander_test.cc | 63 HloInstruction::CreateParameter(0, input_shape, "activation")); in TEST_F()
66 HloInstruction::CreateParameter(1, scale_shape, "scale")); in TEST_F()
69 HloInstruction::CreateParameter(2, offset_shape, "offset")); in TEST_F()
100 HloInstruction::CreateParameter(0, input_shape, "activation")); in TEST_F()
103 HloInstruction::CreateParameter(1, scale_shape, "scale")); in TEST_F()
106 HloInstruction::CreateParameter(2, mean_shape, "mean")); in TEST_F()
109 HloInstruction::CreateParameter(3, var_shape, "var")); in TEST_F()
112 HloInstruction::CreateParameter(4, grad_output_shape, "grad_output")); in TEST_F()
|
| D | hlo_instruction_test.cc | 138 auto parameter = HloInstruction::CreateParameter(1, r0f32_, "foo"); in TEST_F()
152 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
154 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
185 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
187 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); in TEST_F()
224 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
251 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
253 HloInstruction::CreateParameter(1, r0f32_, "param1")); in TEST_F()
296 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
298 HloInstruction::CreateParameter(1, r0f32_, "param1")); in TEST_F()
[all …]
|
| D | reshape_mover_test.cc | 43 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
45 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
112 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
114 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
139 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
141 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
191 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
196 auto param2 = builder.AddInstruction(HloInstruction::CreateParameter(
232 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
237 HloInstruction::CreateParameter(1, root_shape, "param1"));
[all …]
|
| D | hlo_dce_test.cc | 134 auto live_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
136 auto dead_param1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
138 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
221 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
231 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
261 HloInstruction::CreateParameter(0, shape, "cond_param")); in TEST_F()
274 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
288 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
320 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
333 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
[all …]
|
| D | hlo_subcomputation_unification_test.cc | 34 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()
53 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in CreateR1S32AdditionComputation()
55 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "y")); in CreateR1S32AdditionComputation()
134 HloInstruction::CreateParameter(0, r1s32_5_, "param1")); in TEST_F()
136 HloInstruction::CreateParameter(1, r1s32_3_, "param2")); in TEST_F()
160 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in TEST_F()
162 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in TEST_F()
|
| D | buffer_assignment_test.cc | 165 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildMapComputationPlus1()
176 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildReduceComputation()
178 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in BuildReduceComputation()
197 HloInstruction::CreateParameter(0, t_s32_f32v4_, "x")); in BuildWhileConditionComputation()
225 HloInstruction::CreateParameter(0, t_s32_f32v4_, "x")); in BuildWhileBodyComputation()
242 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildR0F32UnaryOpComputation()
391 HloInstruction::CreateParameter(0, f32vec100_, "param0")); in TEST_F()
437 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p")); in TEST_F()
441 HloInstruction::CreateParameter(1, f32vec100_, "p1")); in TEST_F()
443 HloInstruction::CreateParameter(2, f32vec100_, "p2")); in TEST_F()
[all …]
|
| D | while_loop_invariant_code_motion_test.cc | 54 HloInstruction::CreateParameter(0, param_shape, "param")); in MakeAlwaysTrueComputation()
69 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
85 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
111 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
144 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
183 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
198 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
222 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
238 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
261 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 63 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
368 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
524 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
526 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
546 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
548 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
568 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
570 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
595 HloInstruction::CreateParameter(0, scalar_shape, "p0")); in TEST_F()
597 HloInstruction::CreateParameter(1, scalar_shape, "p1")); in TEST_F()
[all …]
|
| D | hlo_alias_analysis_test.cc | 150 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
152 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_F()
199 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
201 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_F()
227 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
276 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
335 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
349 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
357 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
407 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
[all …]
|
| D | map_inliner_test.cc | 46 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
48 HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
83 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
118 HloInstruction::CreateParameter(1, r0f32, "x")); in TEST_F()
120 HloInstruction::CreateParameter(0, r0f32, "y")); in TEST_F()
153 param_builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32, "p0")); in TEST_F()
154 param_builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32, "p1")); in TEST_F()
|
| D | copy_insertion_test.cc | 91 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "x")); in TEST_F()
175 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "x")); in TEST_F()
177 HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {}), "y")); in TEST_F()
237 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {4}), "x")); in TEST_F()
281 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {4}), "x")); in TEST_F()
305 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
341 auto param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
419 HloInstruction::CreateParameter(0, loop_state_shape, "loop_state")); in BuildConditionComputation()
440 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildDependentBodyComputation()
480 HloInstruction::CreateParameter(0, loop_state_shape, "loop_state")); in BuildDependentBodyComputation2()
[all …]
|
| D | hlo_dataflow_analysis_test.cc | 89 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in CreateR0F32UnaryOpComputation()
151 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
153 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
259 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
261 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
322 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
324 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
332 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
334 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
395 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 52 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
54 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
71 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
73 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
90 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
92 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
111 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
113 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
133 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
135 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 101 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
104 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
143 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
146 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
174 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
177 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
204 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
207 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
233 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
236 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
|
| D | gpu_layout_assignment_test.cc | 61 HloInstruction::CreateParameter(0, ashape, "x")); in TEST_F()
63 HloInstruction::CreateParameter(1, ashape, "y")); in TEST_F()
132 HloInstruction::CreateParameter(0, shape, "operand")); in TEST_F()
134 HloInstruction::CreateParameter(1, aux_shape, "scale")); in TEST_F()
136 HloInstruction::CreateParameter(2, aux_shape, "offset")); in TEST_F()
138 HloInstruction::CreateParameter(3, aux_shape, "mean")); in TEST_F()
140 HloInstruction::CreateParameter(4, aux_shape, "variance")); in TEST_F()
206 HloInstruction::CreateParameter(0, shape, "operand")); in TEST_F()
208 HloInstruction::CreateParameter(1, offset_scale_shape, "scale")); in TEST_F()
210 HloInstruction::CreateParameter(2, offset_scale_shape, "offset")); in TEST_F()
[all …]
|
| D | horizontal_loop_fusion_test.cc | 209 HloInstruction::CreateParameter(i * 3 + 0, shape, "var.in")); in TEST_F()
211 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
214 HloInstruction::CreateParameter(i * 3 + 2, shape, "delta")); in TEST_F()
300 HloInstruction::CreateParameter(i * 9 + 0, shape, "grad")); in TEST_F()
302 HloInstruction::CreateParameter(i * 9 + 1, shape, "ms")); in TEST_F()
304 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
307 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
324 HloInstruction::CreateParameter(i * 9 + 4, shape, "momemtum")); in TEST_F()
326 HloInstruction::CreateParameter(i * 9 + 5, shape, "mom")); in TEST_F()
327 HloInstruction* lr = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
|
| D | stream_assignment_test.cc | 47 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()
68 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
70 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
101 params.push_back(builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 51 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in TEST_F()
78 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in TEST_F()
119 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
121 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
132 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F()
|