Searched refs:XlaCompilationDevice (Results 1 – 6 of 6) sorted by relevance
71 XlaCompilationDevice::XlaCompilationDevice(const SessionOptions& options, in XlaCompilationDevice() function in tensorflow::XlaCompilationDevice80 XlaCompilationDevice::~XlaCompilationDevice() {} in ~XlaCompilationDevice()82 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()101 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()144 Status XlaCompilationDevice::Sync() { return Status::OK(); } in Sync()146 Status XlaCompilationDevice::MakeTensorFromProto( in MakeTensorFromProto()
47 class XlaCompilationDevice : public LocalDevice {49 XlaCompilationDevice(const SessionOptions& options, DeviceType type);51 ~XlaCompilationDevice() override;
58 GraphCompiler(XlaCompilationDevice* device, Graph* graph, in GraphCompiler()81 XlaCompilationDevice* device_;
329 XlaCompilationDevice* device_; // Owned by device_mgr_
123 XlaCompilationDevice* device, FunctionLibraryRuntime* flib, in ExecuteGraph()505 device_(new XlaCompilationDevice(SessionOptions(), options_.device_type)), in XlaCompiler()
295 auto device = absl::make_unique<tensorflow::XlaCompilationDevice>( in CreateDeviceMgr()