Home | History | Annotate | Download | only in kernels

Lines Matching defs:val

138     Tensor val;
139 TF_EXPECT_OK(reader.Lookup("tensor_bool", &val));
140 EXPECT_EQ(DT_BOOL, val.dtype());
142 EXPECT_EQ((i != 0), val.template flat<bool>()(i));
154 Tensor val;
155 TF_EXPECT_OK(reader.Lookup("tensor_int", &val));
156 EXPECT_EQ(DT_INT32, val.dtype());
158 EXPECT_EQ(i + 1, val.template flat<int>()(i));
170 Tensor val;
171 TF_EXPECT_OK(reader.Lookup("tensor_float", &val));
172 EXPECT_EQ(DT_FLOAT, val.dtype());
174 EXPECT_EQ(static_cast<float>(i) / 10, val.template flat<float>()(i));
186 Tensor val;
187 TF_EXPECT_OK(reader.Lookup("tensor_double", &val));
188 EXPECT_EQ(DT_DOUBLE, val.dtype());
190 EXPECT_EQ(static_cast<double>(i) / 20, val.template flat<double>()(i));
202 Tensor val;
203 TF_EXPECT_OK(reader.Lookup("tensor_qint8", &val));
204 EXPECT_EQ(DT_QINT8, val.dtype());
206 EXPECT_EQ(*reinterpret_cast<qint8*>(&i), val.template flat<qint8>()(i));
218 Tensor val;
219 TF_EXPECT_OK(reader.Lookup("tensor_qint32", &val));
220 EXPECT_EQ(DT_QINT32, val.dtype());
223 val.template flat<qint32>()(i));
235 Tensor val;
236 TF_EXPECT_OK(reader.Lookup("tensor_uint8", &val));
237 EXPECT_EQ(DT_UINT8, val.dtype());
239 EXPECT_EQ(i + 1, val.template flat<uint8>()(i));
251 Tensor val;
252 TF_EXPECT_OK(reader.Lookup("tensor_int8", &val));
253 EXPECT_EQ(DT_INT8, val.dtype());
255 EXPECT_EQ(i - 7, val.template flat<int8>()(i));
267 Tensor val;
268 TF_EXPECT_OK(reader.Lookup("tensor_int16", &val));
269 EXPECT_EQ(DT_INT16, val.dtype());
271 EXPECT_EQ(i - 8, val.template flat<int16>()(i));
283 Tensor val;
284 TF_EXPECT_OK(reader.Lookup("tensor_int64", &val));
285 EXPECT_EQ(DT_INT64, val.dtype());
287 EXPECT_EQ(i - 9, val.template flat<int64>()(i));
299 Tensor val;
300 TF_EXPECT_OK(reader.Lookup("tensor_complex64", &val));
301 EXPECT_EQ(DT_COMPLEX64, val.dtype());
303 EXPECT_EQ(100 + i, val.template flat<complex64>()(i).real());
304 EXPECT_EQ(200 + i, val.template flat<complex64>()(i).imag());
316 Tensor val;
317 TF_EXPECT_OK(reader.Lookup("tensor_complex128", &val));
318 EXPECT_EQ(DT_COMPLEX128, val.dtype());
320 EXPECT_EQ(100 + i, val.template flat<complex128>()(i).real());
321 EXPECT_EQ(200 + i, val.template flat<complex128>()(i).imag());
332 Tensor val;
333 TF_EXPECT_OK(reader.Lookup("tensor_half", &val));
334 EXPECT_EQ(DT_HALF, val.dtype());
337 val.template flat<Eigen::half>()(i));