| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/experimental/ |
| D | map_parameter.py | 76 …def __new__(cls, key_dtype=None, value_dtype=None, value_shape=None, key_tensor=None, value_tensor… argument
128 def __getitem__(self, key_tensor): argument
131 def __setitem__(self, key_tensor, value_tensor): argument
143 def _check_map_parameter_args(key_tensor, key_dtype, value_tensor, value_dtype, value_shape): argument
185 def get(self, key_tensor, insert_default_value=True): argument
227 def put(self, key_tensor, value_tensor): argument
241 def erase(self, key_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | map_tensor.h | 45 TensorPtr key_tensor; member
186 const TensorPtr &key_tensor() const { return key_tensor_; } in key_tensor() function
198 void set_key_tensor(const TensorPtr key_tensor) { key_tensor_ = key_tensor; } in set_key_tensor()
|
| D | map_tensor.cc | 215 auto key_tensor = std::make_shared<Tensor>(key_dtype(), key_shape); in Export() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _ps_context.py | 241 def _store_warm_up_ptr_by_tensor_list(param_key, key_tensor, value_tensor, status_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | map_tensor_py.cc | 145 const py::object &permit_filter_obj, const py::object &evict_filter_obj) { in RegMapTensor()
|
| /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/tests/ut/python/parallel/ |
| D | test_auto_parallel_frontend_scheduling_with_pangu_alpha.py | 268 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
302 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| D | test_auto_parallel_pangu_alpha.py | 269 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
303 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/ |
| D | transformer.py | 1127 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
1233 def _check_inputs(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument
1261 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|
| /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/core/proto/ |
| D | mind_ir.proto | 220 required TensorProto key_tensor = 3; field
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 230 auto key_tensor = map_tensor->key_tensor(); in CopyMapTensorDataToDevice() local
|
| /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
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/ |
| D | mindir_exporter.cc | 772 auto &key_tensor = export_data.key_tensor; in ConvertMapParameterToMapTensorProto() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/ |
| D | load_model.cc | 894 auto key_tensor = GenerateTensorPtrFromTensorProto(key_tensor_proto); in BuildMapParameterFromMapTensorProto() local
|