Home
last modified time | relevance | path

Searched refs:ExtractVector (Results 1 – 25 of 170) 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.cc75 auto vec = ExtractVector<T>(index); in ExtractDequantVector()
129 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
138 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
147 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
159 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({ in TEST()
243 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
252 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
362 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
371 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()
Dadd_n_test_common.h53 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
60 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
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()
511 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutput()
682 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
725 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
760 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Ddetection_postprocess_test.cc93 return ExtractVector<T>(output1_); in GetOutput1()
98 return ExtractVector<T>(output2_); in GetOutput2()
103 return ExtractVector<T>(output3_); in GetOutput3()
108 return ExtractVector<T>(output4_); in GetOutput4()
435 return ExtractVector<T>(output1_); in GetOutput1()
440 return ExtractVector<T>(output2_); in GetOutput2()
445 return ExtractVector<T>(output3_); in GetOutput3()
450 return ExtractVector<T>(output4_); in GetOutput4()
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()
Dreduce_test_common.h35 return ExtractVector<T>(output_); in GetOutput()
40 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
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()
569 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), in GetOutput()
572 return FloatVector(ExtractVector<T>(output_)); in GetOutput()
Ddensify_test.cc60 std::vector<T> GetInput() { return ExtractVector<T>(input_); } in GetInput()
61 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()
79 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()
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()
Dconv_test.cc145 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
907 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
909 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
1165 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
1639 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
1641 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
1779 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
1783 return ExtractVector<T>(filter_); in GetFilter()
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()
Dtopk_v2_test.cc59 return ExtractVector<int32_t>(output_indexes_); in GetIndexes()
63 return ExtractVector<InputType>(output_values_); in GetValues()
Drandom_uniform_test.cc80 return ExtractVector<T>(output_); in GetOutput()
109 return ExtractVector<T>(output_); in GetOutput()
Ddynamic_update_slice_test.cc70 return ExtractVector<T>(output_); in GetOutput()
74 return ExtractVector<string>(output_); in GetStringOutput()
Dmul_test.cc58 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
66 return ExtractVector<std::complex<float>>(output_); in GetOutput()
74 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
93 return Dequantize<integer_dtype>(ExtractVector<integer_dtype>(output_), in GetDequantizedOutput()
98 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutputInt16()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc132 *output = ExtractVector<float>(index); in GetData()
135 *output = Dequantize<uint8_t>(ExtractVector<uint8_t>(index), in GetData()
186 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
548 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
587 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
685 return ExtractVector<float>(output_); in GetOutput()
687 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetOutput()
696 return ExtractVector<uint8_t>(output_); in GetQuantizedOutput()
955 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
957 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
[all …]
Dnnapi_delegate_signed_quantization_test.cc114 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
333 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
335 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
436 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
438 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
720 return ExtractVector<T>(output_); in GetOutput()
821 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args_test.cc56 std::vector<int> GetOutput1() { return ExtractVector<int>(output1_); } in GetOutput1()
58 std::vector<int> GetOutput2() { return ExtractVector<int>(output2_); } in GetOutput2()
158 std::vector<int64_t> GetOutput1() { return ExtractVector<int64_t>(output1_); } in GetOutput1()
160 std::vector<int64_t> GetOutput2() { return ExtractVector<int64_t>(output2_); } in GetOutput2()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dl2_norm_test.cc50 return ExtractVector<T>(output_); in GetOutput()
55 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