Home
last modified time | relevance | path

Searched defs:tuple_index (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/
D_compile_utils.py384 def tensor_itemset_by_tuple_with_number(data, tuple_index, nubmer_value): argument
406 def _transform_ellipsis_to_slice(data, tuple_index, op_name): argument
504 def _expand_data_dims(data, tuple_index): argument
701 def judge_tuple_index_dim(data, tuple_index): argument
713 def tensor_index_by_tuple(data, tuple_index): argument
754 def cal_tuple_slice_mask(data_shape, tuple_index): argument
768 def _generate_indices_from_tuple_of_tensor(tuple_index, op_name): argument
793 def _generate_indices_from_tuple(data, tuple_index, op_name, fancy_position): argument
1007 def tensor_setitem_by_tuple_with_number(data, tuple_index, value): argument
1069 def tensor_setitem_by_tuple_with_tensor(data, tuple_index, value): argument
[all …]
Dsetitem_impl.py323 def _tensor_setitem_by_tuple_with_number(data, tuple_index, value): argument
344 def _tensor_setitem_by_tuple_with_tensor(data, tuple_index, value): argument
365 def _tensor_setitem_by_tuple_with_tuple(data, tuple_index, value): argument
387 def _tensor_setitem_by_tuple_with_list(data, tuple_index, value): argument
Dgetitem_impl.py368 def _tensor_getitem_by_tuple(data, tuple_index): argument
D_constexpr_utils.py838 def use_copy_slice(tuple_index): argument
/third_party/python/Objects/clinic/
Dtupleobject.c.h21 tuple_index(PyTupleObject *self, PyObject *const *args, Py_ssize_t nargs) in tuple_index() function
/third_party/mindspore/mindspore-src/source/tests/st/fallback/
Dtest_graph_fallback_print.py620 def judge_tuple_index_dim_lambda(data, tuple_index): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Ddecrease_transfer_precision.cc116 auto tuple_index = ObtainGetItemIndex(in_node); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc535 const ShapeVector &data_shape, const std::vector<TensorIndex> &tuple_index) { in GetStrideInfoFromTuple()
661 …x::TensorGetitemByTuple(const ShapeVector &data_shape, const std::vector<TensorIndex> &tuple_index, in TensorGetitemByTuple()
803const std::vector<TensorIndex> &tuple_index, in GenerateIndicesFromTupleOfTensor()
1020const std::vector<TensorIndex> &tuple_index, int64_t py_fancy_position, in GenerateIndicesFromTuple()
1454 TensorIndex tuple_index = SequenceToTensor(tensor_index, data_shape[0]); in GetItemByList() local
/third_party/mindspore/mindspore-src/source/tests/st/graph_syntax/statements/
Dtest_graph_raise_with_variable.py654 def judge_tuple_index_dim(data, tuple_index, x): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dtransform.cc353 const size_t tuple_index = 2; in AddSwitchLayer() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/shard/
Dshard.cc307 auto tuple_index = to_insert_cnode->input(2); in SetInputLayout() local
/third_party/python/Objects/
Dgenericaliasobject.c172 tuple_index(PyObject *self, Py_ssize_t len, PyObject *item) in tuple_index() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_assigned_parallel.cc110 auto tuple_index = GetTupleGetItemIndex(cnode); in FindPrevLayout() local
Dstep_parallel_utils.cc905 auto tuple_index = LongToSize(GetValue<int64_t>(GetValueNode(cnode->input(2)))); in BFSParallelCareNode() local
Dstep_parallel.cc2313 auto tuple_index = GetTupleGetItemIndex(cnode); in FindPrevLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc603 CNodePtr PipelineTransformer::GraphOutNode(const AnfNodePtr &node, int tuple_index) { in GraphOutNode()
615 OperatorInfoPtr PipelineTransformer::CreateOpInfo(const CNodePtr &cnode, int tuple_index = 0) { in CreateOpInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dtensor_index.cc470 const std::vector<int64_t> &tuple_index_types, size_t tuple_index) { in NormalizeStrideInfoFromTuple()
Dcomposite.cc1794 constexpr size_t tuple_index = 0; in GenerateFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dgraph_build.cc2953 for (size_t tuple_index = 0; tuple_index < iterable_nodes.size(); ++tuple_index) { in TraceRunForIterZip() local