Searched defs:device_tensors (Results 1 – 3 of 3) sorted by relevance
128 std::vector<DeviceTensor *> device_tensors; in FillDataBuffer() local138 auto &device_tensors = buffers_.back(); in SendMemoryAllocReq() local144 auto &device_tensors = buffers_.front(); in SendMemoryFreeReq() local157 auto &device_tensors = buffers_.back(); in OnMemoryAllocFinish() local223 std::vector<DeviceTensor *> device_tensors; in FillDataBuffer() local234 auto &device_tensors = buffers_.back(); in SendMemoryAllocReq() local245 auto &device_tensors = buffers_.front(); in SendMemoryFreeReq() local265 auto &device_tensors = buffers_.back(); in OnMemoryAllocFinish() local
444 const auto &device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in PrepareDataForWeightNode() local478 auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in PrepareDataForControlWeightNode() local
1597 …auto device_tensors = DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second.get()); in LinkDeviceTensorStoreForAutoMonadActor() local2611 const auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in DumpDeviceTensorStore() local2633 const auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in DumpDeviceTensorStore() local