Searched refs:MutableBuffers (Results 1 – 5 of 5) sorted by relevance
202 auto it = arguments[parameter].MutableBuffers()->find({index}); in ExecuteAsyncOnStream()203 CHECK(it != arguments[parameter].MutableBuffers()->end()); in ExecuteAsyncOnStream()
121 ShapeTree<MaybeOwningDeviceMemory>* MutableBuffers() { return &buffers_; } in MutableBuffers() function
306 for (auto& index_buffer : *argument.MutableBuffers()) { in MarkToBeReleasedArguments()
110 auto* arg_buffers = arg.MutableBuffers(); in ExecuteAsyncOnStream()
1518 execution_input.MutableBuffers()->begin(); in MakeTupleHelper()1520 execution_input.MutableBuffers()->end(); in MakeTupleHelper()1660 execution_input.MutableBuffers()->begin(); in MakeExecutionInputsAndWaitForEvents()1662 execution_input.MutableBuffers()->end(); in MakeExecutionInputsAndWaitForEvents()