Searched defs:flat_outer_dims (Results 1 – 2 of 2) sorted by relevance
581 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local588 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local596 auto flat_outer_dims = t.flat_outer_dims<float, 5>(); in TEST_F() local606 auto flat_outer_dims = zero_t.flat_outer_dims<float>(); in TEST_F() local611 auto flat_outer_dims = zero_t.flat_outer_dims<float, 3>(); in TEST_F() local617 auto flat_outer_dims = zero_t.flat_outer_dims<float, 5>(); in TEST_F() local
918 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function935 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function