Searched refs:mutable_input_mem (Results 1 – 1 of 1) sorted by relevance
225 xla::MaybeOwningDeviceMemory* mutable_input_mem = in UpdateDynamicInputs() local232 se::DeviceMemory<int8>(mutable_input_mem->AsDeviceMemoryBase()), in UpdateDynamicInputs()277 *mutable_input_mem = in UpdateDynamicInputs()