Searched refs:decompressed_tensor (Results 1 – 2 of 2) sorted by relevance
83 decompressed_tensor = buffer87 decompressed_tensor.copy_(value)88 return decompressed_tensor132 decompressed_tensor = buffer136 decompressed_tensor.copy_(value)137 return decompressed_tensor177 decompressed_tensor = bucket.buffer()180 decompressed_tensor.copy_(fut.value())181 return decompressed_tensor216 decompressed_tensor = bucket.buffer()[all …]
27 auto decompressed_tensor = bucket.getBufferRef(); in runHook() local28 auto decompress = [decompressed_tensor](c10::ivalue::Future& allreduce_fut) { in runHook()39 decompressed_tensor.copy_(reduce_tensor); in runHook()40 return c10::IValue(decompressed_tensor); in runHook()