Searched refs:unwrapped_output (Results 1 – 2 of 2) sorted by relevance
533 unwrapped_output = distribution.experimental_local_results(loss_output)534 self.assertLen(unwrapped_output, 1)535 loss_tensor = unwrapped_output[0]
434 BufferHandle* unwrapped_output = in ExecuteProgram() local436 unwrapped_outputs.push_back(unwrapped_output); in ExecuteProgram()