Searched refs:use_node (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/fx/ |
| D | node.py | 620 for use_node in to_process:
621 if not delete_user_cb(use_node):
622 skipped.append(use_node)
632 m._replace_hook(old=self, new=replace_with.name, user=use_node)
634 new_args = map_arg(use_node.args, maybe_replace_node)
635 new_kwargs = map_arg(use_node.kwargs, maybe_replace_node)
638 use_node.__update_args_kwargs(new_args, new_kwargs)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_view.cc | 322 Node* use_node = e->dst(); in SetScopedAllocatorAttrs() local
323 NodeItem* item = node(use_node->id()); in SetScopedAllocatorAttrs()
326 Status s = GetNodeAttr(use_node->attrs(), "_scoped_allocator", in SetScopedAllocatorAttrs()
330 << use_node->name(); in SetScopedAllocatorAttrs()
335 for (const auto& e : use_node->out_edges()) { in SetScopedAllocatorAttrs()
|
| /external/pytorch/torch/fx/passes/ |
| D | split_module.py | 189 def_node: Node, use_node: Optional[Node]
194 used = getattr(use_node, "_fx_partition", None)
198 def_node.name, defined, use_node.name if use_node is not None else "-", used
|
| /external/coreboot/src/vendorcode/cavium/bdk/libbdk-dram/ |
| D | bdk-dram-test.c | 553 int use_node = (flags & BDK_DRAM_TEST_USE_CCPI) ? node ^ 1 : node; in bdk_dram_test()
554 if (bdk_get_running_coremask(use_node) == 0) in bdk_dram_test()
555 bdk_init_cores(use_node, 1); in bdk_dram_test()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_divergence_analysis.c | 81 nir_cf_node *use_node = nir_src_get_block(src)->cf_node.parent; in nir_src_is_divergent() local
85 if (def_node == use_node) in nir_src_is_divergent()
95 for (nir_cf_node *node = use_node; node != NULL; node = node->parent) { in nir_src_is_divergent()
|