| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 211 string node_input = it->second->input(i); in AddGraph() local
215 auto prefix_pos = node_input.find(':'); in AddGraph()
216 if (prefix_pos != node_input.npos) { in AddGraph()
217 std::vector<string> input_parts = absl::StrSplit(node_input, ':'); in AddGraph()
219 << "Unknown NodeDef.input format: " << node_input; in AddGraph()
220 node_input = input_parts[0]; in AddGraph()
224 if (node_input.substr(0, 1) == "^") { in AddGraph()
225 node_input = node_input.substr(1); in AddGraph()
230 node->AddInput(node_input, output_idx, i); in AddGraph()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 266 for (const GraphTransferNodeInput& node_input : in SortParams() local
267 input_map.at(node_id)->node_input()) { in SortParams()
268 dependency_map.at(node_id).emplace(node_input.node_id()); in SortParams()
771 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in RegisterPadNode() local
772 node_input.set_node_id(id); in RegisterPadNode()
773 node_input.set_output_port(0); in RegisterPadNode()
881 GraphTransferNodeInput& node_input = *node_input_info->add_node_input(); in AddNodeInputByInputIndex() local
882 node_input.set_node_id(src_id); in AddNodeInputByInputIndex()
883 node_input.set_output_port(port); in AddNodeInputByInputIndex()
897 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in AppendNodeInputParams() local
[all …]
|
| D | hexagon_control_wrapper.cc | 180 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in SetupGraph() local
181 node_input.set_node_id(node_info->node_id()); in SetupGraph()
182 node_input.set_output_port(port); in SetupGraph()
215 const GraphTransferNodeInput& node_input = input_params.node_input(i); in SetupGraph() local
216 node_ids[i] = node_input.node_id() + NODE_ID_OFFSET; in SetupGraph()
217 ports[i] = node_input.output_port(); in SetupGraph()
|
| D | graph_transferer_test.cc | 247 for (const GraphTransferNodeInput& node_input : in SanityCheckNodes() local
248 input_params->node_input()) { in SanityCheckNodes()
249 EXPECT_GE(node_input.output_port(), 0); in SanityCheckNodes()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 45 for (const std::string& node_input : node.input()) { in AddNodeToGraph() local
48 if (StrContains(node_input, cluster_name) && in AddNodeToGraph()
56 new_node->add_input(node_input); in AddNodeToGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 196 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()
197 const auto& input = ParseNodeConnection(node_input); in GetUniqueSignature()
200 node_input = *new_name + ParseOutputNode(node_input); in GetUniqueSignature()
217 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()
218 auto parsed_name = ParseNodeConnection(node_input); in FuseFunctionNodes()
225 node_input = in FuseFunctionNodes()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_test.py | 142 for (input_index, node_input) in enumerate(node.input):
143 if node_input == "b":
145 if node_input == "y":
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_impl.py | 341 for node_input in node.input:
342 if "^" in node_input:
343 control_input_names.add(node_input.replace("^", ""))
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_transfer_info.proto | 35 repeated GraphTransferNodeInput node_input = 2; field
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 927 const string& node_input = node_def->input(i); in AddControlDependencies() local
928 if (node_input[0] != '^') { in AddControlDependencies()
933 if (node_input == input) { in AddControlDependencies()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2170 std::vector<NodeInput> node_input{ in TEST_P() local
2214 for (int i = 0; i < node_input.size(); i++) { in TEST_P()
2235 node_input[i].name, in TEST_P()
2244 partial_input_shape[1] = node_input[i].dims[1]; in TEST_P()
2246 AddTestTensor(node_input[i].name, node_input[i].dims, tf_type_, in TEST_P()
2247 node_input[i].val, partial_input_shape, in TEST_P()
2251 AddTestWeights(node_input[i].name, node_input[i].dims, in TEST_P()
2252 node_input[i].val, tf_type_); in TEST_P()
2255 TestOpConverter("my_batchnorm", node_def, node_input[0].dims, in TEST_P()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 1832 GraphView::InputPort node_input(&node, 0); in NodeImplicitlyReadsNonResourceVariable() local
1834 graph_view_.GetRegularFanin(node_input); in NodeImplicitlyReadsNonResourceVariable()
|