| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | map_tensor_py.cc | 59 void MapTensorPy::UpdateFromNumpy(const MapTensorPtr &map_tensor, in UpdateFromNumpy()
72 …::tuple<py::array, py::array, py::array> MapTensorPy::ExportAsNumpy(const MapTensorPtr &map_tensor, in ExportAsNumpy()
80 std::tuple<py::bytes, py::bytes, py::bytes> MapTensorPy::ExportBytes(const MapTensorPtr &map_tensor… in ExportBytes()
87 …:array, py::array, py::array, bool> MapTensorPy::ExportSliceAsNumpy(const MapTensorPtr &map_tensor, in ExportSliceAsNumpy()
97 const MapTensorPtr &map_tensor, int32_t param_key, bool incremental) { in ExportPersistentSliceAsNumpy()
110 static tensor::TensorPtr PyMapTensorGetKeys(const MapTensorPtr &map_tensor) { in PyMapTensorGetKeys()
115 static tensor::TensorPtr PyMapTensorGetValues(const MapTensorPtr &map_tensor) { in PyMapTensorGetValues()
120 …td::pair<tensor::TensorPtr, tensor::TensorPtr> PyMapTensorGetData(const MapTensorPtr &map_tensor) { in PyMapTensorGetData()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | make_map_parameter.cc | 102 …auto map_tensor = std::make_shared<tensor::MapTensor>(key_dtype_id, value_dtype_id, value_shape, d… in MakeMapParameterInfer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_sparse_embedding_operation.cc | 240 auto map_tensor = std::make_shared<tensor::MapTensor>(key_type, value_type, value_shape, nullptr); in NewMapParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | getitem_impl.py | 383 def _map_tensor_getitem(map_tensor, key_tensor): argument
|
| D | setitem_impl.py | 864 def _map_tensor_setitem(map_tensor, key_tensor, value_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 145 auto map_tensor = input_tensor->cast<tensor::MapTensorPtr>(); in UpdateInputNodeDeviceAddress() local
227 void CopyMapTensorDataToDevice(const tensor::MapTensorPtr &map_tensor, const AnfNodePtr &input_node, in CopyMapTensorDataToDevice()
261 auto map_tensor = input_tensors[i]->cast<tensor::MapTensorPtr>(); in CopyParameterDataToDevice() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_sparse_ops.py | 396 def bprop(map_tensor, key_tensor, out, dout): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 3936 def map_tensor_get(map_tensor, key_tensor, insert_default_value=True): argument
3943 def map_tensor_put(map_tensor, key_tensor, value_tensor): argument
3950 def map_tensor_erase(map_tensor, key_tensor): argument
3957 def map_tensor_get_keys(map_tensor): argument
3964 def map_tensor_get_values(map_tensor): argument
3971 def map_tensor_get_data(map_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/debug/ |
| D | trace.cc | 233 …} else if (auto map_tensor = abstract->cast_ptr<abstract::AbstractMapTensor>(); map_tensor != null… in GetNodeTypeOrigin() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/ |
| D | resolve.cc | 147 …auto map_tensor = py::cast<MapTensorPtr>(python_adapter::GetPyObjAttr(param_obj, attr_map_tensor)); in GetParameterValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | data_prepare_actor.cc | 143 const auto &map_tensor = host_tensor->cast<tensor::MapTensorPtr>(); in SyncTensorData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/ |
| D | load_model.cc | 813 auto map_tensor = std::make_shared<tensor::MapTensor>(key_tensor_abs->BuildType()->type_id(), in BuildAbstractMapTensorFromAttrProto() local
922 …auto map_tensor = std::make_shared<tensor::MapTensor>(key_tensor, value_tensor, status_tensor, def… in BuildMapParameterFromMapTensorProto() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ |
| D | abstract_value.cc | 2686 AbstractMapTensor::AbstractMapTensor(const MapTensorPtr &map_tensor) in AbstractMapTensor()
2695 AbstractMapTensor::AbstractMapTensor(const MapTensorPtr &map_tensor, const ValuePtr &ref_key_value) in AbstractMapTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/ |
| D | mindir_exporter.cc | 757 auto map_tensor = param_default->cast<tensor::MapTensorPtr>(); in ConvertMapParameterToMapTensorProto() local
|
| D | anf_ir_dump.cc | 313 …} else if (auto map_tensor = abstract->cast_ptr<abstract::AbstractMapTensor>(); map_tensor != null… in PrintNodeOutputType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.cc | 233 auto map_tensor = input_tensor->cast<tensor::MapTensorPtr>(); in ConstructRunOpParameter() local
|
| D | anf_runtime_algorithm.cc | 2195 …auto map_tensor = std::make_shared<tensor::MapTensor>(*key_type, *value_type, *shape_vector, defau… in CreateMapTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_utils.cc | 1871 …::ConvertMapTensor(const FrontendOpRunInfoPtr &op_run_info, const tensor::MapTensorPtr &map_tensor, in ConvertMapTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/ |
| D | guard_utils.cc | 1386 std::string map_tensor = ToStringIntern(); in ToString() local
|