Searched refs:unaligned_shaped (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 466 return unaligned_shaped<T, 1>({NumElements()}); in unaligned_flat()
505 typename TTypes<T, NDIMS>::UnalignedTensor unaligned_shaped(
555 return unaligned_shaped<T, 1>({NumElements()}); in unaligned_flat()
572 typename TTypes<T, NDIMS>::UnalignedConstTensor unaligned_shaped(
858 typename TTypes<T, NDIMS>::UnalignedTensor Tensor::unaligned_shaped( in unaligned_shaped() function
886 typename TTypes<T, NDIMS>::UnalignedConstTensor Tensor::unaligned_shaped( in unaligned_shaped() function
|
| D | tensor_test.cc | 451 &Tensor::unaligned_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
453 &Tensor::unaligned_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
506 EXPECT_DEATH((t.unaligned_shaped<float, 0>({})), "1 vs. 120"); in TEST_F()
507 EXPECT_DEATH((t.unaligned_shaped<float, 1>({119})), "119 vs. 120"); in TEST_F()
508 EXPECT_DEATH((t.unaligned_shaped<float, 4>({2, 3, 4, 6})), "144 vs. 120"); in TEST_F()
|