| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander_test.cc | 49 HloInstruction::CreateParameter(0, input_shape, "activiation"));
52 HloInstruction::CreateParameter(1, scale_shape, "scale"));
55 HloInstruction::CreateParameter(2, offset_shape, "offset"));
85 HloInstruction::CreateParameter(0, input_shape, "activation"));
88 HloInstruction::CreateParameter(1, scale_shape, "scale"));
91 HloInstruction::CreateParameter(2, mean_shape, "mean"));
94 HloInstruction::CreateParameter(3, var_shape, "var"));
97 HloInstruction::CreateParameter(4, grad_output_shape, "grad_output"));
|
| hlo_scheduling_test.cc | 44 HloInstruction::CreateParameter(0, tuple_shape, "cond_param"));
59 HloInstruction::CreateParameter(0, tuple_shape, "body_param"));
66 HloInstruction::CreateParameter(0, scalar_shape, "param_iter"));
68 HloInstruction::CreateParameter(1, scalar_shape, "param_data"));
|
| bfloat16_normalization_test.cc | 83 HloInstruction::CreateParameter(0, f32_shape, "a"));
85 HloInstruction::CreateParameter(1, bf16_shape, "b"));
87 HloInstruction::CreateParameter(2, f32_shape, "c"));
111 HloInstruction::CreateParameter(0, f32_shape, "a"));
113 HloInstruction::CreateParameter(1, bf16_shape, "b"));
115 HloInstruction::CreateParameter(2, f32_shape, "c"));
141 HloInstruction::CreateParameter(0, f32_shape, "a"));
143 HloInstruction::CreateParameter(1, bf16_shape, "b"));
145 HloInstruction::CreateParameter(2, f32_shape, "c"));
173 HloInstruction::CreateParameter(0, bf16_scalar_shape, "param0"))
[all...] |
| bfloat16_conversion_folding_test.cc | 82 HloInstruction::CreateParameter(0, f32_shape, "a"));
84 HloInstruction::CreateParameter(1, f32_shape, "b"));
86 HloInstruction::CreateParameter(2, f32_shape, "c"));
116 HloInstruction::CreateParameter(0, f32_shape, "a"));
118 HloInstruction::CreateParameter(1, f32_shape, "b"));
120 HloInstruction::CreateParameter(2, f32_shape, "c"));
151 HloInstruction::CreateParameter(0, f32_shape, "a"));
153 HloInstruction::CreateParameter(1, f32_shape, "b"));
155 HloInstruction::CreateParameter(2, f32_shape, "c"));
186 HloInstruction::CreateParameter(0, f32_shape, "a"))
[all...] |
| tuple_simplifier_test.cc | 56 HloInstruction::CreateParameter(0, scalar_shape_, "param0"));
58 HloInstruction::CreateParameter(1, scalar_shape_, "param1"));
60 HloInstruction::CreateParameter(2, scalar_shape_, "param2"));
72 HloInstruction::CreateParameter(0, tuple_shape_, "param"));
85 HloInstruction::CreateParameter(0, scalar_shape_, "param0"));
87 HloInstruction::CreateParameter(1, scalar_shape_, "param1"));
89 HloInstruction::CreateParameter(2, scalar_shape_, "param2"));
109 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
139 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
169 HloInstruction::CreateParameter(0, tuple_shape_, "param"))
[all...] |
| defuser_test.cc | 48 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
50 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
63 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
65 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
86 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
88 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
111 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"));
113 builder.AddInstruction(HloInstruction::CreateParameter(1, shape_, "p1"));
115 builder.AddInstruction(HloInstruction::CreateParameter(2, shape_, "p2"));
149 builder.AddInstruction(HloInstruction::CreateParameter(0, shape_, "p0"))
[all...] |
| implicit_broadcast_remover_test.cc | 38 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "p0"));
40 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "p1"));
56 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
59 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "p1"));
83 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "p0"));
84 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
104 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
107 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "p1"));
126 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
128 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
[all...] |
| hlo_dce_test.cc | 97 auto live_param = builder.AddInstruction(HloInstruction::CreateParameter(
99 auto dead_param1 = builder.AddInstruction(HloInstruction::CreateParameter(
101 builder.AddInstruction(HloInstruction::CreateParameter(
184 HloInstruction::CreateParameter(0, shape, "param"));
194 HloInstruction::CreateParameter(0, shape, "param"));
224 HloInstruction::CreateParameter(0, shape, "cond_param"));
236 HloInstruction::CreateParameter(0, shape, "param"));
248 HloInstruction::CreateParameter(0, shape, "param"));
280 HloInstruction::CreateParameter(0, shape, "param"));
291 HloInstruction::CreateParameter(0, shape, "param"))
[all...] |
| reshape_mover_test.cc | 42 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
44 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
105 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
107 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
131 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
133 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
181 auto param1 = builder.AddInstruction(HloInstruction::CreateParameter(
186 auto param2 = builder.AddInstruction(HloInstruction::CreateParameter(
221 auto param0 = builder.AddInstruction(HloInstruction::CreateParameter(
226 HloInstruction::CreateParameter(1, root_shape, "param1"))
[all...] |
| graphviz_example.cc | 47 auto x_value = builder.AddInstruction(HloInstruction::CreateParameter(
61 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "lhs"));
63 HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {}), "rhs"));
75 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "arg"));
96 builder.AddInstruction(HloInstruction::CreateParameter(0, vshape, "foo"));
98 builder.AddInstruction(HloInstruction::CreateParameter(1, vshape, "bar"));
100 builder.AddInstruction(HloInstruction::CreateParameter(2, vshape, "baz"));
102 builder.AddInstruction(HloInstruction::CreateParameter(3, sshape, "qux"));
104 builder.AddInstruction(HloInstruction::CreateParameter(4, mshape, "zzz"));
|
| 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"));
54 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x"));
56 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "y"));
155 HloInstruction::CreateParameter(0, r1s32_5_, "param1"));
157 HloInstruction::CreateParameter(1, r1s32_5_, "param2"));
191 builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32_, "x"));
193 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32_, "y"));
|
| hlo_tfgraph_builder_test.cc | 37 HloInstruction::CreateParameter(0, r0f32_, "param0"));
48 HloInstruction::CreateParameter(0, r0f32_, "param0"));
67 HloInstruction::CreateParameter(0, shape, "param0"));
69 HloInstruction::CreateParameter(1, shape, "param1"));
125 HloInstruction::CreateParameter(0, r0f32_, "param0"));
127 HloInstruction::CreateParameter(1, r0f32_, "param1"));
143 HloInstruction::CreateParameter(0, r0f32_, "param0"));
145 HloInstruction::CreateParameter(1, r0f32_, "param1"));
170 HloInstruction::CreateParameter(0, r0f32_, "param0"));
|
| instruction_fusion_test.cc | 28 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(S32, {}), "0"));
44 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(S32, {}), "0"));
60 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(S32, {}), "0"));
77 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "0"));
79 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "1"));
99 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "0"));
120 HloInstruction::CreateParameter(0, small_shape, "0"));
122 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "1"));
|
| algebraic_simplifier_test.cc | 58 HloInstruction::CreateParameter(0, r0f32, "param0"));
79 HloInstruction::CreateParameter(0, r0f32, "param0"));
100 HloInstruction::CreateParameter(0, r0f32, "param0"));
125 HloInstruction::CreateParameter(0, r2f32, "param0"));
147 HloInstruction::CreateParameter(0, r2f32, "param0"));
170 HloInstruction::CreateParameter(0, r0f32, "param0"));
191 HloInstruction::CreateParameter(0, r0f32, "param0"));
212 HloInstruction::CreateParameter(0, r0f32, "param0"));
214 HloInstruction::CreateParameter(1, r0f32, "param1"));
216 HloInstruction::CreateParameter(2, r0f32, "param2"))
[all...] |
| hlo_verifier_test.cc | 42 HloInstruction::CreateParameter(0, scalar_shape, "param"));
61 HloInstruction::CreateParameter(0, scalar_shape, "param"));
80 HloInstruction::CreateParameter(0, scalar_shape, "param"));
88 HloInstruction::CreateParameter(0, scalar_shape, "param"));
106 builder.AddInstruction(HloInstruction::CreateParameter(0, s1, "param"));
|
| inliner_test.cc | 46 HloInstruction::CreateParameter(0, r0f32, "x"));
48 HloInstruction::CreateParameter(1, r0f32, "y"));
83 HloInstruction::CreateParameter(0, r0f32, "x"));
118 HloInstruction::CreateParameter(1, r0f32, "x"));
120 HloInstruction::CreateParameter(0, r0f32, "y"));
|
| while_loop_invariant_code_motion_test.cc | 53 HloInstruction::CreateParameter(0, param_shape, "param"));
67 HloInstruction::CreateParameter(0, while_shape, "param"));
83 HloInstruction::CreateParameter(0, while_shape, "init_value"));
109 HloInstruction::CreateParameter(0, while_shape, "param"));
142 HloInstruction::CreateParameter(0, while_shape, "init_value"));
181 HloInstruction::CreateParameter(0, while_shape, "param"));
196 HloInstruction::CreateParameter(0, while_shape, "init_value"));
219 HloInstruction::CreateParameter(0, while_shape, "param"));
235 HloInstruction::CreateParameter(0, while_shape, "init_value"));
255 HloInstruction::CreateParameter(0, while_shape, "param"))
[all...] |
| reduce_precision_insertion_test.cc | 53 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
81 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
83 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b"));
112 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
141 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
177 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
206 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "a"));
211 builder.AddInstruction(HloInstruction::CreateParameter(1, shape, "b"));
241 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x"));
267 builder.AddInstruction(HloInstruction::CreateParameter(0, shape, "x"))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker_test.cc | 40 HloInstruction::CreateParameter(0, scalar_shape, "param0"));
42 HloInstruction::CreateParameter(1, scalar_shape, "param1"));
55 HloInstruction::CreateParameter(0, sparse_shape, "param0"));
57 HloInstruction::CreateParameter(1, sparse_shape, "param1"));
|
| cpu_instruction_fusion_test.cc | 44 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
46 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
63 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
65 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
82 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
84 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
103 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
105 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
125 HloInstruction* arg0 = builder.AddInstruction(HloInstruction::CreateParameter(
127 HloInstruction* arg1 = builder.AddInstruction(HloInstruction::CreateParameter(
[all...] |
| cpu_copy_insertion_test.cc | 71 HloInstruction::CreateParameter(0, scalar_shape_, "param_0"));
75 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
82 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
109 HloInstruction::CreateParameter(0, scalar_shape_, "param_0"));
115 HloInstruction::CreateParameter(0, scalar_shape_, "param"));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_support_checker_test.cc | 40 HloInstruction::CreateParameter(0, scalar_shape, "param0"));
42 HloInstruction::CreateParameter(1, scalar_shape, "param1"));
55 HloInstruction::CreateParameter(0, sparse_shape, "param0"));
57 HloInstruction::CreateParameter(1, sparse_shape, "param1"));
|
| gpu_layout_assignment_test.cc | 54 HloInstruction::CreateParameter(0, ashape, "x"));
56 HloInstruction::CreateParameter(1, ashape, "y"));
123 HloInstruction::CreateParameter(0, shape, "operand"));
125 HloInstruction::CreateParameter(1, aux_shape, "scale"));
127 HloInstruction::CreateParameter(2, aux_shape, "offset"));
129 HloInstruction::CreateParameter(3, aux_shape, "mean"));
131 HloInstruction::CreateParameter(4, aux_shape, "variance"));
195 HloInstruction::CreateParameter(0, shape, "operand"));
197 HloInstruction::CreateParameter(1, offset_scale_shape, "scale"));
199 HloInstruction::CreateParameter(2, offset_scale_shape, "offset"))
[all...] |
| cudnn_convolution_rewriter_test.cc | 98 builder.AddInstruction(HloInstruction::CreateParameter(
101 builder.AddInstruction(HloInstruction::CreateParameter(
127 builder.AddInstruction(HloInstruction::CreateParameter(
130 builder.AddInstruction(HloInstruction::CreateParameter(
156 builder.AddInstruction(HloInstruction::CreateParameter(
159 builder.AddInstruction(HloInstruction::CreateParameter(
186 builder.AddInstruction(HloInstruction::CreateParameter(
189 builder.AddInstruction(HloInstruction::CreateParameter(
214 builder.AddInstruction(HloInstruction::CreateParameter(
217 builder.AddInstruction(HloInstruction::CreateParameter(
[all...] |
| 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(
58 HloInstruction* x = builder.AddInstruction(HloInstruction::CreateParameter(
60 HloInstruction* y = builder.AddInstruction(HloInstruction::CreateParameter(
91 params.push_back(builder.AddInstruction(HloInstruction::CreateParameter(
|
