Searched refs:executable_parameter_shapes (Results 1 – 2 of 2) sorted by relevance
1798 absl::Span<const Shape> executable_parameter_shapes, in MakeExecutionInputsAndWaitForEvents() argument1813 executable_parameter_shapes[0], argument_handles, in MakeExecutionInputsAndWaitForEvents()1818 if (argument_handles.size() != executable_parameter_shapes.size()) { in MakeExecutionInputsAndWaitForEvents()1820 executable_parameter_shapes.size(), in MakeExecutionInputsAndWaitForEvents()1830 executable_parameter_shapes[i], *transfer_manager, i)); in MakeExecutionInputsAndWaitForEvents()1831 execution_inputs.emplace_back(executable_parameter_shapes[i]); in MakeExecutionInputsAndWaitForEvents()
764 absl::Span<const Shape> executable_parameter_shapes,