Searched refs:numpy_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | tensor_test.py | 117 numpy_tensor = np.asarray(tensor, dtype=np.int32)
118 self.assertAllEqual(numpy_tensor, [1, 2, 3])
121 numpy_tensor = np.asarray(1.0)
122 tensor = constant_op.constant(numpy_tensor)
123 self.assertAllEqual(numpy_tensor.ndim, tensor.ndim)
125 numpy_tensor = np.asarray([1.0, 2.0, 3.0])
126 tensor = constant_op.constant(numpy_tensor)
127 self.assertAllEqual(numpy_tensor.ndim, tensor.ndim)
129 numpy_tensor = np.asarray([[1.0, 2.0, 3.0], [1.0, 2.0, 3.0]])
130 tensor = constant_op.constant(numpy_tensor)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _tensor.py | 292 numpy_tensor = (
299 (numpy_tensor, self.dtype, str(self.device), self.requires_grad),
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 1438 def save_tensor_data(cls, numpy_tensor, output_path): argument
1441 proto_tensor = numpy_helper.from_array(numpy_tensor)
|