Home
last modified time | relevance | path

Searched refs:mutable_backend (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc336 Backend* LocalClient::mutable_backend() { in mutable_backend() function in xla::LocalClient
337 return local_service_->mutable_backend(); in mutable_backend()
359 std::move(executable), local_service_->mutable_backend(), in Compile()
376 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()
384 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()
425 mutable_backend()->BorrowStream(device_ordinal)); in TransferToLocalServer()
Dlocal_client.h187 Backend* mutable_backend();
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_device.h47 xla::Backend* backend() { return client_->mutable_backend(); } in backend()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc264 xla::Backend* backend = GetOrCreateClient().ValueOrDie()->mutable_backend(); in GetAllocatorLocked()
294 xla::Backend* backend = client->mutable_backend(); in GetDeviceContextLocked()
Dxla_device_context.cc227 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.h184 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc206 stream = local_client_->mutable_backend() in ExecuteLocally()
Dxla_hlo_profile_test.cc136 Backend* backend = service->mutable_backend(); in ExecuteAndFetchProfile()
Dtransfer_manager_test.cc47 stream_ptr_ = local_client_->mutable_backend()
Dlocal_client_execute_test.cc775 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()
951 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
Ddynamic_ops_test.cc755 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()