OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:GetOutputShape
(Results
1 - 12
of
12
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/kernels/
exp_test.cc
45
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
57
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3, 1, 2}));
reshape_test.cc
43
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
71
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2, 2, 2}));
79
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2, 1, 4}));
space_to_depth_test.cc
46
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
64
EXPECT_THAT(m.
GetOutputShape
(), ElementsAre(1, 1, 1, 8));
72
EXPECT_THAT(m.
GetOutputShape
(), ElementsAre(1, 1, 1, 4));
81
EXPECT_THAT(m.
GetOutputShape
(), ElementsAre(1, 1, 1, 12));
91
EXPECT_THAT(m.
GetOutputShape
(), ElementsAre(1, 2, 2, 4));
batch_to_space_nd_test.cc
42
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
101
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 4, 1}));
112
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 4, 1}));
gather_test.cc
62
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
85
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
96
EXPECT_TRUE(m.
GetOutputShape
().empty());
131
ASSERT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
mean_test.cc
40
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
85
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
97
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 3, 1}));
113
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
128
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 3, 1}));
pad_test.cc
37
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
113
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 4, 1}));
123
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 4, 1}));
133
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 7, 1}));
144
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 4, 7, 1}));
space_to_batch_nd_test.cc
42
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
106
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({4, 2, 2, 1}));
117
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({4, 2, 2, 1}));
126
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({8, 1, 2, 1}));
137
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({8, 1, 2, 1}));
146
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({6, 2, 2, 1}));
159
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({6, 2, 2, 1}));
170
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({6, 2, 4, 1}));
184
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({6, 2, 4, 1}));
split_test.cc
58
std::vector<int>
GetOutputShape
(int i) { return GetTensorShape(outputs_[i]); }
84
EXPECT_THAT(m.
GetOutputShape
(i), ElementsAreArray(output_shape))
94
EXPECT_THAT(const_m.
GetOutputShape
(i), ElementsAreArray(output_shape))
squeeze_test.cc
56
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
67
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({24}));
83
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 24}));
99
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({24}));
113
EXPECT_THAT(m.
GetOutputShape
(), IsEmpty());
strided_slice_test.cc
62
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
92
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
103
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({0}));
113
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2}));
124
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3}));
135
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1}));
146
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3}));
157
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3}));
168
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1}));
179
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1}))
[
all
...]
transpose_test.cc
139
std::vector<int>
GetOutputShape
() { return GetTensorShape(output_); }
202
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3}));
211
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3}));
219
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2, 3}));
228
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({2, 3}));
237
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({4, 2, 3}));
249
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({4, 2, 3}));
264
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 2, 3, 1}));
273
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({1, 2, 3, 1}));
281
EXPECT_THAT(m.
GetOutputShape
(), ElementsAreArray({3, 2, 1, 1}))
[
all
...]
Completed in 330 milliseconds