Searched refs:current_tensor (Results 1 – 2 of 2) sorted by relevance
223 Tensor current_tensor(type_id, shape, input_host_data, size_in_byte); in GetString()226 auto out = current_tensor.data().ToString(type_id, shape, true); in GetString()234 return current_tensor.ToStringNoLimit(); in GetString()
254 auto current_tensor = input_tensors[input_index]; in RunOpMallocPre() local255 MS_EXCEPTION_IF_NULL(current_tensor); in RunOpMallocPre()256 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpMallocPre()269 current_tensor->set_device_address(device_address); in RunOpMallocPre()270 current_tensor->set_sync_status(kNeedSyncHostToDevice); in RunOpMallocPre()405 auto current_tensor = input_tensors[input_index]; in RunOpAssignInputMemory() local406 MS_EXCEPTION_IF_NULL(current_tensor); in RunOpAssignInputMemory()407 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpAssignInputMemory()