Searched refs:VirtualDevice (Results 1 – 3 of 3) sorted by relevance
68 VirtualDevice::VirtualDevice(Env* env, in VirtualDevice() function in tensorflow::VirtualDevice72 Status VirtualDevice::Sync() { return Status::OK(); } in Sync()74 Allocator* VirtualDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()92 Status VirtualDevice::TryGetDeviceContext(DeviceContext** out_context) { in TryGetDeviceContext()
25 class VirtualDevice : public Device {27 VirtualDevice(Env* env, const DeviceAttributes& device_attributes);
437 devices->push_back(absl::make_unique<VirtualDevice>(options.env, attrs)); in CreateDevices()