Python c2_native_run_op 예제들

예제 #1

파일: c2_ref_test.py 프로젝트: zhuwenxing/pytorch

 def test_tensor_filling_ops(self):
     for dtype in [
             onnx.TensorProto.FLOAT,
             onnx.TensorProto.DOUBLE,
             onnx.TensorProto.BOOL,
             onnx.TensorProto.INT8,
             onnx.TensorProto.INT16,
             onnx.TensorProto.INT32,
             onnx.TensorProto.INT64,
             onnx.TensorProto.UINT8,
             onnx.TensorProto.UINT16,
             onnx.TensorProto.UINT32,
     ]:
         shape = (1, 2, 3)
         vals = np.random.randn(*shape)
         if dtype != onnx.TensorProto.BOOL:
             vals *= 5
         vals = vals.astype(mapping.TENSOR_TYPE_TO_NP_TYPE[dtype])
         tensor = make_tensor(
             name='test-tensor-{}'.format(dtype),
             data_type=dtype,
             dims=[1, 2, 3],
             vals=vals.flatten().tolist(),
         )
         op = c2.Caffe2Backend._create_tensor_filling_op(tensor)
         self.assertEqual(len(op.input), 0)
         self.assertEqual(op.output, [tensor.name])
         ws, output = c2_native_run_op(op, inputs=[])
         self.assertEqual(len(output), 1)
         np.testing.assert_almost_equal(output[0], vals)
         np.testing.assert_almost_equal(ws.FetchBlob(op.output[0]), vals)

예제 #2

파일: c2_ref_test.py 프로젝트: Yangqing/caffe2

 def test_tensor_filling_ops(self):
     for dtype in [
             onnx.TensorProto.FLOAT,
             onnx.TensorProto.DOUBLE,
             onnx.TensorProto.BOOL,
             onnx.TensorProto.INT8,
             onnx.TensorProto.INT16,
             onnx.TensorProto.INT32,
             onnx.TensorProto.INT64,
             onnx.TensorProto.UINT8,
             onnx.TensorProto.UINT16,
             onnx.TensorProto.UINT32,
     ]:
         shape = (1, 2, 3)
         vals = np.random.randn(*shape)
         if dtype != onnx.TensorProto.BOOL:
             vals *= 5
         vals = vals.astype(
             mapping.TENSOR_TYPE_TO_NP_TYPE[dtype])
         tensor = make_tensor(
             name='test-tensor-{}'.format(dtype),
             data_type=dtype,
             dims=[1, 2, 3],
             vals=vals.flatten().tolist(),
         )
         op = c2.Caffe2Backend._create_tensor_filling_op(tensor)
         self.assertEqual(len(op.input), 0)
         self.assertEqual(op.output, [tensor.name])
         ws, output = c2_native_run_op(op, inputs=[])
         self.assertEqual(len(output), 1)
         np.testing.assert_almost_equal(output[0], vals)
         np.testing.assert_almost_equal(ws.FetchBlob(op.output[0]), vals)