HomeSort by relevance Sort by last modified time
    Searched refs:merged_dims (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_2d.h 151 Eigen::DSizes<IndexType, 3> merged_dims; local
152 merged_dims[0] = in.dimension(0); // spatial dimensions
154 merged_dims[0] *= in.dimension(i);
156 merged_dims[1] = in.dimension(NDIMS - 2); // input filters
157 merged_dims[2] = in.dimension(NDIMS - 1); // output filters
166 out.device(d) = in.reshape(merged_dims)
178 Eigen::DSizes<IndexType, 3> merged_dims; local
184 merged_dims[0] = in.dimension(0);
185 merged_dims[1] = in.dimension(1);
186 merged_dims[2] = in.dimension(2) * in.dimension(3)
    [all...]
  /external/tensorflow/tensorflow/core/framework/
shape_inference_test.cc 488 auto merged_dims = c.MergedDims(); local
489 ASSERT_EQ(3, merged_dims.size());
490 EXPECT_TRUE(merged_dims[0].first.SameHandle(d2));
491 EXPECT_TRUE(merged_dims[0].second.SameHandle(d_unknown));
492 EXPECT_TRUE(merged_dims[1].first.SameHandle(d_unknown));
493 EXPECT_TRUE(merged_dims[1].second.SameHandle(d2));
494 EXPECT_TRUE(merged_dims[2].first.SameHandle(d_unknown));
495 EXPECT_TRUE(merged_dims[2].second.SameHandle(d_unknown_b));
503 merged_dims = c.MergedDims();
504 EXPECT_EQ(3, merged_dims.size())
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc     [all...]

Completed in 257 milliseconds