Searched refs:ConsumeResult (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 96 ctx, result, execution_output.ConsumeResult(), in Run()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 366 executable, input_tuples, input_output_alias, output.ConsumeResult(), in DoWork()
463 input_output_alias, output.ConsumeResult(), in DoWork()
|
| D | xrt_execute_op.cc | 292 (void)run_result.ConsumeResult().release(); in CreateOutputTuple()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 293 ctx, compilation_result, execution_output->ConsumeResult(), in Compute()
542 ctx, closure.compilation_result(), execution_output->ConsumeResult(), in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.h | 194 ScopedShapedBuffer ConsumeResult() { in ConsumeResult() function
|
| D | executable.cc | 115 return out.ConsumeResult(); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 413 return out.ConsumeResult(); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 767 context, output.ConsumeResult(), executable.output_tensor_shapes(), in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1818 return result_buffer_or_status.ConsumeValueOrDie().ConsumeResult(); in EnqueueExecution()
|