Searched defs:flat_inner_outer_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 629 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 4>(0); in TEST_F() local
638 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(-2); in TEST_F() local
649 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(0); in TEST_F() local
660 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 8>(-2); in TEST_F() local
673 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(1); in TEST_F() local
681 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(1); in TEST_F() local
691 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(0); in TEST_F() local
699 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(-2); in TEST_F() local
709 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 2>(1); in TEST_F() local
716 auto flat_inner_outer_dims = zero_t.flat_inner_outer_dims<float, 2>(0); in TEST_F() local
[all …]
|
| D | tensor.h | 923 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_outer_dims(int64 begin) { in flat_inner_outer_dims() function
940 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_outer_dims( in flat_inner_outer_dims() function
|