Searched defs:flat_inner_dims (Results 1 – 2 of 2) sorted by relevance
533 auto flat_inner_dims = t.flat_inner_dims<float>(); in TEST_F() local540 auto flat_inner_dims = t.flat_inner_dims<float, 3>(); in TEST_F() local548 auto flat_inner_dims = t.flat_inner_dims<float, 5>(); in TEST_F() local558 auto flat_inner_dims = zero_t.flat_inner_dims<float>(); in TEST_F() local563 auto flat_inner_dims = zero_t.flat_inner_dims<float, 3>(); in TEST_F() local569 auto flat_inner_dims = zero_t.flat_inner_dims<float, 5>(); in TEST_F() local
952 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_dims() { in flat_inner_dims() function969 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_dims() const { in flat_inner_dims() function