Commit graph

2794 commits

Author SHA1 Message Date
Xianyan
2d593c46a2 Clone graph keep their original prevs nodes and next nodes sequence (#1763)
* Clone graph keep their original prevs nodes and next nodes sequence

* add comments

* update

* simplify table merge
2017-10-27 11:24:40 +08:00
Ian Wong
f2018c6614 fix shape output tensor type from T to Int (#1737) 2017-10-25 20:36:48 +08:00
zhangxiaoli73
a5ae91c6d2 add toGraph (#1189)
* add toGraph

* refactor toGraph

* add getEndNodes for ParallelTable, MapTable

* add unit test
2017-10-23 10:45:34 +08:00
Yanzhang Wang
e8244efb6c fix: make the length of weight and gradWeight in quantized conv same length (#1692)
* fix: the different length of weight and gradWeight in quantized conv is
very confused. So make them same length.

* fix: return grad
2017-10-20 04:33:06 -04:00
Ian Wong
0cebe9d5f0 Support load more tensorflow operations (#1656)
* Support load more tf operation

* refine session

* fix style issue

* meet code reivew
2017-10-19 14:35:17 +08:00
zhangxiaoli73
c58cf4b348 add ptb model example (#1672)
* add ptb model example

* meet review
2017-10-19 14:24:36 +08:00
Yang Wang
a1508bc7f6 fix decode raw and StridedSlice and Transpose serilization issue (#1673) 2017-10-19 10:08:24 +08:00
Ian Wong
e72e2fae82 add more layers and operations (#1678)
* add more layers and operations

* fix compile error

* fix compile error
2017-10-19 09:56:14 +08:00
Jerry Wu
d48acf9ab9 Fix duplicated msg proto (#1659)
* fix protobuf duplicate msgs

* fix test cases
2017-10-18 09:56:11 +08:00
Yao Zhang
ec031c72a0 Add BroadcastGradientArgs (#1658)
* finish BroadcastGradientArgs

* remove breeze

* pass the tests

* refine the function

* fix scala style

* while instead of for
2017-10-16 17:29:30 +08:00
Xianyan
f4ccc9f754 move model freeze to AbstractModule (#1647)
* move model freeze to container

* update

* update freeze API

* refactor freeze API

* meet code review and add unit test

* fix doc

* meet code review
2017-10-15 22:28:35 -05:00
Xin Qiu
cd9c7b6cf7 SparseLinear SparseJoinTable DenseToSparse (#1652)
* SparseLinear SparseJoinTable DenseToSparse

* Python api

* add DenseToSparseSpec

* update to upstream

* add some method

* meet code review

* fix python unit test

* fix python unit test
2017-10-13 16:44:26 +08:00
zhangxiaoli73
7cbfdb88d4 throw clear exception in lookuptable (#1586) 2017-10-13 12:51:24 +08:00
dding3
e3b1518c41 RecurrentDecoder enhancement (#1619)
* RecurrentDecoder refactor
2017-10-12 22:03:22 -04:00
Yang Wang
5c66b437be serveral fix for preprocessing (#1609)
* serveral fix for preprocessing

* load all

* fix tests

* more fixes

* meet code review

* fix style

* fix style

* fix tests

* fix tests

* rebase and fix api

* refine test case

* fix tests

* add transpose

* fix tests

* fix expandims

* fix serializer

* fix doc
2017-10-12 15:49:35 +08:00
zhangxiaoli73
39113e9a2a add Variational Auto Encoder (#1432)
* vae

* add python api and unit tests

* add docs
2017-10-12 13:16:43 +08:00
Yao Zhang
2c4b9d83db Add tile operation (#1646)
* Add tile operation

* reset core

* remove unnecessary comments
2017-10-11 11:36:14 -05:00
Yanzhang Wang
1ddcaa63fc feat: quantize a whole graph/modules (#1618)
* feat: quantize a whole graph/modules

* feat: python supports

* fix: delete unusage
2017-10-11 03:16:48 -04:00
Yanzhang Wang
05283d3b60 feat: quantized dilated conv (#1617)
* feat: quantized dilated conv

* fix: serialization api changes
2017-10-10 07:43:28 -04:00
Yanzhang Wang
1f5c0a5701 feat: quantized linear (#1615)
* feat: quantized linear

* fix: serialization api changes
2017-10-10 06:45:17 -04:00
Yanzhang Wang
3470d5e205 feat: quantized convolution (#1614)
* feat: quantized convolution

* fix: serialization api changes
2017-10-10 06:02:05 -04:00
Yanzhang Wang
8bf29dc119 feat: serilization for quantized modules (#1613)
* feat: serilization for quantized modules

All quantized modules extending QuantModule, which has an empty Tensor
for gradient. And the object mixes QuantSerializer for protobuf
 supporting.

* refactor: serialization api changes
2017-10-10 04:21:36 -04:00
Jerry Wu
af0aaccf68 support graph (#1636) 2017-10-10 10:26:06 +08:00
Yanzhang Wang
7a5dfa350b feat: quantized tensor support (#1611)
* feat: quantized tensor support

QuantizedTensor inherits from Tensor with supporting part of methods.
The memory is allocated from native code, so it only matains a
pointer `getNativeStorage`.

Note,

We don't support automatically `free` method. Anyone using this method
should think about the memory lacking, especially for memory
optimization.

test: quantized tensor tests

* fix: test case failed on spark 1.6
2017-10-09 08:20:01 -04:00
Yanzhang Wang
2a23278691 feat: quantization methods (#1610)
Quantization object support:

1. quantize a value with max and min.
2. quantize an array
3. quantize a Tensor[Float]

And for test, there're relative dequantize methods.
2017-10-09 04:33:18 -04:00
Jerry Wu
99da09f0ed Tensor types support (#1630)
* refinemnt to support more tensor types

* refinemnt

* refinemnt

* support object sharing

* support shared memory

* remove unused type

* refinement
2017-10-09 15:14:31 +08:00
jenniew
b63bea22a6 fix MapTable issue 2017-10-05 17:59:03 -07:00
Ian Wong
34a3b0f5fa Refine scheduler (#1607)
don't execute relevant nodes in forward/backward
fix a bug when handle const node
2017-09-29 13:23:31 +08:00
dding3
da17fa7586 Support feed prior prediction back into recurrent (#1476)
* Implement RecurrentDecoder
* add ut for lstm with recurrent decoder
2017-09-27 13:42:18 -04:00
Ian Wong
46a7195bbe 1. throw exception when ConcatTable contains no submodule (#1598)
2. rescursive remove stopGradient nodes
2017-09-27 17:13:42 +08:00
Ian Wong
ce4f777a39 Refine tensorflow loader code strcuture (#1593)
* revise tensorflow loaders

* refine more layers

* fix sytle issue

* fix style issue

* more ops

* meet code review

* fix style issue
2017-09-27 16:54:11 +08:00
zhangxiaoli73
1e1466b33d refactor predictor (#1582)
* refactor predictor

* predictClass share model output memory

* refactor repeatMemory to shareBuffer
2017-09-26 14:48:56 +08:00
Yang Wang
423027c89d Support Tensorflow Preprocessing (#1572)
* fix tensor bug

* preprocessing

* test

* fix tests

* refine

* fix tests

* fix style

* fix tests

* meet code review

* meet code review

* meet code review

* add doc
2017-09-25 16:23:09 +08:00
Yao Zhang
e66d7f6e8f Add Realdiv, BiasAdd and Pad (#1579)
* Add Pad

* refactor biasAdd
2017-09-25 15:33:33 +08:00
Yao Zhang
08161f96ed Add Assign, Cast, ExpandDims, MaxPool operations (#1562)
* add assign

* Add Cast and ExpandDims

* meet code review

* meet code reiew

* meet code review

* meet code review

* meet code review

* add comment to NumericWildcard type

* try to fix an AbstractMethod error in Cast method

* fix test failure
2017-09-25 15:22:15 +08:00
Xin Qiu
aa31da4797 fix Logsoftmax infinity error (#1576) 2017-09-21 16:02:03 +08:00
Ian Wong
f7ea6315e5 Support load tf unet (#1575)
* add deconv2d, resizebilinear and support broadcast in add

* fix test

* fix style error and meet code review

* fix python failure

* fix unit test
2017-09-21 11:53:21 +08:00
Jerry Wu
956593ad02 Handle null array (#1571)
* fix type comparision

* handle null value array

* fix type diff
2017-09-21 10:16:16 +08:00
Yao Zhang
cc21eaf0cf Add slice and onehot (#1547)
* Add slice and onehot

* refactor the code

* fit new code

* fix a typo

* meet code review

* fix test failed
2017-09-20 21:47:37 +08:00
Yang Wang
4cb832c141 Add assert, image decoders, parseExample operation (#1559)
* add some ops

* add parse example

* style

* add variable

* add png decoder

* add raw decoder

* remove resources

* fix ops

* test sig

* remove useless code

* remove import

* fix style

* fix conflict

* fix api
2017-09-20 10:02:05 +08:00
Yang Wang
42fafc3ba0 support different type in tensorflow loader (#1569)
* support different type in tensorflow

* meet code review

* meet code review

* fix tests
2017-09-19 18:51:59 +08:00
zhangxiaoli73
4d0aaf96ea optimize copy in recurrent (#1556)
* optimize copy in recurrent

* add more unit test

* select optimize

* meet review
2017-09-19 13:34:00 +08:00
Yang Wang
29626d863a support multi output in graph follow up (#1564)
* support multi-output follow up

* meet code review
2017-09-18 13:02:57 +08:00
Ian Wong
1b5ad05658 Fix failed tensorflow unit test (#1563)
* fix tf unit test

* fix unit test

* meet code review
2017-09-15 17:23:04 +08:00
Yao Zhang
e7e78e5ea9 Implement and test Conv2d operation (#1539)
* finish conv2d

* add conv2d test

* fit new code

* meet code review
2017-09-15 15:41:49 +08:00
Yao Zhang
88e2e4583b Add Floor L2Loss RandomUniform Rank MatMul SoftMax operations (#1533)
* make a layer support different type

* finish equal and make layer suppor different type

* fix conflict

* add Floor L2Loss RandomUniform Rank MatMul SoftMax

* add TruncatedNormal

* fit new code

* make jenkens pass
2017-09-15 10:34:58 +08:00
Ian Wong
b4919497fc Support switch and merge operation (#1555)
* Support control flow

* remove loop

* fix unit test

* fix unit test
2017-09-14 13:21:05 +08:00
Ian Wong
e16f246980 Fix operation logical related bugs (#1549)
* Support control flow

* Decouple input/output tensor numeric type from the module numeric type(parameter type).

* FIx unit tests

* while loop api refactor

* fix unit test

* make breeze version configurable as spark 2.1 bump breeze version

* Add scheduler to exclude layers

* remove control-flow change

* remove Schduler and ControlOps

* fix broken unit test
2017-09-12 13:39:27 +08:00
Ian Wong
d213b4df98 Display bigdl model in tensorboard (#1545)
* Display bigdl model in tensorboard

* add document, python API and meet code review.

* meet code review

* meet code review
2017-09-12 11:32:51 +08:00
Yao Zhang
416dbb4e4b Add Greater and Less operations (#1525)
* finish greater and less

* fix scala style
2017-09-07 09:52:58 +08:00