Lines Matching refs:builder
52 ComputationBuilder builder(client_, TestName());
53 auto x = builder.ConstantR1<float>(
55 auto exp = builder.Exp(x);
61 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
66 ComputationBuilder builder(client_, TestName());
72 auto x = builder.ConstantR1<float>(exponents);
73 auto exp = builder.Exp(x);
81 ComputeAndCompareR1<float>(&builder, expected, {},
87 ComputationBuilder builder(client_, TestName());
101 auto x = builder.ConstantR4FromArray4D<float>(exponents);
102 auto exp = builder.Exp(x);
104 ComputeAndCompareR4<float>(&builder, expected, {},
109 ComputationBuilder builder(client_, TestName());
110 auto x = builder.ConstantR1<float>(
112 builder.Neg(x);
116 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
120 ComputationBuilder builder(client_, TestName());
121 auto x = builder.ConstantR1<int32>({2, -2, 12, -4, 5, 20, -15, 0, -2, 1});
122 builder.Neg(x);
125 ComputeAndCompareR1<int32>(&builder, expected, {});
129 ComputationBuilder builder(client_, TestName());
130 auto x = builder.ConstantR1<uint32>(
132 builder.Neg(x);
135 ComputeAndCompareR1<uint32>(&builder, expected, {});
139 ComputationBuilder builder(client_, TestName());
140 auto x = builder.ConstantR1<float>(
142 builder.SquareF32(x);
146 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
150 ComputationBuilder builder(client_, TestName());
151 auto x = builder.ConstantR1<float>(
153 builder.ReciprocalF32(x);
158 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
162 ComputationBuilder builder(client_, TestName());
163 auto x = builder.ConstantR1<float>({0.0, -0.0});
164 auto exp = builder.SqrtF32(x);
166 ComputeAndCompareR1<float>(&builder, {0, 0}, {}, error_spec_);
170 ComputationBuilder builder(client_, TestName());
171 auto x = builder.ConstantR1<float>({16.0, 1.0, 1024.0, 0.16, 0.2, 12345});
172 auto exp = builder.SqrtF32(x);
175 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
179 ComputationBuilder builder(client_, TestName());
181 builder.ConstantR1<float>({16.0, 1.0, 1024.0, 0.16, 0.2, 12345, 1.2345});
182 auto exp = builder.Pow(x, builder.ConstantR0<float>(-.5f));
187 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
191 ComputationBuilder builder(client_, TestName());
192 auto add = CreateScalarAddComputation(F32, &builder);
194 auto x = builder.ConstantR1<float>(
196 auto y = builder.ConstantR1<float>(
198 auto max = builder.Map({x, y}, add, {0});
202 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_);
206 ComputationBuilder builder(client_, TestName());
207 auto x = builder.ConstantR1<float>(
209 auto y = builder.ConstantR1<float>(
211 auto max = builder.Max(x, y);
215 ComputeAndCompareR1<float>(&builder, expected, {});
221 ComputationBuilder builder(client_, TestName());
225 /*builder=*/&builder, /*data_handle=*/&v1);
228 /*builder=*/&builder, /*data_handle=*/&v2);
230 auto max = builder.Max(v1, v2);
231 ComputeAndCompareR1<float>(&builder, {41.0f, 22.0f, 23.0f, 84.0f},
239 ComputationBuilder builder(client_, TestName());
265 /*builder=*/&builder, /*data_handle=*/&v1);
268 /*builder=*/&builder, /*data_handle=*/&v2);
270 auto max = builder.Max(v1, v2);
271 ComputeAndCompareR1<float>(&builder, expected_vec,
277 ComputationBuilder builder(client_, TestName());
278 auto x = builder.ConstantR1<float>(
280 auto y = builder.ConstantR0<float>(0);
281 auto max = builder.Max(x, y);
285 ComputeAndCompareR1<float>(&builder, expected, {});
289 ComputationBuilder builder(client_, TestName());
290 auto x = builder.ConstantR1<float>(
292 auto y = builder.ConstantR1<float>(
294 auto min = builder.Min(x, y);
298 ComputeAndCompareR1<float>(&builder, expected, {});
302 ComputationBuilder builder(client_, TestName());
303 auto zero = builder.ConstantR0<float>(0);
304 auto one = builder.ConstantR0<float>(1);
305 auto x = builder.ConstantR1<float>(
307 auto clamp = builder.Min(builder.Max(x, zero), one);
311 ComputeAndCompareR1<float>(&builder, expected, {});
315 ComputationBuilder builder(client_, TestName());
316 auto zero = builder.ConstantR0<float>(0);
317 auto one = builder.ConstantR0<float>(1);
318 auto x = builder.ConstantR1<float>(
320 auto clamp = builder.Clamp(zero, x, one);
324 ComputeAndCompareR1<float>(&builder, expected, {});
328 ComputationBuilder builder(client_, TestName());
329 auto zero = builder.ConstantR1<float>({0.0f, 0.0f});
330 auto one = builder.ConstantR1<float>({1.0f, 1.0f});
331 auto x = builder.ConstantR1<float>({2.1, -2.6});
332 auto clamp = builder.Clamp(zero, x, one);
335 ComputeAndCompareR1<float>(&builder, expected, {});
339 ComputationBuilder builder(client_, TestName());
340 auto one = builder.ConstantR0<float>(1);
341 auto two = builder.ConstantR0<float>(2);
342 auto x = builder.ConstantR1<float>(
344 auto clamp = builder.Clamp(one, x, two);
348 ComputeAndCompareR1<float>(&builder, expected, {});
355 ComputationBuilder builder(client_, "add_half");
357 builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x_value");
358 auto half = builder.ConstantR0<float>(0.5);
359 builder.Add(x_value, half);
360 auto computation_status = builder.Build();
368 ComputationBuilder builder(client_, "clamp");
370 builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "y_value");
371 auto zero = builder.ConstantR0<float>(0.0);
372 auto clamped = builder.Clamp(zero, y_value, builder.ConstantR0<float>(5));
373 auto computation_status = builder.Build();
381 ComputationBuilder builder(client_, "mult_relu_add");
383 builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "z_value");
384 auto zero = builder.ConstantR0<float>(0.0);
385 auto two = builder.ConstantR0<float>(2.0);
386 auto max = builder.Max(z_value, zero);
387 auto mult = builder.Mul(two, max);
388 auto inner = builder.Map({mult}, add_half, {});
389 builder.Map({inner}, clamp, {});
390 auto computation_status = builder.Build();
395 ComputationBuilder builder(client_, "map10");
397 auto x = builder.ConstantR1<float>(
399 auto activations = builder.Map({x}, mult_relu_add, {0});
404 ComputeAndCompareR1<float>(&builder, expected, {});
408 ComputationBuilder builder(client_, TestName());
409 auto x = builder.ConstantR1<int32>({-5, -4, -3, -2, -1, 0, 1, 2, 3, 4});
410 auto y = builder.ConstantR0<int32>(3);
411 builder.Rem(x, y);
414 ComputeAndCompareR1<int32>(&builder, expected, {});
418 ComputationBuilder builder(client_, TestName());
419 auto x = builder.ConstantR1<bool>({false, true});
420 auto y = builder.ConstantR1<bool>({true, false});
421 builder.Eq(x, y);
424 ComputeAndCompareR1<bool>(&builder, expected, {});
428 ComputationBuilder builder(client_, TestName());
429 auto x = builder.ConstantR1<bool>({false, true});
430 auto y = builder.ConstantR1<bool>({true, false});
431 builder
434 ComputeAndCompareR1<bool>(&builder, expected, {});
