Lines Matching refs:builder
40 ComputationBuilder builder(client_, TestName());
41 auto arg = builder.ConstantR1<T>({});
42 auto abs = builder.Abs(arg);
45 ComputeAndCompareR1<float>(&builder, {}, {});
47 ComputeAndCompareR1<T>(&builder, {}, {});
53 ComputationBuilder builder(client_, TestName());
54 auto arg = builder.ConstantR1<T>({-2, 25, 0, -123, inf<T>(), -inf<T>()});
55 auto abs = builder.Abs(arg);
57 ComputeAndCompareR1<T>(&builder, {2, 25, 0, 123, inf<T>(), inf<T>()}, {});
62 ComputationBuilder builder(client_, TestName());
63 auto arg = builder.ConstantR1<T>(
65 auto sign = builder.Sign(arg);
67 ComputeAndCompareR1<T>(&builder, {-1, 1, 0, 0, -1, 1, -1}, {});
72 ComputationBuilder builder(client_, TestName());
73 auto arg = builder.ConstantR1<T>({-2, 25, 0, -123});
74 auto sign = builder.Sign(arg);
75 auto abs = builder.Abs(arg);
76 builder.Sub(builder.Mul(sign, abs), arg);
78 ComputeAndCompareR1<T>(&builder, {0, 0, 0, 0}, {});
89 ComputationBuilder builder(client_, TestName());
90 auto arg = builder.ConstantR1<complex64>({{-2, 0},
96 auto abs = builder.Abs(arg);
100 ComputeAndCompareLiteral(&builder, *expected, {}, ErrorSpec(1e-6f));
105 ComputationBuilder builder(client_, TestName());
106 auto arg = builder.ConstantR1<complex64>(
108 auto sign = builder.Sign(arg);
112 ComputeAndCompareLiteral(&builder, *expected, {}, ErrorSpec(1e-6f));
117 ComputationBuilder builder(client_, TestName());
119 builder.ConstantR1<complex64>({{-2, 0}, {0, 25}, {0, 0}, {-0.4, 0.3}});
120 auto sign = builder.Sign(arg);
121 auto abs = builder.Abs(arg);
122 builder.Sub(builder.Mul(sign, builder.ConvertElementType(abs, C64)), arg);
126 ComputeAndCompareLiteral(&builder, *expected, {}, ErrorSpec(1e-6f));
142 ComputationBuilder builder(client_, TestName());
143 auto argi = builder.ConstantR0<int>(-5);
144 auto absi = builder.Abs(argi);
145 auto argf = builder.ConstantR0<float>(-3.0f);
146 auto absf = builder.Abs(argf);
147 auto argf0 = builder.ConstantR0<float>(-0.0f);
148 auto absf0 = builder.Abs(argf0);
149 auto argc = builder.ConstantR0<complex64>({-0.3f, 0.4f});
150 auto absc = builder.Abs(argc);
151 builder.Add(builder.Add(absc, absf0),
152 builder.Add(absf, builder.ConvertElementType(absi, F32)));
154 ComputeAndCompareR0<float>(&builder, 8.5f, {});
158 ComputationBuilder builder(client_, TestName());
159 auto argi = builder.ConstantR0<int>(-5);
160 auto sgni = builder.Sign(argi); // -1
161 auto argf = builder.ConstantR0<float>(-4.0f);
162 auto sgnf = builder.Sign(argf); // -1
163 auto argf0 = builder.ConstantR0<float>(-0.0f);
164 auto sgnf0 = builder.Sign(argf0); // 0
165 auto argc = builder.ConstantR0<complex64>({-.3, .4});
166 auto sgnc = builder.Sign(argc); // (-.6, .8)
167 builder.Add(sgnc, builder.ConvertElementType(
168 builder.Add(builder.Add(sgnf0, sgnf),
169 builder.ConvertElementType(sgni, F32)),
174 ComputeAndCompareLiteral(&builder, *expected, {}, ErrorSpec(1e-6f));
190 ComputationBuilder builder(client_, TestName());
191 auto arg = builder.ConstantR1<unsigned int>(
193 auto abs = builder.Abs(arg);
196 &builder, {2, 25, 0, 123, std::numeric_limits<unsigned int>::max()}, {});
200 ComputationBuilder builder(client_, TestName());
201 auto arg = builder.ConstantR1<unsigned int>(
203 auto sign = builder.Sign(arg);
205 ComputeAndCompareR1<unsigned int>(&builder, {1, 1, 0, 1, 1}, {});
209 ComputationBuilder builder(client_, TestName());
210 auto arg = builder.ConstantR2<float>({{1.0, -2.0}, {-3.0, 4.0}});
211 auto sign = builder.Sign(arg);
212 auto abs = builder.Abs(arg);
213 builder.Sub(builder.Mul(sign, abs), arg);
215 ComputeAndCompareR2<float>(&builder, {{0, 0}, {0, 0}}, {});
219 ComputationBuilder builder(client_, TestName());
220 auto lhs = builder.ConstantR1<int32>({0, 1});
221 auto rhs = builder.ConstantR1<int32>({1, 1});
222 builder.ConvertElementType(builder.Eq(lhs, rhs), S32);
224 ComputeAndCompareR1<int32>(&builder, {0, 1}, {});
228 ComputationBuilder builder(client_, TestName());
229 auto lhs = builder.ConstantR1<int32>({0, 1});
230 auto rhs = builder.ConstantR1<int32>({1, 1});
231 builder.ConvertElementType(builder.Eq(lhs, rhs), F32);
233 ComputeAndCompareR1<float>(&builder, {0.0, 1.0}, {});
