| /external/tensorflow/tensorflow/lite/ |
| D | simple_planner.cc | 131 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations() local
132 for (int j = 0; j < node_inputs->size; ++j) { in PlanAllocations()
133 int tensor_index = node_inputs->data[j]; in PlanAllocations()
153 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations() local
154 for (int j = 0; j < node_inputs->size; ++j) { in PlanAllocations()
155 int tensor_index = node_inputs->data[j]; in PlanAllocations()
|
| D | arena_planner.cc | 149 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations() local
150 for (int j = 0; j < node_inputs->size; ++j) { in PlanAllocations()
151 int tensor_index = node_inputs->data[j]; in PlanAllocations()
172 TfLiteIntArray* node_inputs = node.inputs; in PlanAllocations() local
173 for (int j = 0; j < node_inputs->size; ++j) { in PlanAllocations()
174 int tensor_index = node_inputs->data[j]; in PlanAllocations()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_data.py | 837 if datum.node_name not in debug_graph.node_inputs:
842 for node in debug_graph.node_inputs:
844 inputs = debug_graph.node_inputs[node]
1036 nodes.extend(self._debug_graphs[device_name].node_inputs.keys())
1041 return self._debug_graphs[device_name].node_inputs.keys()
1063 def node_inputs(self, node_name, is_control=False, device_name=None): member in DebugDumpDir
1088 return self._debug_graphs[device_name].node_inputs[node_name]
1122 input_lists = [self._debug_graphs[device_name].node_inputs]
1197 input_lists = [self._debug_graphs[dst_device_name].node_inputs]
1288 if node_name in debug_graph.node_inputs:
|
| D | session_debug_testlib.py | 805 self.assertEqual([], dump.node_inputs(u_name))
806 self.assertEqual([u_name], dump.node_inputs(u_read_name))
807 self.assertEqual([u_read_name] * 2, dump.node_inputs(v_name))
808 self.assertEqual([v_name] * 2, dump.node_inputs(w_name))
810 self.assertEqual([], dump.node_inputs(u_name, is_control=True))
811 self.assertEqual([], dump.node_inputs(u_read_name, is_control=True))
812 self.assertEqual([], dump.node_inputs(v_name, is_control=True))
813 self.assertEqual([], dump.node_inputs(w_name, is_control=True))
828 dump.node_inputs(u_name + "foo")
|
| D | debug_graphs.py | 457 def node_inputs(self): member in DebugGraph
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 295 TensorValueVec node_inputs; in Run() local
416 node_inputs.clear(); in Run()
417 node_inputs.resize(num_inputs); in Run()
428 node_inputs[j].tensor = const_cast<Tensor*>(input.const_tensor); in Run()
431 node_inputs[j].tensor = input.val.get(); in Run()
438 params.inputs = node_inputs; in Run()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_topology_view_test.cc | 34 const auto& node_inputs = node.second; in CreateGraph() local
38 for (const string& input : node_inputs) { in CreateGraph()
|
| D | mutable_graph_view_test.cc | 310 absl::flat_hash_map<string, std::vector<string>> node_inputs; in GetNodeInputsFromGraph() local
315 node_inputs[node.name()] = in GetNodeInputsFromGraph()
318 return node_inputs; in GetNodeInputsFromGraph()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | hash_utils.cc | 228 node_rep.node_inputs.push_back(std::make_pair(input_node, suffix)); in Init()
266 for (const auto& input : node_rep->node_inputs) { in HashNode()
311 auto& this_node_inputs = nodes_[this_node].node_inputs; in CheckNodesEqualHelper()
312 auto& that_node_inputs = that->nodes_[that_node].node_inputs; in CheckNodesEqualHelper()
694 std::vector<std::pair<const NodeDef*, absl::string_view>> node_inputs; member
|
| /external/pytorch/torch/csrc/jit/passes/utils/ |
| D | subgraph_utils.cpp | 511 std::set<Value*> node_inputs(n->inputs().begin(), n->inputs().end()); in unmergeNode() local
531 if (node_inputs.count(subgraph->outputs().at(i)) != 0) { in unmergeNode()
542 node_inputs.erase(subgraph->outputs().at(i)); in unmergeNode()
550 for (auto ni : node_inputs) { in unmergeNode()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | symbolic_shape_analysis.cpp | 980 std::vector<Value*> node_inputs; in joinPartialEvaluatedShapeGraphToLargeShapeGraph() local
986 node_inputs.push_back(v); in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
988 node_inputs.push_back(curr->namedInput("dim")); in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
991 node_inputs.push_back(curr->input(i)); in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
996 for (size_t i = 0; i < node_inputs.size(); ++i) { in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
997 auto node_input = node_inputs[i]; in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
|
| /external/federated-compute/fcp/artifact_building/ |
| D | graph_helpers.py | 401 node_inputs = new_node.input
402 if init_op_control_dep not in node_inputs:
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli.py | 794 self._debug_dump.node_inputs(node_name))
796 self._debug_dump.node_inputs(node_name, is_control=True))
1302 tracker = self._debug_dump.node_inputs
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 210 TfLiteIntArrayView node_inputs(node->inputs);
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 1203 TfLiteIntArray* node_inputs = node.inputs; in RemoveUnusedInputs() local
1204 for (int j = 0; j < node_inputs->size; ++j) { in RemoveUnusedInputs()
1205 int tensor_index = node_inputs->data[j]; in RemoveUnusedInputs()
|