Searched defs:flat_inner_outer_dims (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | tensor_test.cc | 595 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 4>(0); in TEST_F() local 604 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(-2); in TEST_F() local 615 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(0); in TEST_F() local 626 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 8>(-2); in TEST_F() local 639 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(1); in TEST_F() local 647 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(1); in TEST_F() local 657 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 3>(0); in TEST_F() local 665 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 5>(-2); in TEST_F() local 675 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 2>(1); in TEST_F() local 682 auto flat_inner_outer_dims = zero_t.flat_inner_outer_dims<float, 2>(0); in TEST_F() local [all …]
|
D | tensor.h | 738 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_outer_dims(int64 begin) { in flat_inner_outer_dims() function 755 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_outer_dims( in flat_inner_outer_dims() function
|