Searched defs:flat_outer_dims (Results 1 – 2 of 2) sorted by relevance
548 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local555 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local563 auto flat_outer_dims = t.flat_outer_dims<float, 5>(); in TEST_F() local573 auto flat_outer_dims = zero_t.flat_outer_dims<float>(); in TEST_F() local578 auto flat_outer_dims = zero_t.flat_outer_dims<float, 3>(); in TEST_F() local584 auto flat_outer_dims = zero_t.flat_outer_dims<float, 5>(); in TEST_F() local
884 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function901 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function