Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/lite/kernels/
Dcast_test.cc50 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
58 EXPECT_THAT(m.ExtractVector<int>(m.output()), in TEST()
66 EXPECT_THAT(m.ExtractVector<bool>(m.output()), in TEST()
74 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
86 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
95 m.ExtractVector<std::complex<float>>(m.output()), in TEST()
110 EXPECT_THAT(m.ExtractVector<int>(m.output()), in TEST()
119 m.ExtractVector<std::complex<float>>(m.output()), in TEST()
135 m.ExtractVector<std::complex<float>>(m.output()), in TEST()
Delementwise_test.cc63 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
72 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
81 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
93 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({ in TEST()
103 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
112 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
121 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
130 EXPECT_THAT(m.ExtractVector<bool>(m.output()), in TEST()
Dtopk_v2_test.cc62 return ExtractVector<int32_t>(output_indexes_); in GetIndexes()
66 return ExtractVector<float>(output_values_); in GetValuesFloat()
70 return ExtractVector<uint8_t>(output_values_); in GetValuesUInt8()
74 return ExtractVector<int8_t>(output_values_); in GetValuesInt8()
78 return ExtractVector<int32_t>(output_values_); in GetValuesInt32()
82 return ExtractVector<int64_t>(output_values_); in GetValuesInt64()
Dfill_test.cc53 EXPECT_THAT(m.ExtractVector<int32_t>(m.output()), in TEST()
63 EXPECT_THAT(m.ExtractVector<int64_t>(m.output()), in TEST()
74 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
83 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({4.0})); in TEST()
Dzeros_like_test.cc48 EXPECT_THAT(m.ExtractVector<float>(m.output()), in TEST()
57 EXPECT_THAT(m.ExtractVector<int32_t>(m.output()), in TEST()
66 EXPECT_THAT(m.ExtractVector<int64_t>(m.output()), in TEST()
Dpooling_test.cc59 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
74 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
76 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
93 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
95 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Ddetection_postprocess_test.cc94 return ExtractVector<T>(output1_); in GetOutput1()
99 return ExtractVector<T>(output2_); in GetOutput2()
104 return ExtractVector<T>(output3_); in GetOutput3()
109 return ExtractVector<T>(output4_); in GetOutput4()
305 return ExtractVector<T>(output1_); in GetOutput1()
310 return ExtractVector<T>(output2_); in GetOutput2()
315 return ExtractVector<T>(output3_); in GetOutput3()
320 return ExtractVector<T>(output4_); in GetOutput4()
Ddepthwise_conv_test.cc129 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
278 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
280 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
483 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
485 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Dactivations_test.cc79 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
112 return ExtractVector<T>(output_); in GetOutput()
116 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
864 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
882 return ExtractVector<T>(output_); in GetOutput()
886 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
953 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dmul_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
59 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
78 return Dequantize<integer_dtype>(ExtractVector<integer_dtype>(output_), in GetDequantizedOutput()
83 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutputInt16()
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()
Dconv_test.cc123 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
608 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
610 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
807 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
1126 std::vector<int8_t> GetOutput() { return ExtractVector<int8_t>(output_); } in GetOutput()
1128 return Dequantize<int8_t>(ExtractVector<int8_t>(output_), GetScale(output_), in GetDequantizedOutput()
Dunique_test.cc41 std::vector<T> GetOutput() { return ExtractVector<T>(output_id_); } in GetOutput()
43 return ExtractVector<I>(output_index_id_); in GetIndexesOutput()
Dconcatenation_test.cc58 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
87 return ExtractVector<T>(output_); in GetOutput()
91 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
Dl2norm_test.cc55 return ExtractVector<T>(output_); in GetOutput()
60 return Dequantize<T>(ExtractVector<T>(output_), GetScale(output_), in GetDequantizedOutput()
Dhashtable_lookup_test.cc90 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
91 std::vector<uint8_t> GetHit() { return ExtractVector<uint8_t>(hit_); } in GetHit()
Dadd_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
59 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
68 return Dequantize<integer_dtype>(ExtractVector<integer_dtype>(output_), in GetDequantizedOutput()
73 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutputInt16()
Dsquared_difference_test.cc53 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
60 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
Dsub_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
59 std::vector<int32_t> GetOutput() { return ExtractVector<int32_t>(output_); } in GetOutput()
68 return Dequantize<integer_dtype>(ExtractVector<integer_dtype>(output_), in GetDequantizedOutput()
73 return Dequantize<int16_t>(ExtractVector<int16_t>(output_), in GetDequantizedOutputInt16()
/external/libtextclassifier/utils/tflite/
Dtext_encoder_test.cc57 return ExtractVector<int>(output_encoding_); in GetOutputEncoding()
60 return ExtractVector<int>(output_positions_); in GetOutputPositions()
63 return ExtractVector<int>(output_attributes_int32_); in GetOutputAttributeInt32()
66 return ExtractVector<float>(output_attributes_float_); in GetOutputAttributeFloat()
68 int GetEncodedLength() { return ExtractVector<int>(output_length_)[0]; } in GetEncodedLength()
Dtoken_encoder_test.cc48 return ExtractVector<int>(output_positions_); in GetOutputPositions()
51 return ExtractVector<int>(output_attributes_int32_); in GetOutputAttributeInt32()
54 return ExtractVector<float>(output_attributes_float_); in GetOutputAttributeFloat()
56 int GetOutputLength() { return ExtractVector<int>(output_length_)[0]; } in GetOutputLength()
Ddist_diversification_test.cc38 int GetOutputLen() { return ExtractVector<int>(output_len_).front(); } in GetOutputLen()
40 auto res = ExtractVector<int>(output_indexes_); in GetOutputIndexes()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc107 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
177 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
217 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
321 return ExtractVector<float>(output_); in GetOutput()
323 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetOutput()
332 return ExtractVector<uint8_t>(output_); in GetQuantizedOutput()
490 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
569 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
643 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
689 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
[all …]
/external/tensorflow/tensorflow/lite/models/smartreply/ops/
Dextract_feature_test.cc47 std::vector<int> GetSignature() { return ExtractVector<int>(signature_); } in GetSignature()
48 std::vector<float> GetWeight() { return ExtractVector<float>(weight_); } in GetWeight()
Dpredict_test.cc72 std::vector<int> GetLabel() { return ExtractVector<int>(output_label_); } in GetLabel()
74 return ExtractVector<float>(output_weight_); in GetWeight()

1234