/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | exp_test.cc | 42 std::vector<T> GetOutput() { function in class:tflite::__anon25932::ExpOpModel 58 EXPECT_THAT(m.GetOutput<float>(),
|
H A D | l2norm_test.cc | 41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25940::L2NormOpModel 52 EXPECT_THAT(m.GetOutput(),
|
H A D | local_response_norm_test.cc | 44 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 D | reshape_test.cc | 42 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 D | resize_bilinear_test.cc | 53 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 D | space_to_depth_test.cc | 43 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 D | squeeze_test.cc | 55 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 D | activations_test.cc | 67 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 D | concatenation_test.cc | 57 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 D | pad_test.cc | 36 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 D | pooling_test.cc | 59 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 D | skip_gram_test.cc | 48 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 D | add_test.cc | 52 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 D | batch_to_space_nd_test.cc | 41 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 D | depthwise_conv_test.cc | 90 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 D | embedding_lookup_test.cc | 61 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } function in class:tflite::__anon25931::EmbeddingLookupOpModel 79 EXPECT_THAT(m.GetOutput(),
|
H A D | kernel_util.h | 31 inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, function in namespace:tflite
|
H A D | lsh_projection_test.cc | 69 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 D | mean_test.cc | 36 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 D | mul_test.cc | 52 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 D | softmax_test.cc | 51 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 D | space_to_batch_nd_test.cc | 41 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 D | split_test.cc | 55 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 D | strided_slice_test.cc | 61 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 D | ops_testutil.cc | 44 Tensor* OpsTestBase::GetOutput(int output_index) { function in class:tensorflow::OpsTestBase
|