Searched refs:XlaComputationLaunchContext (Results 1 – 5 of 5) sorted by relevance
125 class XlaComputationLaunchContext {135 XlaComputationLaunchContext(xla::LocalClient* client,
56 XlaComputationLaunchContext launch_context( in Run()155 args = XlaComputationLaunchContext::BuildXlaCompilerArguments( in Compile()
189 XlaComputationLaunchContext::XlaComputationLaunchContext( in XlaComputationLaunchContext() function in tensorflow::XlaComputationLaunchContext220 XlaComputationLaunchContext::PopulateInputs( in PopulateInputs()420 Status XlaComputationLaunchContext::PopulateOutputs( in PopulateOutputs()579 XlaComputationLaunchContext::BuildXlaCompilerArguments( in BuildXlaCompilerArguments()
117 XlaComputationLaunchContext::BuildXlaCompilerArguments( in GetCompilerIr()
209 XlaComputationLaunchContext::BuildXlaCompilerArguments( in CompileToLocalExecutable()257 XlaComputationLaunchContext launch_context( in Compute()475 XlaComputationLaunchContext launch_context( in Compute()