Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dtopk_v2_test.cc58 return ExtractVector<int32>(output_indexes_); in GetIndexes()
62 return ExtractVector<float>(output_values_); in GetValuesFloat()
66 return ExtractVector<uint8>(output_values_); in GetValuesUInt8()
70 return ExtractVector<int32>(output_values_); in GetValuesInt32()
74 return ExtractVector<int64_t>(output_values_); in GetValuesInt64()
Dpooling_test.cc59 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
70 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
72 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Ddepthwise_conv_test.cc90 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
136 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
138 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dgather_test.cc55 std::vector<float> GetOutputFloat() { return ExtractVector<float>(output_); } in GetOutputFloat()
57 return ExtractVector<uint8_t>(output_); in GetOutputUint8()
60 return ExtractVector<string>(output_); in GetOutputString()
Dconcatenation_test.cc57 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
66 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
68 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dactivations_test.cc67 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
83 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
85 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dfully_connected_test.cc187 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
204 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
206 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dconv_test.cc101 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
393 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput()
395 return Dequantize<uint8_t>(ExtractVector<uint8_t>(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()
Dmul_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
67 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dadd_test.cc52 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
60 return Dequantize<uint8_t>(ExtractVector<uint8_t>(output_), in GetDequantizedOutput()
Dl2norm_test.cc41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dexp_test.cc43 return ExtractVector<T>(output_); in GetOutput()
Dtest_util.h183 std::vector<T> ExtractVector(int index) { in ExtractVector() function
254 std::vector<string> SingleOpModel::ExtractVector(int index);
Dlocal_response_norm_test.cc44 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dreshape_test.cc42 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dembedding_lookup_test.cc61 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dspace_to_depth_test.cc44 return ExtractVector<T>(output_); in GetOutput()
Dlsh_projection_test.cc69 std::vector<int> GetOutput() { return ExtractVector<int>(output_); } in GetOutput()
Dsqueeze_test.cc55 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dbidirectional_sequence_rnn_test.cc735 std::vector<float> GetFwOutput() { return ExtractVector<float>(fw_output_); } in GetFwOutput()
736 std::vector<float> GetBwOutput() { return ExtractVector<float>(bw_output_); } in GetBwOutput()
Dbatch_to_space_nd_test.cc41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
Dsoftmax_test.cc51 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput()
/external/tensorflow/tensorflow/contrib/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()

12