| /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
| trt_shfn.cc | 63 std::vector<string> output_nodes; local
64 TF_RETURN_IF_ERROR(context->GetAttr("output_nodes", &output_nodes));
65 for (size_t i = 0; i < output_nodes.size(); i++) {
66 int binding_index = trt_engine->getBindingIndex(output_nodes[i].c_str());
76 LOG(FATAL) << "TensorRT engine cannot find binding: " << output_nodes[i];
|
| /system/libufdt/tests/src/ |
| ufdt_gen_test_dts.c | 114 static void output_nodes(FILE *fp, int count, const char *prop, const char *value) { function
141 output_nodes(fp, params->node_num, "status", "disabled");
|
| /external/tensorflow/tensorflow/cc/framework/ |
| gradients.cc | 99 std::unordered_set<int> output_nodes);
194 std::unordered_set<int> output_nodes) {
224 if (output_nodes.find(node_id) != output_nodes.end()) {
238 // Finally, we set stop_backprop_nodes to all output_nodes that aren't also
241 for (int output_node : output_nodes) {
266 std::unordered_set<int> output_nodes; local
267 output_nodes.reserve(outputs_.size());
269 output_nodes.insert(outputs_[i].node()->id());
273 GetStopBackpropNodes(reachable_nodes, output_nodes);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| dependency_optimizer.cc | 160 const std::set<NodeDef*> output_nodes = node_map_->GetOutputs(node_name); local
161 for (NodeDef* fanout : output_nodes) {
278 const std::vector<NodeDef*> output_nodes(output_node_set.begin(),
280 const int num_outputs = output_nodes.size();
298 output_nodes[0]->device() != node->device())) {
305 const string& out0_dev = output_nodes[0]->device();
306 const string& out1_dev = output_nodes[1]->device();
331 for (auto consumer : output_nodes) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_utils.cc | 396 std::set<string> output_nodes; local
415 output_nodes.insert(matched_node.name());
424 node_generator(*match, input_nodes, output_nodes, &new_nodes));
433 for (const string& expected_output : output_nodes) {
|
| /external/adhd/cras/src/common/ |
| cras_types.h | 292 * output_nodes - Output nodes currently attached.
329 struct cras_ionode_info output_nodes[CRAS_MAX_IONODES]; member in struct:cras_server_state
|
| /external/v8/src/compiler/ |
| instruction-selector.cc | 630 output_nodes(zone),
634 output_nodes.reserve(descriptor->ReturnCount());
643 NodeVector output_nodes; member in struct:v8::internal::compiler::CallBuffer
676 buffer->output_nodes.push_back(call);
678 buffer->output_nodes.resize(buffer->descriptor->ReturnCount(), nullptr);
682 DCHECK_LT(index, buffer->output_nodes.size());
683 DCHECK(!buffer->output_nodes[index]);
684 buffer->output_nodes[index] = use;
694 for (size_t i = 0; i < buffer->output_nodes.size(); i++) {
695 bool output_is_live = buffer->output_nodes[i] != nullptr |
[all...] |
