Searched refs:AddToInput (Results 1 – 2 of 2) sorted by relevance
412 void AddToInput(ShapeTree<MaybeOwningDeviceMemory>::iterator* iterator,
563 void PjRtStreamExecutorBuffer::ScopedHold::AddToInput( in AddToInput() function in xla::PjRtStreamExecutorBuffer::ScopedHold1529 device_buffer.AddToInput(&input_iterator, iterator_end, &execution_input, in MakeTupleHelper()1663 device_buffers[i].AddToInput( in MakeExecutionInputsAndWaitForEvents()