| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 55 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
57 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetAdd()
66 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
68 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetGe()
85 HloInstruction::CreateParameter(0, input_shape, "param")); in TEST_F()
87 HloInstruction::CreateParameter(1, scalar_shape_, "param")); in TEST_F()
107 auto param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
130 auto param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
160 HloInstruction::CreateParameter(0, input_shape, "data_param")); in TEST_F()
162 HloInstruction::CreateParameter(1, scalar_shape_, "size_param")); in TEST_F()
[all …]
|
| D | bfloat16_normalization_test.cc | 98 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
100 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
102 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
126 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
128 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
130 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
156 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
158 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
160 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
188 HloInstruction::CreateParameter(0, bf16_scalar_shape, "param0")); in TEST_F()
[all …]
|
| D | dynamic_padder_test.cc | 58 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetScalarAddComputation()
60 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in GetScalarAddComputation()
77 HloInstruction::CreateParameter(0, input_shape, "data_param")); in TEST_F()
79 HloInstruction::CreateParameter(1, scalar_shape_, "size_param")); in TEST_F()
111 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
113 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
115 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
154 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
156 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
158 builder.AddInstruction(HloInstruction::CreateParameter( 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 | 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 | hlo_instruction_test.cc | 136 auto parameter = HloInstruction::CreateParameter(1, r0f32_, "foo"); in TEST_F()
150 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
152 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "bar")); 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()
222 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "foo")); in TEST_F()
249 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
251 HloInstruction::CreateParameter(1, r0f32_, "param1")); in TEST_F()
294 HloInstruction::CreateParameter(0, r0f32_, "param0")); in TEST_F()
296 HloInstruction::CreateParameter(1, r0f32_, "param1")); 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 | hlo_dce_test.cc | 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()
183 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
193 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
223 HloInstruction::CreateParameter(0, shape, "cond_param")); in TEST_F()
236 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
248 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
280 HloInstruction::CreateParameter(0, shape, "param")); in TEST_F()
293 HloInstruction::CreateParameter(0, shape, "param")); 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 | 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 | algebraic_simplifier_test.cc | 61 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
284 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
303 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
305 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
325 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
327 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
347 HloInstruction::CreateParameter(0, r0s32, "param0")); in TEST_F()
349 HloInstruction::CreateParameter(1, r0s32, "param1")); in TEST_F()
374 HloInstruction::CreateParameter(0, scalar_shape, "p0")); in TEST_F()
376 HloInstruction::CreateParameter(1, scalar_shape, "p1")); 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_5_, "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 | 158 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildMapComputationPlus1()
169 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildReduceComputation()
171 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y")); in BuildReduceComputation()
190 HloInstruction::CreateParameter(0, t_s32_f32v4_, "x")); in BuildWhileConditionComputation()
218 HloInstruction::CreateParameter(0, t_s32_f32v4_, "x")); in BuildWhileBodyComputation()
235 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x")); in BuildR0F32UnaryOpComputation()
384 HloInstruction::CreateParameter(0, f32vec100_, "param0")); in TEST_F()
430 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "p")); in TEST_F()
434 HloInstruction::CreateParameter(1, f32vec100_, "p1")); in TEST_F()
436 HloInstruction::CreateParameter(2, f32vec100_, "p2")); in TEST_F()
[all …]
|
| D | reduce_precision_insertion_test.cc | 53 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
80 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
108 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
110 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
139 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
168 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
204 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
233 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a")); in TEST_F()
238 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b")); in TEST_F()
268 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); 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 | 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()
477 HloInstruction::CreateParameter(0, loop_state_shape, "loop_state")); in BuildDependentBodyComputation2()
[all …]
|
| D | hlo_dataflow_analysis_test.cc | 86 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in CreateR0F32UnaryOpComputation()
148 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
150 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
256 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
258 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
306 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
308 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
361 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
363 HloInstruction::CreateParameter(1, scalar_shape_, "param1")); in TEST_P()
415 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_P()
[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 | 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()
279 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
341 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
355 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
363 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
415 HloInstruction::CreateParameter(0, scalar_shape_, "param0")); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 51 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
53 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
70 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
72 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
89 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
91 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
109 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
111 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
131 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
133 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter_test.cc | 101 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
104 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
142 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
145 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
172 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
175 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
202 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
205 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
231 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
234 builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
[all …]
|
| D | gpu_layout_assignment_test.cc | 59 HloInstruction::CreateParameter(0, ashape, "x")); in TEST_F()
61 HloInstruction::CreateParameter(1, ashape, "y")); in TEST_F()
130 HloInstruction::CreateParameter(0, shape, "operand")); in TEST_F()
132 HloInstruction::CreateParameter(1, aux_shape, "scale")); in TEST_F()
134 HloInstruction::CreateParameter(2, aux_shape, "offset")); in TEST_F()
136 HloInstruction::CreateParameter(3, aux_shape, "mean")); in TEST_F()
138 HloInstruction::CreateParameter(4, aux_shape, "variance")); in TEST_F()
204 HloInstruction::CreateParameter(0, shape, "operand")); in TEST_F()
206 HloInstruction::CreateParameter(1, offset_scale_shape, "scale")); in TEST_F()
208 HloInstruction::CreateParameter(2, offset_scale_shape, "offset")); 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 | 46 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in TEST_F()
68 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x")); in TEST_F()
104 auto lhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
106 auto rhs = embedded_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()
117 HloInstruction::CreateParameter(0, param_shape, "x")); in TEST_F()
|