Searched refs:MutableBuffer (Results 1 – 7 of 7) sorted by relevance
130 input.MutableBuffer(alias->parameter_index); in AllocateOutputMemoryWithInputReuse()
200 input.MutableBuffer(alias->parameter_index); in AllocateOutputMemoryWithInputReuse()
121 MaybeOwningDeviceMemory* MutableBuffer(const ShapeIndex& index) { in MutableBuffer() function
258 input.MutableBuffer(alias->parameter_index); in CreateResultShapedBuffer()
228 runtime_input.MutableBuffer(index); in UpdateDynamicInputs()
233 execution_input.MutableBuffer(index); in PopulateExecutionInputBuffer()
759 return input.MutableBuffer(allocation->param_shape_index()); in ExecuteAsyncOnStreamImpl()