Home
last modified time | relevance | path

Searched refs:GetOutputShape (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dstrided_slice_test.cc62 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5c572b230111::StridedSliceOpModel
92 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
103 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({0})); in TEST()
113 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
124 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
135 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
146 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
157 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
168 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
179 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1})); in TEST()
[all …]
Dspace_to_batch_nd_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon09a4ca450111::SpaceToBatchNDOpModel
106 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
117 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 2, 1})); in TEST()
126 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
137 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({8, 1, 2, 1})); in TEST()
146 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
159 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 2, 1})); in TEST()
170 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
184 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({6, 2, 4, 1})); in TEST()
Dtranspose_test.cc139 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon940c88840111::TransposeOpModel
202 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
211 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); in TEST()
219 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
228 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); in TEST()
237 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
249 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})); in TEST()
264 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST()
273 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 2, 3, 1})); in TEST()
281 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 2, 1, 1})); in TEST()
[all …]
Dspace_to_depth_test.cc46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon9fabf0e70111::SpaceToDepthOpModel
64 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 8)); in TEST()
72 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 4)); in TEST()
81 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 1, 1, 12)); in TEST()
91 EXPECT_THAT(m.GetOutputShape(), ElementsAre(1, 2, 2, 4)); in TEST()
Dsqueeze_test.cc56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anondda50d070111::FloatSqueezeOpModel
67 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24})); in TEST()
83 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 24})); in TEST()
99 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24})); in TEST()
113 EXPECT_THAT(m.GetOutputShape(), IsEmpty()); in TEST()
Dpad_test.cc37 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anond79f939a0111::PadOpModel
113 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
123 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
133 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
144 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 7, 1})); in TEST()
Dmean_test.cc40 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1603f8660111::BaseMeanOpModel
85 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
97 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1})); in TEST()
113 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
128 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 3, 1})); in TEST()
Dgather_test.cc62 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5c6b54800111::GatherOpModel
85 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
96 EXPECT_TRUE(m.GetOutputShape().empty()); in TEST()
131 ASSERT_THAT(m.GetOutputShape(), ElementsAreArray({2})); in TEST()
Dreshape_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon299a708d0111::ReshapeOpModel
71 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 2, 2})); in TEST()
79 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 1, 4})); in TEST()
Dbatch_to_space_nd_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb3939b050111::BatchToSpaceNDOpModel
101 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
112 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); in TEST()
Dsplit_test.cc58 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape() function in tflite::__anon02dcc3910111::SplitOpModel
84 EXPECT_THAT(m.GetOutputShape(i), ElementsAreArray(output_shape)) in Check()
94 EXPECT_THAT(const_m.GetOutputShape(i), ElementsAreArray(output_shape)) in Check()
Dexp_test.cc45 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonef6c1fd20111::ExpOpModel
57 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dreduce_join_op.cc99 TensorShape GetOutputShape(gtl::InlinedVector<bool, 8> index_is_reduced, in GetOutputShape() function
156 GetOutputShape(index_is_reduced, input_shape, keep_dims_); in Compute()