Searched refs:MkTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1343 Tensor MkTensor(DataType dt, const TensorShape& shape, in MkTensor() function
1364 Tensor x = MkTensor<int>(DT_INT32, TensorShape({5}), {1, 2, 3, 4, 0}); in TEST()
1366 x = MkTensor<int>(DT_INT32, TensorShape({2, 2}), {1, 2, 3, 4, 0}); in TEST()
1368 x = MkTensor<int>(DT_INT32, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); in TEST()
1371 x = MkTensor<int>(DT_INT32, TensorShape({0}), {}); in TEST()
1376 Tensor x = MkTensor<int>(DT_INT32, TensorShape({3, 4}), in TEST()
1383 Tensor x = MkTensor<float>(DT_FLOAT, TensorShape({5}), {1, 2, 3, 4, 0}); in TEST()
1385 x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2}), {1, 2, 3, 4, 0}); in TEST()
1387 x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); in TEST()
1390 x = MkTensor<float>(DT_FLOAT, TensorShape({0}), {}); in TEST()
[all …]
|