Searched refs:extra_inputs (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/lang_id/common/ |
| D | embedding-network.cc | 252 const std::vector<float> &extra_inputs, std::vector<float> *scores) const { in ComputeFinalScores() argument
256 if (!extra_inputs.empty()) { in ComputeFinalScores()
257 input.reserve(input.size() + extra_inputs.size()); in ComputeFinalScores()
258 for (int i = 0; i < extra_inputs.size(); i++) { in ComputeFinalScores()
259 input.push_back(extra_inputs[i]); in ComputeFinalScores()
|
| D | embedding-network.h | 61 const std::vector<float> &extra_inputs,
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 178 const std::vector<int>& extra_inputs,
185 const std::vector<int>& extra_inputs);
197 const int inputs_size, const std::vector<int>& extra_inputs,
|
| D | graph_transferer.cc | 657 std::vector<int> extra_inputs{stride_id}; in RegisterNodeWithPaddingAndStrides() local
662 extra_inputs.insert(extra_inputs.begin(), ksize_id); in RegisterNodeWithPaddingAndStrides()
673 static_cast<int>(padding), node.num_inputs(), extra_inputs, in RegisterNodeWithPaddingAndStrides()
692 std::vector<int> extra_inputs{const_val_id}; in RegisterNodeWithRank() local
706 padding_id, node.num_inputs(), extra_inputs, node.num_outputs(), in RegisterNodeWithRank()
856 const std::vector<int>& extra_inputs, in AppendNodeParams() argument
865 static_cast<int>(extra_inputs.size())); in AppendNodeParams()
887 const int id, const Node& node, const std::vector<int>& extra_inputs) { in AppendNodeInputParams() argument
889 << ", " << extra_inputs.size(); in AppendNodeInputParams()
896 for (const int extra_input : extra_inputs) { in AppendNodeInputParams()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 365 self._extra_inputs = temp_graph.extra_inputs
673 self.extra_inputs = []
777 self.extra_inputs.append(tensor)
1211 return g.extra_inputs
|