Searched defs:DeviceTensor (Results 1 – 6 of 6) sorted by relevance
28 using DeviceTensor = mindspore::device::DeviceAddress; variable
33 class DeviceTensor; variable
28 DeviceTensor::DeviceTensor(const TensorShape &shape, const DataType &type) in DeviceTensor() function in mindspore::dataset::DeviceTensor
37 using DeviceTensor = mindspore::device::DeviceAddress; variable
114 OpContext<DeviceTensor> *) { in FreeMemory()
377 void KernelActor::PreLaunchKernel(OpContext<DeviceTensor> *) { in PreLaunchKernel()