| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.h | 61 class ExecutionInput {
63 explicit ExecutionInput(xla::Shape shape) : buffers_(std::move(shape)) { in ExecutionInput() function
67 ExecutionInput(xla::Shape shape, xla::Shape host_shape) in ExecutionInput() function
72 explicit ExecutionInput(ShapeTree<MaybeOwningDeviceMemory> buffers) in ExecutionInput() function
77 ExecutionInput(ShapeTree<MaybeOwningDeviceMemory> buffers, in ExecutionInput() function
83 ExecutionInput(ExecutionInput&&) = default;
85 ~ExecutionInput();
87 ExecutionInput& operator=(ExecutionInput&&) = default;
290 std::vector<ExecutionInput> arguments,
295 std::vector<ExecutionInput> arguments,
[all …]
|
| D | executable.cc | 36 ExecutionInput::~ExecutionInput() { in ~ExecutionInput()
45 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape()
56 void ExecutionInput::SetUnownedBuffer(const ShapeIndex& index, in SetUnownedBuffer()
62 StatusOr<ShapedBuffer> ExecutionInput::ToShapedBuffer( in ToShapedBuffer()
93 static ExecutionInput MakeMaybeOwningDeviceMemoryTree( in MakeMaybeOwningDeviceMemoryTree()
95 ExecutionInput result(shaped_buffer.on_device_shape()); in MakeMaybeOwningDeviceMemoryTree()
107 std::vector<ExecutionInput> args; in ExecuteAsyncOnStream()
120 std::vector<ExecutionInput> arguments, in ExecuteOnStream()
175 std::vector<ExecutionInput> arguments) { in ExecuteOnStreamWrapper()
292 std::vector<ExecutionInput> arguments) { in ExecuteAsyncOnStreamWrapper()
[all …]
|
| D | hlo_runner.cc | 121 static std::vector<ExecutionInput> ExecutionInputsFromScopedShapedBuffers( in ExecutionInputsFromScopedShapedBuffers()
125 std::vector<ExecutionInput> execution_inputs; in ExecutionInputsFromScopedShapedBuffers()
170 std::vector<ExecutionInput> execution_arguments = in ExecuteWithDeviceBuffers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 60 std::vector<ExecutionInput> arguments,
106 absl::Span<ExecutionInput const> arguments);
122 absl::Span<ExecutionInput> arguments);
|
| D | cpu_executable.cc | 93 absl::Span<ExecutionInput const> arguments, in MemoryForAllocation()
129 absl::Span<ExecutionInput const> arguments) { in CreateBufferTable()
228 absl::Span<ExecutionInput> arguments) { in CreateResultShapedBuffer()
263 ExecutionInput& input = arguments[alias->parameter_number]; in CreateResultShapedBuffer()
327 std::vector<ExecutionInput> arguments, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 56 StatusOr<ExecutionOutput> Run(std::vector<ExecutionInput> arguments,
67 StatusOr<ExecutionOutput> RunAsync(std::vector<ExecutionInput> arguments,
79 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options);
|
| D | local_client.cc | 187 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in Run()
190 for (const ExecutionInput& arg : arguments) { in Run()
282 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
323 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
326 for (const ExecutionInput& arg : arguments) { in RunAsync()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 49 std::vector<ExecutionInput> arguments,
69 std::vector<ExecutionInput>* arguments, se::Stream* stream,
|
| D | tpu_executable_interface.cc | 57 std::vector<ExecutionInput>* arguments, se::Stream* stream, in AllocateOutputMemoryWithInputReuse()
116 ExecutionInput& input = (*arguments)[alias->parameter_number]; in AllocateOutputMemoryWithInputReuse()
170 std::vector<ExecutionInput> arguments, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 116 std::vector<ExecutionInput> arguments,
125 absl::Span<ExecutionInput>>;
|
| D | gpu_executable.cc | 318 return absl::get<absl::Span<ExecutionInput>>(arguments)[param_no] in BufferForAllocation()
400 std::vector<ExecutionInput> arguments, in ExecuteAsyncOnStream()
480 absl::get<absl::Span<ExecutionInput>>(arguments); in ExecuteAsyncOnStreamImpl()
481 ExecutionInput& input = in ExecuteAsyncOnStreamImpl()
561 if (auto args = absl::get_if<absl::Span<ExecutionInput>>(&arguments)) { in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.h | 44 std::vector<xla::ExecutionInput> arguments,
|
| D | tpu_execute.cc | 141 xla::ExecutionInput* mem, in FixTupleTableAsync()
196 std::vector<xla::ExecutionInput>* runtime_inputs, in UpdateDynamicInputs()
416 std::vector<xla::ExecutionInput> arguments, in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.h | 42 std::vector<ExecutionInput> arguments,
|
| D | executable_base.cc | 43 std::vector<ExecutionInput> arguments, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 312 xla::StatusOr<std::vector<xla::ExecutionInput>> GetArgumentsBuffers( in GetArgumentsBuffers()
319 std::vector<xla::ExecutionInput> arguments; in GetArgumentsBuffers()
338 TF_ASSIGN_OR_RETURN(xla::ExecutionInput exec_input, in GetArgumentsBuffers()
|
| D | xrt_util.h | 88 xla::StatusOr<std::vector<xla::ExecutionInput>> GetArgumentsBuffers(
|
| D | xrt_state.h | 231 xla::StatusOr<xla::ExecutionInput> ToExecutionInput(
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 204 static void PopulateExecutionInputBuffer(xla::ExecutionInput& execution_input, in PopulateExecutionInputBuffer()
219 xla::StatusOr<std::vector<xla::ExecutionInput>>
226 std::vector<xla::ExecutionInput> arguments; in PopulateInputs()
269 xla::ExecutionInput& execution_input = arguments.back(); in PopulateInputs()
|
| D | xla_launch_util.h | 156 xla::StatusOr<std::vector<xla::ExecutionInput>> PopulateInputs(
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer.h | 163 ExecutionInput* execution_input,
|
| D | pjrt_stream_executor_client.cc | 566 ExecutionInput* execution_input, in AddToInput()
1473 ExecutionInput execution_input;
1516 ExecutionInput execution_input(on_device_shape); in MakeTupleHelper()
1634 StatusOr<std::vector<ExecutionInput>>
1640 std::vector<ExecutionInput> execution_inputs; in MakeExecutionInputsAndWaitForEvents()
1658 ExecutionInput& execution_input = execution_inputs.back(); in MakeExecutionInputsAndWaitForEvents()
1744 std::vector<ExecutionInput> execution_inputs, in EnqueueExecution()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 89 std::vector<ExecutionInput> args; in RunAndCheck()
122 args.emplace_back(ExecutionInput(std::move(owned_buffers))); in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 203 std::vector<xla::ExecutionInput>* execution_inputs, in UpdateDynamicInputs()
214 xla::ExecutionInput* execution_input = &(*execution_inputs)[i]; in UpdateDynamicInputs()
306 std::vector<xla::ExecutionInput> execution_inputs, in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | aliasing.md | 55 [`ExecutionInput`](https://www.tensorflow.org/code/tensorflow/compiler/xla/service/executable.h),
|