Searched refs:MutableBuffer (Results 1 – 6 of 6) sorted by relevance
118 input.MutableBuffer(alias->parameter_index); in AllocateOutputMemoryWithInputReuse()
123 MaybeOwningDeviceMemory* MutableBuffer(const ShapeIndex& index) { in MutableBuffer() function
265 input.MutableBuffer(alias->parameter_index); in CreateResultShapedBuffer()
226 runtime_input.MutableBuffer(index); in UpdateDynamicInputs()
483 return input.MutableBuffer(allocation->param_shape_index()); in ExecuteAsyncOnStreamImpl()
210 execution_input.MutableBuffer(index); in PopulateExecutionInputBuffer()