Home
last modified time | relevance | path

Searched refs:executable_parameter_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1844 absl::Span<const Shape> executable_parameter_shapes, in MakeExecutionInputsAndWaitForEvents() argument
1859 executable_parameter_shapes[0], argument_handles, in MakeExecutionInputsAndWaitForEvents()
1864 if (argument_handles.size() != executable_parameter_shapes.size()) { in MakeExecutionInputsAndWaitForEvents()
1866 executable_parameter_shapes.size(), in MakeExecutionInputsAndWaitForEvents()
1876 executable_parameter_shapes[i], *transfer_manager, i)); in MakeExecutionInputsAndWaitForEvents()
1877 execution_inputs.emplace_back(executable_parameter_shapes[i]); in MakeExecutionInputsAndWaitForEvents()
H A Dpjrt_stream_executor_client.h811 absl::Span<const Shape> executable_parameter_shapes,