Lines Matching refs:output_tensor
36 output_tensor = input_tensor.reshape((6, 2))
41 self.assertAllEqual(result, output_tensor)
47 output_tensor = input_tensor.reshape((6, 2))[:-1]
52 self.assertAllEqual(result, output_tensor)
58 output_tensor = numpy.array([[[0], [2], [4], [6]], [[1], [3], [5], [7]],
62 # NOTE: output_tensor != input_tensor.reshape((4, 4, -1))
70 self.assertAllEqual(result, output_tensor)
76 output_tensor = numpy.array(
87 # NOTE: output_tensor != input_tensor.reshape((4, 4, 4, -1))
91 self.assertAllEqual(result, output_tensor)
