Home | History | Annotate | Download | only in service

Lines Matching defs:Backend

18 #include "tensorflow/compiler/xla/service/backend.h"
65 struct Backend::EigenThreadPoolWrapper {
78 /* static */ StatusOr<std::unique_ptr<Backend>> Backend::CreateBackend(
88 std::unique_ptr<Backend> backend(
89 new Backend(platform, compiler, stream_executors, transfer_manager,
91 return std::move(backend);
94 /* static */ StatusOr<std::unique_ptr<Backend>>
95 Backend::CreateDefaultBackend() {
103 StatusOr<Backend::StreamPtr> Backend::BorrowStream(int device_ordinal) {
108 StatusOr<Backend::StreamPtr> Backend::BorrowStream(
123 Backend::Backend(
142 << "Service found no devices for backend " << platform_->Name() << '.';
156 Backend::~Backend() {}
158 int Backend::default_device_ordinal() const {
162 tensorflow::thread::ThreadPool* Backend::inter_op_thread_pool() const {
166 const Eigen::ThreadPoolDevice* Backend::eigen_intra_op_thread_pool_device()
174 tensorflow::thread::ThreadPool* Backend::eigen_intra_op_thread_pool() const {
181 StatusOr<perftools::gputools::StreamExecutor*> Backend::stream_executor(
198 StatusOr<bool> Backend::devices_equivalent(int device_ordinal_a,
212 Status Backend::ResetDevices() {