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
|