Searched refs:output_datas (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unpack_test.cc | 49 std::vector<std::vector<T>> output_datas; in GetOutputDatas() local
52 output_datas.push_back(ExtractVector<T>(output)); in GetOutputDatas()
54 return output_datas; in GetOutputDatas()
84 const std::vector<std::vector<float>>& output_datas = model.GetOutputDatas(); in TEST() local
85 EXPECT_EQ(output_datas.size(), 3); in TEST()
86 EXPECT_THAT(output_datas[0], ElementsAre(1, 2)); in TEST()
87 EXPECT_THAT(output_datas[1], ElementsAre(3, 4)); in TEST()
88 EXPECT_THAT(output_datas[2], ElementsAre(5, 6)); in TEST()
103 const std::vector<std::vector<float>>& output_datas = model.GetOutputDatas(); in TEST() local
104 EXPECT_EQ(output_datas.size(), 2); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1067 int dimensions, int outputs_count, Scalar* const* output_datas, in Unpack() argument
1074 DimsToShape(output_dims), output_datas); in Unpack()
|
| D | reference_ops.h | 1743 Scalar* const* output_datas) { in Unpack() argument
1766 Scalar* output_ptr = output_datas[i] + copy_size * k; in Unpack()
|