Searched refs:unaligned_shaped (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 403 return unaligned_shaped<T, 1>({NumElements()}); in unaligned_flat()
442 typename TTypes<T, NDIMS>::UnalignedTensor unaligned_shaped(
492 return unaligned_shaped<T, 1>({NumElements()}); in unaligned_flat()
509 typename TTypes<T, NDIMS>::UnalignedConstTensor unaligned_shaped(
830 typename TTypes<T, NDIMS>::UnalignedTensor Tensor::unaligned_shaped( in unaligned_shaped() function
858 typename TTypes<T, NDIMS>::UnalignedConstTensor Tensor::unaligned_shaped( in unaligned_shaped() function
|
| D | tensor_test.cc | 420 &Tensor::unaligned_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
422 &Tensor::unaligned_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
473 EXPECT_DEATH((t.unaligned_shaped<float, 0>({})), "1 vs. 120"); in TEST_F()
474 EXPECT_DEATH((t.unaligned_shaped<float, 1>({119})), "119 vs. 120"); in TEST_F()
475 EXPECT_DEATH((t.unaligned_shaped<float, 4>({2, 3, 4, 6})), "144 vs. 120"); in TEST_F()
|