Home
last modified time | relevance | path

Searched refs:ExtractVector (Results 1 – 25 of 153) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/kernels/
Dcast_test.cc53 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
61 EXPECT_THAT(m.ExtractVector<int32_t>(m.output()), in TEST()
69 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
77 EXPECT_THAT(m.ExtractVector<int32_t>(m.output()), in TEST()
85 EXPECT_THAT(m.ExtractVector<int16_t>(m.output()), in TEST()
93 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
101 EXPECT_THAT(m.ExtractVector<int64_t>(m.output()), in TEST()
109 EXPECT_THAT(m.ExtractVector<bool>(m.output()), in TEST()
117 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
125 EXPECT_THAT(m.ExtractVector<uint8_t>(m.output()), in TEST()
[all …]
Delementwise_test.cc74 auto vec = ExtractVector<T>(index); in ExtractDequantVector()
128 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
137 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
146 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
158 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({ in TEST()
242 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
251 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
361 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
370 EXPECT_THAT(m.ExtractVector<bool>(m.output()), in TEST()
Dzeros_like_test.cc53 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
62 EXPECT_THAT(m.ExtractVector<int32_t>(m.output()), in TEST()
71 EXPECT_THAT(m.ExtractVector<int64_t>(m.output()), in TEST()
Dcomplex_support_test.cc45 std::vector<T> GetOutput() { return ExtractVector<T>(output_); } in GetOutput()
106 std::vector<T> GetOutput() { return ExtractVector<T>(output_); } in GetOutput()
167 std::vector<T> GetOutput() { return ExtractVector<T>(output_); } in GetOutput()
Dtranspose_conv_test.cc121 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
306 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
389 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
502 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutput()
673 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
716 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
751 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Ddetection_postprocess_test.cc95 return ExtractVector<T>(output1_); in GetOutput1()
100 return ExtractVector<T>(output2_); in GetOutput2()
105 return ExtractVector<T>(output3_); in GetOutput3()
110 return ExtractVector<T>(output4_); in GetOutput4()
370 return ExtractVector<T>(output1_); in GetOutput1()
375 return ExtractVector<T>(output2_); in GetOutput2()
380 return ExtractVector<T>(output3_); in GetOutput3()
385 return ExtractVector<T>(output4_); in GetOutput4()
Dadd_n_test.cc58 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
65 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
Dnon_max_suppression_test.cc44 return ExtractVector<int>(output_selected_indices_); in GetSelectedIndices()
48 return ExtractVector<float>(output_selected_scores_); in GetSelectedScores()
52 return ExtractVector<int>(output_num_selected_indices_); in GetNumSelectedIndices()
Ddensify_test.cc58 std::vector<T> GetInput() { return ExtractVector<T>(input_); } in GetInput()
59 std::vector<T> GetOutput() { return ExtractVector<T>(output_); } in GetOutput()
Dsquared_difference_test.cc57 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
64 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
78 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Dunique_test.cc44 std::vector<T> GetOutput() { return ExtractVector<T>(output_id_); } in GetOutput()
46 return ExtractVector<I>(output_index_id_); in GetIndexesOutput()
Dconv_test.cc145 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
880 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
882 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
1104 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
1544 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
1546 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
1662 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
1666 return ExtractVector<T>(filter_); in GetFilter()
Dpooling_test.cc62 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
77 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
79 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
96 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
98 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
115 std::vector<int16_t> GetOutput() { return ExtractVector<int16_t>(output_); } in GetOutput()
117 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutput()
Dtopk_v2_test.cc61 return ExtractVector<int32_t>(output_indexes_); in GetIndexes()
65 return ExtractVector<InputType>(output_values_); in GetValues()
Dconcatenation_test.cc71 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
84 return ExtractVector<T>(output_); in GetOutput()
88 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
99 std::vector<bool> GetOutput() { return ExtractVector<bool>(output_); } in GetOutput()
Dhashtable_lookup_test.cc96 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
97 std::vector<uint8_t> GetHit() { return ExtractVector<uint8_t>(hit_); } in GetHit()
Drandom_uniform_test.cc77 return ExtractVector<T>(output_); in GetOutput()
106 return ExtractVector<T>(output_); in GetOutput()
Dwhere_test.cc51 std::vector<int64_t> GetOutput() { return ExtractVector<int64_t>(output_); } in GetOutput()
66 std::vector<int64_t> GetOutput() { return ExtractVector<int64_t>(output_); } in GetOutput()
Dbatch_matmul_test.cc58 std::vector<T> GetOutput() { return ExtractVector<T>(output_id_); } in GetOutput()
315 return ExtractVector<float>(neg_output_id_); in GetOutput()
384 std::vector<float> GetOutput() { return ExtractVector<float>(output_id_); } in GetOutput()
750 return ExtractVector<T>(output_id_); in GetOutput()
755 return Dequantize<T>(ExtractVector<T>(output_id_), GetScale(output_id_), in GetDequantizedOutput()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc122 *output = ExtractVector<float>(index); in GetData()
125 *output = Dequantize<uint8_t>(ExtractVector<uint8_t>(index), in GetData()
173 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
504 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
543 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
641 return ExtractVector<float>(output_); in GetOutput()
643 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetOutput()
652 return ExtractVector<uint8_t>(output_); in GetQuantizedOutput()
878 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
880 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
[all …]
Dnnapi_delegate_signed_quantization_test.cc110 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
329 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
331 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
432 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
434 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
716 return ExtractVector<T>(output_); in GetOutput()
817 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dadd_n_test.cc57 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
64 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc223 return ExtractVector<T>(sparse_indices_outputs_[i]); in GetSparseIndicesOutput()
228 return ExtractVector<T>(sparse_values_outputs_[i]); in GetSparseValuesOutput()
233 return ExtractVector<T>(sparse_shapes_outputs_[i]); in GetSparseShapesOutput()
238 return ExtractVector<T>(dense_outputs_[i]); in GetDenseOutput()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dl2_norm_test.cc48 return ExtractVector<T>(output_); in GetOutput()
53 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite_test.cc65 return ExtractVector<int32>(output_splits_); in GetSplits()
67 std::vector<T> GetValues() const { return ExtractVector<T>(output_values_); } in GetValues()

1234567