Searched refs:execution_inputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 90 std::vector<ExecutionInput> execution_inputs; in RunMlirModule() local
95 execution_inputs.emplace_back(shape); in RunMlirModule()
96 execution_inputs.back().SetBuffer({}, MaybeOwningDeviceMemory(arg)); in RunMlirModule()
101 &run_options, std::move(execution_inputs), in RunMlirModule()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 263 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Compute() local
267 OP_REQUIRES_OK(ctx, execution_inputs.status()); in Compute()
282 executable->Run(std::move(*execution_inputs), run_options); in Compute()
285 executable->RunAsync(std::move(*execution_inputs), run_options); in Compute()
485 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs; in Compute() local
500 execution_inputs = launch_context.PopulateInputs( in Compute()
504 OP_REQUIRES_OK(ctx, execution_inputs.status()); in Compute()
518 closure.executable()->Run(std::move(*execution_inputs), run_options); in Compute()
521 std::move(*execution_inputs), run_options); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 71 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Run() local
75 TF_RETURN_IF_ERROR(execution_inputs.status()); in Run()
88 executable->Run(execution_inputs.ConsumeValueOrDie(), run_options); in Run()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 203 std::vector<xla::ExecutionInput>* execution_inputs, in UpdateDynamicInputs() argument
205 TF_RET_CHECK(execution_inputs->size() == compile_time_shapes.size()); in UpdateDynamicInputs()
214 xla::ExecutionInput* execution_input = &(*execution_inputs)[i]; in UpdateDynamicInputs()
306 std::vector<xla::ExecutionInput> execution_inputs, in RunExecutable()
354 &execution_inputs, shape_layouts)); in RunExecutable()
357 executable->Run(std::move(execution_inputs), run_options)); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 125 std::vector<ExecutionInput> execution_inputs; in ExecutionInputsFromScopedShapedBuffers() local
145 execution_inputs.emplace_back(std::move(buffer_tree)); in ExecutionInputsFromScopedShapedBuffers()
147 return execution_inputs; in ExecutionInputsFromScopedShapedBuffers()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1640 std::vector<ExecutionInput> execution_inputs; in MakeExecutionInputsAndWaitForEvents() local
1650 execution_inputs.emplace_back(std::move(tuple_handle->execution_input)); in MakeExecutionInputsAndWaitForEvents()
1652 execution_inputs.reserve(argument_handles.size()); in MakeExecutionInputsAndWaitForEvents()
1657 execution_inputs.emplace_back(handle->on_device_shape()); in MakeExecutionInputsAndWaitForEvents()
1658 ExecutionInput& execution_input = execution_inputs.back(); in MakeExecutionInputsAndWaitForEvents()
1675 return execution_inputs; in MakeExecutionInputsAndWaitForEvents()
1744 std::vector<ExecutionInput> execution_inputs, in EnqueueExecution()
1772 executables_[executable_idx]->RunAsync(std::move(execution_inputs), in EnqueueExecution()
|