Searched defs:flat_outer_dims (Results 1 – 2 of 2) sorted by relevance
733 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function750 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function
547 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local554 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local562 auto flat_outer_dims = t.flat_outer_dims<float, 5>(); in TEST_F() local572 auto flat_outer_dims = zero_t.flat_outer_dims<float>(); in TEST_F() local577 auto flat_outer_dims = zero_t.flat_outer_dims<float, 3>(); in TEST_F() local583 auto flat_outer_dims = zero_t.flat_outer_dims<float, 5>(); in TEST_F() local