HomeSort by relevance Sort by last modified time
    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