Home
last modified time | relevance | path

Searched refs:GetForPlatform (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.cc85 TF_ASSIGN_OR_RETURN(auto compiler, Compiler::GetForPlatform(platform)); in CreateBackend()
90 TransferManager::GetForPlatform(platform)); in CreateBackend()
92 ComputationPlacer::GetForPlatform(platform)); in CreateBackend()
Dplatform_util.cc69 auto compiler_status = Compiler::GetForPlatform(platform); in GetSupportedPlatforms()
124 TF_RETURN_IF_ERROR(Compiler::GetForPlatform(platform).status()); in GetPlatform()
Dcomputation_placer.h99 static StatusOr<ComputationPlacer*> GetForPlatform(
Dcompiler.cc82 /* static */ StatusOr<Compiler*> Compiler::GetForPlatform( in GetForPlatform() function in xla::Compiler
Dcompile_only_service.cc53 TF_ASSIGN_OR_RETURN(auto compiler, Compiler::GetForPlatform(platform)); in NewService()
Dcomputation_placer.cc157 /* static */ StatusOr<ComputationPlacer*> ComputationPlacer::GetForPlatform( in GetForPlatform() function in xla::ComputationPlacer
Dtransfer_manager.cc210 Compiler::GetForPlatform(stream->parent()->platform())); in ReadDynamicShapes()
265 /* static */ StatusOr<TransferManager*> TransferManager::GetForPlatform( in GetForPlatform() function in xla::TransferManager
Dtransfer_manager.h295 static StatusOr<TransferManager*> GetForPlatform(
Dcompiler.h287 static StatusOr<Compiler*> GetForPlatform(const se::Platform* platform);
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager_interface.cc30 auto tm = xla::TransferManager::GetForPlatform(platform); in GetRegisteredTpuTransferManager()
Dtpu_executable_interface.cc36 TF_ASSIGN_OR_RETURN(auto transfer_manager, TransferManager::GetForPlatform( in PopulateResultTupleBuffers()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc158 TF_ASSIGN_OR_RETURN(auto compiler, xla::Compiler::GetForPlatform( in UpdateMetadata()
162 xla::TransferManager::GetForPlatform(stream->parent()->platform())); in UpdateMetadata()
206 TF_ASSIGN_OR_RETURN(auto compiler, xla::Compiler::GetForPlatform( in UpdateDynamicInputs()
259 xla::TransferManager::GetForPlatform(stream->parent()->platform())); in UpdateDynamicInputs()
278 xla::TransferManager::GetForPlatform(stream->parent()->platform())); in CreateOutputTuple()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc106 TransferManager::GetForPlatform(platform)); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc39 xla::Compiler::GetForPlatform(platform.ValueOrDie()); in BuildXlaCompilationCache()
Dxla_launch_util.cc466 xla::TransferManager::GetForPlatform(stream->parent()->platform())); in PopulateOutputs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc133 TransferManager::GetForPlatform(local_client_->platform()).ValueOrDie(); in LocalClientTestBase()
Ddynamic_ops_test.cc760 TransferManager::GetForPlatform(platform).ValueOrDie(); in BM_DynamicSlice()
Dlocal_client_execute_test.cc956 TransferManager::GetForPlatform(platform).ValueOrDie(); in BM_LocalClientOverhead()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc287 xla::TransferManager::GetForPlatform(stream->parent()->platform())); in UpdateDynamicInputs()