Home
last modified time | relevance | path

Searched refs:to_merge (Results 1 – 17 of 17) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/utils/
Dsubgraph_utils.cpp39 Node* to_merge, in ValueMapper()
42 last_uses_ = gatherLastUses(to_merge->outputs()); in ValueMapper()
46 WithInsertPoint guard(to_merge); in ValueMapper()
47 auto g = to_merge->owningGraph(); in ValueMapper()
51 for (size_t i = 0; i < to_merge->outputs().size(); ++i) { in ValueMapper()
52 Value* existing = to_merge->outputs().at(i); in ValueMapper()
54 to_merge->outputs().at(i)); in ValueMapper()
100 Node* to_merge, in executeSubgraphMergeAndUpdateAliasing() argument
109 ValueMapper vm(to_merge, db, existing); in executeSubgraphMergeAndUpdateAliasing()
418 Node* to_merge, in mergeNodeIntoSubgraphAndUpdateAliasing() argument
[all …]
Dsubgraph_utils.h27 Node* to_merge,
45 Node* to_merge,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc78 bool MergeSharding(const HloSharding& old, HloSharding* to_merge, in MergeSharding() argument
81 CHECK(to_merge->IsTuple()); in MergeSharding()
85 MergeSharding(old.tuple_elements()[i], &to_merge->tuple_elements()[i], in MergeSharding()
91 !to_merge->HasPartialReplication() || in MergeSharding()
93 to_merge->tile_assignment().num_elements()) { in MergeSharding()
94 return IsShardingMoreSpecific(*to_merge, old); in MergeSharding()
99 /*minimum_tiles=*/std::max(old.NumTiles(), to_merge->NumTiles()) + 1, in MergeSharding()
100 to_merge)) { in MergeSharding()
103 return IsShardingMoreSpecific(*to_merge, old); in MergeSharding()
106 bool MergeShardingIfCompatible(const HloSharding& to_merge, in MergeShardingIfCompatible() argument
[all …]
Dhlo_sharding_util.h50 bool MergeSharding(const HloSharding& old, HloSharding* to_merge,
55 bool MergeShardingIfCompatible(const HloSharding& to_merge,
Dsharding_propagation.cc925 const HloSharding& to_merge, absl::Span<const int64_t> unspecified_dims, in RefineManualAutoShardingFromAuto() argument
937 to_merge, unspecified_dims); in RefineManualAutoShardingFromAuto()
997 const HloSharding& to_merge, absl::Span<const int64_t> unspecified_dims, in RefineManualAutoShardingFromManual() argument
999 if (!to_merge.IsManualSubgroup() || !manual_sharding->IsManualSubgroup() || in RefineManualAutoShardingFromManual()
1007 to_merge, unspecified_dims); in RefineManualAutoShardingFromManual()
/external/pytorch/torch/csrc/jit/runtime/static/
Dfusion.cpp171 Node* to_merge, in tryMerge() argument
173 if (!canMerge(fusion_group, to_merge, aliasDb)) { in tryMerge()
177 std::vector<Node*> nodes_to_merge = {to_merge}; in tryMerge()
179 if (to_merge->kind() == aten::cat) { in tryMerge()
180 Node* listconstruct = to_merge->input(0)->node(); in tryMerge()
/external/python/cpython3/Python/
Dbrc.c103 merge_queued_objects(_PyObjectStack *to_merge) in merge_queued_objects() argument
106 while ((ob = _PyObjectStack_Pop(to_merge)) != NULL) { in merge_queued_objects()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_formation.cc72 bool CanMergeIntoCluster(const Cluster& c, Operation* to_merge) { in CanMergeIntoCluster() argument
73 return llvm::all_of(to_merge->getOperands(), [&](Value operand) { in CanMergeIntoCluster()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc859 std::vector<void*> to_merge; in MergeTimestampedChunks() local
877 to_merge.push_back(c->ptr); in MergeTimestampedChunks()
884 to_merge.push_back(c->ptr); in MergeTimestampedChunks()
886 to_merge.push_back(c->ptr); in MergeTimestampedChunks()
898 for (int ci = 0, end = to_merge.size(); ci < end; ++ci) { in MergeTimestampedChunks()
899 void* ptr = to_merge[ci]; in MergeTimestampedChunks()
/external/grpc-grpc/test/cpp/qps/
Dclient.h178 std::vector<Histogram> to_merge(threads_.size()); in Mark()
182 threads_[i]->BeginSwap(&to_merge[i], &to_merge_status[i]); in Mark()
187 latencies.Merge(to_merge[i]); in Mark()
/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp777 std::optional<Node*> tryMerge(Node* fusion_group, Node* to_merge) { in tryMerge() argument
778 if (!canMerge(fusion_group, to_merge)) { in tryMerge()
782 std::vector<Node*> nodes_to_merge = {to_merge}; in tryMerge()
784 if (to_merge->kind() == aten::cat) { in tryMerge()
785 Node* listconstruct = to_merge->input(0)->node(); in tryMerge()
/external/cronet/stable/testing/libfuzzer/research/domatolpm/
Dgenerator.py602 to_merge = collections.defaultdict(set)
616 to_merge[elt].add(name)
618 for parent, childs in to_merge.items():
647 return len(to_merge) > 0
/external/cronet/tot/testing/libfuzzer/research/domatolpm/
Dgenerator.py602 to_merge = collections.defaultdict(set)
616 to_merge[elt].add(name)
618 for parent, childs in to_merge.items():
647 return len(to_merge) > 0
/external/python/ipaddress/
Dipaddress.py402 to_merge = list(addresses)
404 while to_merge:
405 net = to_merge.pop()
413 to_merge.append(supernet)
/external/python/cpython3/Lib/
Dipaddress.py280 to_merge = list(addresses)
282 while to_merge:
283 net = to_merge.pop()
291 to_merge.append(supernet)
/external/python/cpython3/Objects/
Dtypeobject.c2715 set_mro_error(PyObject **to_merge, Py_ssize_t to_merge_size, int *remain) in set_mro_error() argument
2724 PyObject *L = to_merge[i]; in set_mro_error()
2766 pmerge(PyObject *acc, PyObject **to_merge, Py_ssize_t to_merge_size) in pmerge() argument
2789 PyObject *cur_tuple = to_merge[i]; in pmerge()
2805 PyObject *j_lst = to_merge[j]; in pmerge()
2814 PyObject *j_lst = to_merge[j]; in pmerge()
2825 set_mro_error(to_merge, to_merge_size, remain); in pmerge()
2892 PyObject **to_merge = PyMem_New(PyObject *, n + 1); in mro_implementation_unlocked() local
2893 if (to_merge == NULL) { in mro_implementation_unlocked()
2900 to_merge[i] = lookup_tp_mro(base); in mro_implementation_unlocked()
[all …]
/external/libaom/av1/encoder/
Dpass2_strategy.c1667 int to_merge = 0; in find_blending_regions() local
1693 to_merge = 1; in find_blending_regions()
1697 if (to_merge) { in find_blending_regions()