Searched defs:flat_inner_outer_dims (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | tensor_test.cc | 596 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 4>(0); in TEST_F() local 605 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(-2); in TEST_F() local 616 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(0); in TEST_F() local 627 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 8>(-2); in TEST_F() local 640 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(1); in TEST_F() local 648 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(1); in TEST_F() local 658 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(0); in TEST_F() local 666 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(-2); in TEST_F() local 676 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 2>(1); in TEST_F() local 683 auto flat_inner_outer_dims = zero_t.flat_inner_outer_dims<float, 2>(0); in TEST_F() local [all …]
|
D | tensor.h | 889 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_outer_dims(int64 begin) { in flat_inner_outer_dims() function 906 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_outer_dims( in flat_inner_outer_dims() function
|