Searched refs:extra_inputs (Results 1 – 9 of 9) sorted by relevance
| /external/libtextclassifier/native/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/compiler/tensorrt/test/ |
| D | trt_mode_test.py | 105 extra_inputs=[],
149 extra_inputs=[[[1, 2, 3]], [[1, 4, 6]]],
|
| D | batch_matmul_test.py | 46 extra_inputs=[],
|
| D | tf_trt_integration_test_base.py | 211 input_mask, output_mask, extra_inputs, extra_outputs): argument
247 for extra_in_shape, extra_out_shape in zip(extra_inputs, extra_outputs):
261 input_dims=[input_shapes] + extra_inputs,
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 407 self._extra_inputs = temp_graph.extra_inputs
715 self.extra_inputs = []
845 self.extra_inputs.append(tensor)
1289 return g.extra_inputs
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 369 new_inputs = body_grad_graph.extra_inputs
927 self.extra_inputs = []
946 return self.extra_inputs
1163 self.extra_inputs.append(tensor_list)
|