Searched defs:GetOutput (Results 1 - 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dexp_test.cc42 std::vector<T> GetOutput() { function in class:tflite::__anon25932::ExpOpModel
58 EXPECT_THAT(m.GetOutput<float>(),
H A Dl2norm_test.cc41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25940::L2NormOpModel
52 EXPECT_THAT(m.GetOutput(),
H A Dlocal_response_norm_test.cc44 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25941::LocalResponseNormOpModel
58 m.GetOutput(),
68 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear(
79 m.GetOutput(),
89 m.GetOutput(),
H A Dreshape_test.cc42 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25949::ReshapeOpModel
70 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8}));
78 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8}));
H A Dresize_bilinear_test.cc53 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25950::ResizeBilinearOpModel
66 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 5, 6})));
71 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 5, 6})));
79 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 7, 9})));
84 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 7, 9})));
95 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
107 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({
124 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
141 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({
159 EXPECT_THAT(m.GetOutput(), ElementsAreArra
[all...]
H A Dspace_to_depth_test.cc43 std::vector<T> GetOutput() { function in class:tflite::__anon25955::SpaceToDepthOpModel
62 EXPECT_THAT(m.GetOutput<float>(),
71 EXPECT_THAT(m.GetOutput<uint8_t>(), ElementsAreArray({1, 2, 3, 4}));
79 EXPECT_THAT(m.GetOutput<int32_t>(),
88 EXPECT_THAT(m.GetOutput<int64_t>(),
H A Dsqueeze_test.cc55 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25957::FloatSqueezeOpModel
69 m.GetOutput(),
85 m.GetOutput(),
101 m.GetOutput(),
114 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.85}));
H A Dactivations_test.cc67 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25921::FloatActivationsOpModel
83 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } function in class:tflite::__anon25921::QuantizedActivationsOpModel
98 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
112 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
126 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
140 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
154 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
176 EXPECT_THAT(m.GetOutput(),
188 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({
203 EXPECT_THAT(m2.GetOutput(), ElementsAreArra
[all...]
H A Dconcatenation_test.cc57 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25926::ConcatenationOpModel
66 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } function in class:tflite::__anon25926::QuantizedConcatenationOpModel
78 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 3, 4, 7}));
86 EXPECT_THAT(m0.GetOutput(), ::testing::ElementsAre(5));
94 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6}));
107 EXPECT_THAT(m0.GetOutput(),
115 EXPECT_THAT(m0_negative.GetOutput(),
123 EXPECT_THAT(m1.GetOutput(),
131 EXPECT_THAT(m1_negative.GetOutput(),
143 EXPECT_THAT(m0.GetOutput(),
[all...]
H A Dpad_test.cc36 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25947::PadOpModel
111 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 0, 0, 1, 2, 0, 0, 3, 4,
121 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, 0, 0, 1, 2, 0, 0, 3, 4,
130 EXPECT_THAT(m.GetOutput(),
141 EXPECT_THAT(m.GetOutput(),
H A Dpooling_test.cc59 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25948::FloatPoolingOpModel
70 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } function in class:tflite::__anon25948::QuantizedPoolingOpModel
87 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2.75, 5.75}));
106 EXPECT_THAT(m.GetOutput(), ElementsAreArray({44, 92}));
119 EXPECT_THAT(m.GetOutput(), ElementsAreArray({6, 10}));
138 EXPECT_THAT(m.GetOutput(), ElementsAreArray({96, 160}));
151 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.5, 6.5}));
H A Dskip_gram_test.cc48 std::vector<string> GetOutput() { function in class:tflite::__anon25952::SkipGramOp
70 EXPECT_THAT(m.GetOutput(), testing::UnorderedElementsAreArray(
79 EXPECT_THAT(m.GetOutput(),
89 EXPECT_THAT(m.GetOutput(),
103 EXPECT_THAT(m.GetOutput(),
122 m.GetOutput(),
133 EXPECT_THAT(m.GetOutput(),
147 EXPECT_THAT(m.GetOutput(),
163 EXPECT_THAT(m.GetOutput(),
187 m.GetOutput(),
[all...]
H A Dadd_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25922::FloatAddOpModel
79 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1.9, 0.4, 1.0, 1.3}));
89 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1.0, 0.4, 1.0, 1.0}));
102 EXPECT_THAT(m.GetOutput(),
119 m.GetOutput(),
H A Dbatch_to_space_nd_test.cc41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25924::BatchToSpaceNDOpModel
102 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 5, 2, 6, 9, 13, 10, 14, 3, 7,
113 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 5, 2, 6, 9, 13, 10, 14, 3, 7,
H A Ddepthwise_conv_test.cc90 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25928::DepthwiseConvolutionOpModel
113 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
136 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } function in class:tflite::__anon25928::QuantizedDepthwiseConvolutionOpModel
173 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
H A Dembedding_lookup_test.cc61 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25931::EmbeddingLookupOpModel
79 EXPECT_THAT(m.GetOutput(),
H A Dkernel_util.h31 inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, function in namespace:tflite
H A Dlsh_projection_test.cc69 std::vector<int> GetOutput() { return ExtractVector<int>(output_); } function in class:tflite::__anon25942::LSHProjectionOpModel
89 EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 0, 1, 0, 0));
100 EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 1, 8 + 0));
113 EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 2, 4 + 1, 8 + 1));
H A Dmean_test.cc36 std::vector<T> GetOutput() { function in class:tflite::__anon25944::BaseMeanOpModel
86 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({12, 13})));
98 EXPECT_THAT(m.GetOutput<float>(),
114 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray(ArrayFloatNear({12, 13})));
129 EXPECT_THAT(m.GetOutput<float>(),
H A Dmul_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25945::FloatMulOpModel
79 EXPECT_THAT(m.GetOutput(),
90 EXPECT_THAT(m.GetOutput(),
105 m.GetOutput(),
122 m.GetOutput(),
H A Dsoftmax_test.cc51 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25953::SoftmaxOpModel
72 m.GetOutput(),
104 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear(expected, 1e-6)));
132 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear(expected, 1e-6)));
H A Dspace_to_batch_nd_test.cc41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25954::SpaceToBatchNDOpModel
107 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
118 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
127 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
138 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7,
147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
160 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
171 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
185 EXPECT_THAT(m.GetOutput(), ElementsAreArray({
H A Dsplit_test.cc55 std::vector<float> GetOutput(int i) { function in class:tflite::__anon25956::SplitOpModel
83 EXPECT_THAT(m.GetOutput(i), ElementsAreArray(output_data[i])) << debug(i);
92 EXPECT_THAT(const_m.GetOutput(i), ElementsAreArray(output_data[i]))
H A Dstrided_slice_test.cc61 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25958::StridedSliceOpModel
93 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
114 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3}));
125 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3}));
136 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2}));
147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4}));
158 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3}));
169 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3}));
180 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4}));
191 EXPECT_THAT(m.GetOutput(), ElementsAreArra
[all...]
/external/tensorflow/tensorflow/core/kernels/
H A Dops_testutil.cc44 Tensor* OpsTestBase::GetOutput(int output_index) { function in class:tensorflow::OpsTestBase

Completed in 303 milliseconds

12