Searched defs:tensor_address (Results 1 – 6 of 6) sorted by relevance
254 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareDataForHostTensorQueue() local567 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareHostTensorQueueForControlNode() local
182 auto tensor_address = tensor->device_address(); in LoadInputData() local
344 … auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in LoadInputData() local
330 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in TensorNeedSync() local
326 auto tensor_address = tensor->device_address(); in BindInputTensorAddressPtr() local
936 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignExtraStaticMem() local1402 auto tensor_address = tensor->device_address(); in InitGraphInputTensors() local