HomeSort by relevance Sort by last modified time
    Searched refs:src_node (Results 1 - 25 of 25) sorted by null

  /external/tensorflow/tensorflow/core/framework/
node_def_builder.cc 96 NodeDefBuilder& NodeDefBuilder::Input(StringPiece src_node, int src_index,
99 if (arg != nullptr) SingleInput(arg, src_node, src_index, dt);
116 StringPiece src_node, int src_index,
118 AddInput(src_node, src_index);
173 void NodeDefBuilder::AddInput(StringPiece src_node, int src_index) {
174 if (src_node.empty()) {
176 } else if (src_node[0] == '^') {
178 strings::StrCat("Non-control input starting with ^: ", src_node));
180 node_def_.add_input(strings::StrCat(src_node, ":", src_index));
182 node_def_.add_input(string(src_node));
    [all...]
node_def_builder.h 78 NodeDefBuilder& Input(StringPiece src_node, int src_index, DataType dt);
87 // Specify that this node must only run after src_node.
88 NodeDefBuilder& ControlInput(StringPiece src_node);
153 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
158 // Add "src_node:src_index" to the list of inputs in the node_def_.
159 void AddInput(StringPiece src_node, int src_index);
  /external/tensorflow/tensorflow/core/graph/
node_builder.cc 51 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) {
52 inputs_.emplace_back(src_node, src_index);
54 if (GetOutputType(src_node, src_index, &dt)) {
55 def_builder_.Input(src_node->name(), src_index, dt);
85 NodeBuilder& NodeBuilder::ControlInput(Node* src_node) {
86 control_inputs_.emplace_back(src_node);
87 def_builder_.ControlInput(src_node->name());
94 for (const Node* src_node : src_nodes) {
95 def_builder_.ControlInput(src_node->name());
control_flow.cc 94 const Node* src_node = g->source_node(); local
95 ControlFlowInfo& src_info = (*info)[src_node->id()];
96 src_info.frame = src_node;
97 src_info.parent_frame = src_node;
101 ready.push_back(src_node);
collective_order.cc 100 Node* src_node = collective_nodes[src_idx]; local
102 VLOG(1) << "Adding control dependency from node " << src_node->name()
105 (*dependency_edges)[src_node].insert(dst_node);
106 auto& src_paths = all_paths[src_node];
153 Node* src_node = pair.first; local
155 graph->AddControlEdge(src_node, dst_node);
node_builder.h 91 NodeBuilder& Input(Node* src_node, int src_index = 0);
97 // Require that this node run after src_node(s).
98 NodeBuilder& ControlInput(Node* src_node);
graph_constructor.cc 982 Node* src_node; local
1007 inputs.emplace_back(string(id.first), src_node, src_index); local
1070 Node* src_node = gdef_nodes_[e.src_name].node; local
    [all...]
  /external/tensorflow/tensorflow/core/debug/
debug_graph_utils.cc 106 for (Node* src_node : graph->nodes()) {
109 for (const Edge* edge : src_node->out_edges()) {
119 for (int src_output_slot = 0; src_output_slot < src_node->num_outputs();
122 strings::StrCat(src_node->name(), ":", src_output_slot);
143 const DataType src_dt = src_node->output_type(src_output_slot);
145 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node,
151 graph, device_type, memory_type == HOST_MEMORY, src_node->name(),
162 graph->AddEdge(src_node, src_output_slot, copy_node, 0);
192 const bool is_ref = IsRefType(src_node->output_type(src_output_slot));
206 if (!src_node->IsEnter() && !src_node->IsNextIteration())
    [all...]
  /external/tensorflow/tensorflow/core/grappler/utils/
grappler_test.cc 177 const NodeDef* src_node = node_map.GetNode(src); local
179 EXPECT_TRUE(src_node != nullptr) << src << " node not found";
181 return src_node && dst_node && dst_node->input(position) == src_node->name();
  /external/tensorflow/tensorflow/compiler/jit/
encapsulate_subgraphs_pass.cc 773 Node* src_node = edge->src(); local
808 Node* src_node = edge->src(); local
954 const Node* src_node = input_src.first.node; local
1001 const Node* src_node = input_src.first.node; local
1250 const Node* src_node = input.first.node; local
1305 const Node* src_node = output.first.node; local
1979 Node* src_node = in_edge->src(); local
    [all...]
  /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
resolve_cluster.cc 114 for (const std::unique_ptr<tensorflow::NodeDef>& src_node :
117 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get());
  /external/tensorflow/tensorflow/core/nccl/
nccl_rewrite.cc 61 Node* src_node = edge->src(); local
65 int send_dev = src_node->assigned_device_name_index();
67 recv_builder.Input(src_node);
73 .Input(src_node)
  /external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_utils.cc 774 const Node* src_node = in_edge->src(); local
776 node_names.count(src_node->name()) <= 0 && !src_node->IsSource();
779 strings::StrCat(src_node->name(), ":", in_edge->src_output());
780 CHECK_EQ(1, src_node->num_outputs())
782 << src_node->name();
912 const Node* src_node = edge->src(); local
978 Node* src_node = edge->src(); local
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/
process_function_library_runtime.cc 330 Node* src_node = it->src(); local
331 const string* src_device = AssignedOrRequestedDeviceName(*src_node);
333 GetColocationGroup(src_node, &colocation_group);
334 VLOG(3) << "Considering src: " << src_node->name()
338 src_node->IsIdentity()) {
339 src_node = *src_node->in_nodes().begin();
340 src_device = AssignedOrRequestedDeviceName(*src_node);
341 GetColocationGroup(src_node, &colocation_group);
342 VLOG(3) << "Considering src: " << src_node->name(
    [all...]
  /external/mesa3d/src/compiler/nir/
nir_from_ssa.c 386 merge_node *src_node = get_merge_node(src->src.ssa, state); local
387 if (src_node->set != dest_node->set)
388 merge_merge_sets(dest_node->set, src_node->set);
413 merge_node *src_node = get_merge_node(entry->src.ssa, state); local
416 if (src_node->set == dest_node->set)
419 if (!merge_sets_interfere(src_node->set, dest_node->set))
420 merge_merge_sets(src_node->set, dest_node->set);
    [all...]
  /external/e2fsprogs/lib/ext2fs/
blkmap64_rb.c 244 struct rb_node *dest_node, *src_node, *dest_last, **n; local
256 src_node = ext2fs_rb_first(&src_bp->root);
257 while (src_node) {
258 src_ext = node_to_extent(src_node);
278 src_node = ext2fs_rb_next(src_node);
  /external/u-boot/lib/
fdtdec.c 721 int fdtdec_parse_phandle_with_args(const void *blob, int src_node,
734 list = fdt_getprop(blob, src_node, list_name, &size);
764 fdt_get_name(blob, src_node,
775 fdt_get_name(blob, src_node,
792 fdt_get_name(blob, src_node, NULL));
813 fdt_get_name(blob, src_node,
    [all...]
  /external/u-boot/include/
fdtdec.h 202 * @src_node: Offset of device tree node containing a list
215 int fdtdec_parse_phandle_with_args(const void *blob, int src_node,
    [all...]
  /external/vulkan-validation-layers/layers/
descriptor_sets.cpp 1742 auto src_node = GetSetNode(src_set); local
1779 auto src_node = dev_data->GetSetNode(src_set); local
    [all...]
  /device/linaro/bootloader/edk2/AppPkg/Applications/Python/PyMod-2.7.2/Modules/expat/
xmlparse.c     [all...]
  /device/linaro/bootloader/edk2/AppPkg/Applications/Python/Python-2.7.2/Modules/expat/
xmlparse.c     [all...]
  /device/linaro/bootloader/edk2/AppPkg/Applications/Python/Python-2.7.10/Modules/expat/
xmlparse.c     [all...]
  /external/expat/lib/
xmlparse.c     [all...]
  /external/python/cpython2/Modules/expat/
xmlparse.c     [all...]
  /external/python/cpython3/Modules/expat/
xmlparse.c     [all...]

Completed in 2719 milliseconds