bbdac7f7d8
Polish OpWithKernel
...
* Chage `IndicateDataType` to `GetKernelType`. Make it easier to
understand.
* Change `OpKernelKey` to `OpKernelType`
* Make operator developers can customize which kernel the operator will
use in runtime.
8 years ago
c0d2ca54b9
polish_g_to_l ( #5367 )
8 years ago
73a8b78a72
Correct OpWithKernel's infershape ( #4847 )
...
They are public now
8 years ago
c0a34e1c64
rename InferShapeContextBase to InferShapeContext
8 years ago
88a8eedda1
scatter gather gpu
...
gather scatter gpu
8 years ago
2c05465d2f
Fix unit-tests
8 years ago
9a9d50a6ee
Refactoring InferShape ( #3946 )
...
* init Infershape
* add static InferShape interface
* refactor add-op infershape
* add AttrReader
* add all maker's infershape
* add all InferShape
* add python infer api
* add VarDesc interface
* add python VarDesc and OpDesc interface
* update python code
* use infershape function to do shape inference
* clean code
* do not use pointer
* refine code of op_proto_maker
* add get_dims to VarDesc
* refine the code
* remove the dependency from operator to op registry
* remove OpProtoAndCheckerMaker from operator
* restore complete_add_op
* add shape_infer_impl.h
* code optimization
* remove const return value
* add fake BlockDesc class
* optimize code
* remove infer function in op_info
* move InferShapeContextImpl to operator.h
* optimize the interface of InferShapeContextBase
* add temperary interface of new infershape
* change add_op, clip_op, conv2d_op and activation_op
* change all operators InferShape
* fix SetDim
* update cos_sim_op
* update crop_op
* update lookup_table_op
* allocate tensor when call GetDim in InferShapeContext
* update modified_huber_loss_op
* update rowwise_add_op
* update mean_op
* update sequence_avg_pool_op
* typo
* remove old InferShape interface
* can compile
* fix or unit test
* clean code
* clean code
* remove const before InferShapeContext
* change InferenceContextBase to pointer
* rename RunTime to Runtime, code clean
8 years ago
36aeb30d12
Remove LoDTensor in some operators' InferShape and refine ShareLoD function.
8 years ago
eef1ccbf08
Add the check of inputs and outputs in all operators.
8 years ago
f299206396
Using LoDTensor instead of Tensor in every operator.
8 years ago
9a3c69c268
Revert "Remove `grad_op_type` in `REGISTER_OP`"
8 years ago
79b1f33a02
Remove `grad_op_type` in REGISTER_OP
8 years ago
53e71b44f4
gather op bp passed
8 years ago
6075928d55
gather op added
8 years ago
4d2adab772
gather op added with python unittest
8 years ago
f6bffd4e1f
gather_op modified
8 years ago
2a42a73db1
modify gather_op with test
8 years ago
caaa5f86b9
gather op added
8 years ago
af1eb31afc
add as an operator
8 years ago
Yu Yang
kexinzhao
qiaolongfei
zchen0211
dangqingqing
Liu Yiqun
fengjiayi