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