Searched defs:flat_inner_dims (Results 1 – 2 of 2) sorted by relevance
500 auto flat_inner_dims = t.flat_inner_dims<float>(); in TEST_F() local507 auto flat_inner_dims = t.flat_inner_dims<float, 3>(); in TEST_F() local515 auto flat_inner_dims = t.flat_inner_dims<float, 5>(); in TEST_F() local525 auto flat_inner_dims = zero_t.flat_inner_dims<float>(); in TEST_F() local530 auto flat_inner_dims = zero_t.flat_inner_dims<float, 3>(); in TEST_F() local536 auto flat_inner_dims = zero_t.flat_inner_dims<float, 5>(); in TEST_F() local
879 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_dims() { in flat_inner_dims() function896 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_dims() const { in flat_inner_dims() function