Home
last modified time | relevance | path

Searched refs:XlaComputationLaunchContext (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc194 XlaComputationLaunchContext::XlaComputationLaunchContext( in XlaComputationLaunchContext() function in tensorflow::XlaComputationLaunchContext
207 void XlaComputationLaunchContext::PopulateInputs( in PopulateInputs()
260 Status XlaComputationLaunchContext::PopulateOutputs( in PopulateOutputs()
443 Status XlaComputationLaunchContext::BuildXlaCompilerArguments( in BuildXlaCompilerArguments()
Dxla_launch_util.h134 class XlaComputationLaunchContext {
144 XlaComputationLaunchContext(xla::LocalClient* client,
Dxla_compile_on_demand_op.cc59 XlaComputationLaunchContext launch_context( in Run()
204 TF_RETURN_IF_ERROR(XlaComputationLaunchContext::BuildXlaCompilerArguments( in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc315 TF_RETURN_IF_ERROR(XlaComputationLaunchContext::BuildXlaCompilerArguments( in CompileToLocalExecutable()
351 XlaComputationLaunchContext launch_context( in Compute()
520 XlaComputationLaunchContext launch_context( in Compute()