Searched refs:type_args (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 218 const trans::TypeIdArgs type_args{host.data(), shape_size, type_id_, type, size_}; in SyncDeviceToHost() local
219 sync_ok = trans::TransDataType(type_args, host_ptr); in SyncDeviceToHost()
296 const trans::TypeIdArgs type_args{host.data(), shape_size, type_id_, type, size}; in SyncDeviceToHostAndConvertFormatBasedOnTransData() local
297 sync_ok = trans::TransDataType(type_args, host_ptr); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
342 const trans::TypeIdArgs type_args{host.data(), shape_size, type_id_, type, size}; in SyncDeviceToHostAndConvertFormat() local
343 sync_ok = trans::TransDataType(type_args, host_ptr); in SyncDeviceToHostAndConvertFormat()
383 const trans::TypeIdArgs type_args{host_ptr, shape_size, type, type_id_, size}; in SyncHostToDevice() local
385 sync_ok = trans::TransDataType(type_args, host_tmp.data()); in SyncHostToDevice()
450 const trans::TypeIdArgs type_args{host_ptr, shape_size, type, type_id_, size}; in ConvertFormatAndSyncHostToDevice() local
452 sync_ok = trans::TransDataType(type_args, host_tmp.data()); in ConvertFormatAndSyncHostToDevice()
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 41 def _args_type_validator_check(*type_args, **type_kwargs): argument
46 bound_types = sig.bind_partial(*type_args, **type_kwargs).arguments
|
| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 940 def args_type_check(*type_args, **type_kwargs): argument
945 bound_types = sig.bind_partial(*type_args, **type_kwargs).arguments
|