Searched refs:launch_context (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 59 XlaComputationLaunchContext launch_context( in Run() local
64 launch_context.PopulateInputs(ctx, result, variables, in Run()
72 for (const xla::ShapedBuffer* arg : launch_context.arguments()) { in Run()
82 executable->Run(launch_context.arguments(), run_options); in Run()
85 TF_RETURN_IF_ERROR(launch_context.PopulateOutputs( in Run()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 351 XlaComputationLaunchContext launch_context( in Compute() local
355 launch_context.PopulateInputs(ctx, kernel, variables, in Compute()
368 auto run_result = executable->Run(launch_context.arguments(), run_options); in Compute()
374 OP_REQUIRES_OK(ctx, launch_context.PopulateOutputs( in Compute()
520 XlaComputationLaunchContext launch_context( in Compute() local
528 launch_context.PopulateInputs( in Compute()
543 closure.executable()->Run(launch_context.arguments(), run_options); in Compute()
551 launch_context.PopulateOutputs( in Compute()
|