Searched refs:MkTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1310 Tensor MkTensor(DataType dt, const TensorShape& shape, in MkTensor() function
1331 Tensor x = MkTensor<int>(DT_INT32, TensorShape({5}), {1, 2, 3, 4, 0}); in TEST()
1333 x = MkTensor<int>(DT_INT32, TensorShape({2, 2}), {1, 2, 3, 4, 0}); in TEST()
1335 x = MkTensor<int>(DT_INT32, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); in TEST()
1338 x = MkTensor<int>(DT_INT32, TensorShape({0}), {}); in TEST()
1343 Tensor x = MkTensor<int>(DT_INT32, TensorShape({3, 4}), in TEST()
1350 Tensor x = MkTensor<float>(DT_FLOAT, TensorShape({5}), {1, 2, 3, 4, 0}); in TEST()
1352 x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2}), {1, 2, 3, 4, 0}); in TEST()
1354 x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); in TEST()
1357 x = MkTensor<float>(DT_FLOAT, TensorShape({0}), {}); in TEST()
[all …]
|