Searched refs:merged_dims (Results 1 – 3 of 3) sorted by relevance
218 Eigen::DSizes<IndexType, 3> merged_dims;219 merged_dims[0] = spatial_dims.TotalSize(); // product of spatial dims [H*W]220 merged_dims[1] = in.dimension(NDIMS - 2); // input filters [I]221 merged_dims[2] = in.dimension(NDIMS - 1); // output filters [O]231 expanded_dims[0] = merged_dims[2]; // [O]232 expanded_dims[1] = merged_dims[1]; // [I]240 expanded_dims[0] = merged_dims[2]; // [O]241 expanded_dims[NDIMS - 1] = merged_dims[1]; // [I]252 in.reshape(merged_dims).shuffle(shuffling_perm).reshape(expanded_dims);264 Eigen::DSizes<IndexType, 3> merged_dims;[all …]
503 auto merged_dims = c.MergedDims(); in TEST_F() local504 ASSERT_EQ(3, merged_dims.size()); in TEST_F()505 EXPECT_TRUE(merged_dims[0].first.SameHandle(d2)); in TEST_F()506 EXPECT_TRUE(merged_dims[0].second.SameHandle(d_unknown)); in TEST_F()507 EXPECT_TRUE(merged_dims[1].first.SameHandle(d_unknown)); in TEST_F()508 EXPECT_TRUE(merged_dims[1].second.SameHandle(d2)); in TEST_F()509 EXPECT_TRUE(merged_dims[2].first.SameHandle(d_unknown)); in TEST_F()510 EXPECT_TRUE(merged_dims[2].second.SameHandle(d_unknown_b)); in TEST_F()518 merged_dims = c.MergedDims(); in TEST_F()519 EXPECT_EQ(3, merged_dims.size()); in TEST_F()[all …]
2598 for (const auto& merged_dims : node_ctx->MergedDims()) { in InferStatically() local2599 if (!shape_manager->Merge(merged_dims.first, merged_dims.second).ok()) { in InferStatically()