Searched refs:merged_dims (Results 1 – 3 of 3) sorted by relevance
157 Eigen::DSizes<IndexType, 3> merged_dims;158 merged_dims[0] = in.dimension(0); // spatial dimensions160 merged_dims[0] *= in.dimension(i);162 merged_dims[1] = in.dimension(NDIMS - 2); // input filters163 merged_dims[2] = in.dimension(NDIMS - 1); // output filters175 for (int merged_dim = 0; merged_dim < merged_dims.rank(); ++merged_dim) {188 in.reshape(merged_dims).shuffle(shuffling_perm).reshape(expanded_dims);198 Eigen::DSizes<IndexType, 3> merged_dims;204 merged_dims[0] = in.dimension(0);205 merged_dims[1] = in.dimension(1);[all …]
488 auto merged_dims = c.MergedDims(); in TEST_F() local489 ASSERT_EQ(3, merged_dims.size()); in TEST_F()490 EXPECT_TRUE(merged_dims[0].first.SameHandle(d2)); in TEST_F()491 EXPECT_TRUE(merged_dims[0].second.SameHandle(d_unknown)); in TEST_F()492 EXPECT_TRUE(merged_dims[1].first.SameHandle(d_unknown)); in TEST_F()493 EXPECT_TRUE(merged_dims[1].second.SameHandle(d2)); in TEST_F()494 EXPECT_TRUE(merged_dims[2].first.SameHandle(d_unknown)); in TEST_F()495 EXPECT_TRUE(merged_dims[2].second.SameHandle(d_unknown_b)); in TEST_F()503 merged_dims = c.MergedDims(); in TEST_F()504 EXPECT_EQ(3, merged_dims.size()); in TEST_F()[all …]
2161 for (const auto& merged_dims : node_ctx->MergedDims()) { in InferStatically() local2162 if (!shape_manager->Merge(merged_dims.first, merged_dims.second).ok()) { in InferStatically()