Searched refs:XlaCompilationDevice (Results 1 – 5 of 5) sorted by relevance
75 XlaCompilationDevice::XlaCompilationDevice(const SessionOptions& options, in XlaCompilationDevice() function in tensorflow::XlaCompilationDevice84 XlaCompilationDevice::~XlaCompilationDevice() {} in ~XlaCompilationDevice()86 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()90 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()116 Status XlaCompilationDevice::Sync() { return Status::OK(); } in Sync()118 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_;
456 XlaCompilationDevice* device_; // Owned by device_mgr_
116 XlaCompilationDevice* device, FunctionLibraryRuntime* flib, in ExecuteGraph()427 device_(new XlaCompilationDevice(SessionOptions(), options_.device_type)), in XlaCompiler()