OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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