Searched refs:node_inputs (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/lite/ |
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/core/kernels/data/ |
D | single_threaded_executor.cc | 281 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()
|
D | hash_utils.cc | 195 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/ |
D | debug_data.py | 842 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:
|
D | session_debug_testlib.py | 810 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")
|
D | debug_graphs.py | 461 def node_inputs(self): member in DebugGraph
|
/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/python/debug/cli/ |
D | analyzer_cli.py | 799 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/ |
D | coreml_delegate_kernel.mm | 209 TfLiteIntArrayView node_inputs(node->inputs);
|