Searched refs:execution_output (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 279 xla::StatusOr<xla::ExecutionOutput> execution_output; in Compute() local
281 execution_output = in Compute()
284 execution_output = in Compute()
287 OP_REQUIRES(ctx, execution_output.ok(), execution_output.status()); in Compute()
293 ctx, compilation_result, execution_output->ConsumeResult(), in Compute()
515 xla::StatusOr<xla::ExecutionOutput> execution_output; in Compute() local
517 execution_output = in Compute()
520 execution_output = closure.executable()->RunAsync( in Compute()
523 OP_REQUIRES(ctx, execution_output.ok(), execution_output.status()); in Compute()
542 ctx, closure.compilation_result(), execution_output->ConsumeResult(), in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 90 xla::ExecutionOutput execution_output = run_result.ConsumeValueOrDie(); in Run() local
96 ctx, result, execution_output.ConsumeResult(), in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1783 ExecutionOutput& execution_output = result_buffer_or_status.ValueOrDie(); in EnqueueExecution() local
1792 execution_output.ConsumeToBeReleased(); in EnqueueExecution()
|