Searched refs:tensor_dict (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | composite_test.cc | 264 …abstract::AbstractDictionaryPtr tensor_dict = std::make_shared<abstract::AbstractDictionary>(tenso… in TEST_F() local
266 AbstractBasePtrList args_spec_list = {fn_arg, tensor_tuple, tensor_dict}; in TEST_F()
291 …abstract::AbstractDictionaryPtr tensor_dict = std::make_shared<abstract::AbstractDictionary>(tenso… in TEST_F() local
293 …AbstractBasePtrList args_spec_list = {fn_arg, tensor_dict, tensor_tuple, tensor_dict, tensor_tuple… in TEST_F()
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_initializer.py | 159 …tensor_dict = {tensor1: gain, tensor2: None, tensor3: gain, tensor4: None, tensor5: None, tensor6:…
161 for tensor, gain_value in tensor_dict.items():
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | memory_usage_parser.py | 371 tensor_dict = self._get_tensor_dict(node, tensor, t_id)
372 self.breakdowns[index].append(tensor_dict)
|