Home
last modified time | relevance | path

Searched refs:node_inputs (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/
Darena_planner.cc149 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/core/kernels/data/
Dsingle_threaded_executor.cc281 TensorValueVec node_inputs; in Run() local
299 params.inputs = &node_inputs; in Run()
387 node_inputs.clear(); in Run()
388 node_inputs.resize(num_inputs); in Run()
399 node_inputs[j].tensor = const_cast<Tensor*>(input.const_tensor); in Run()
402 node_inputs[j].tensor = input.val.get(); in Run()
Dhash_utils.cc195 node_rep.node_inputs.push_back(std::make_pair(input_node, suffix)); in Init()
233 for (const auto& input : node_rep->node_inputs) { in HashNode()
274 auto& this_node_inputs = nodes_[this_node].node_inputs; in CheckNodesEqualHelper()
275 auto& that_node_inputs = that->nodes_[that_node].node_inputs; in CheckNodesEqualHelper()
622 std::vector<std::pair<const NodeDef*, std::string>> node_inputs; member
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_data.py842 if datum.node_name not in debug_graph.node_inputs:
847 for node in debug_graph.node_inputs:
849 inputs = debug_graph.node_inputs[node]
1041 nodes.extend(self._debug_graphs[device_name].node_inputs.keys())
1046 return self._debug_graphs[device_name].node_inputs.keys()
1068 def node_inputs(self, node_name, is_control=False, device_name=None): member in DebugDumpDir
1093 return self._debug_graphs[device_name].node_inputs[node_name]
1127 input_lists = [self._debug_graphs[device_name].node_inputs]
1202 input_lists = [self._debug_graphs[dst_device_name].node_inputs]
1293 if node_name in debug_graph.node_inputs:
Dsession_debug_testlib.py810 self.assertEqual([], dump.node_inputs(u_name))
811 self.assertEqual([u_name], dump.node_inputs(u_read_name))
812 self.assertEqual([u_read_name] * 2, dump.node_inputs(v_name))
813 self.assertEqual([v_name] * 2, dump.node_inputs(w_name))
815 self.assertEqual([], dump.node_inputs(u_name, is_control=True))
816 self.assertEqual([], dump.node_inputs(u_read_name, is_control=True))
817 self.assertEqual([], dump.node_inputs(v_name, is_control=True))
818 self.assertEqual([], dump.node_inputs(w_name, is_control=True))
833 dump.node_inputs(u_name + "foo")
Ddebug_graphs.py461 def node_inputs(self): member in DebugGraph
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view_test.cc34 const auto& node_inputs = node.second; in CreateGraph() local
38 for (const string& input : node_inputs) { in CreateGraph()
Dmutable_graph_view_test.cc310 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/python/debug/cli/
Danalyzer_cli.py799 self._debug_dump.node_inputs(node_name))
801 self._debug_dump.node_inputs(node_name, is_control=True))
1307 tracker = self._debug_dump.node_inputs
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate_kernel.mm209 TfLiteIntArrayView node_inputs(node->inputs);