whs
a0e8894137
Merge pull request #7020 from wanghaoshuang/fix_transpose_doc
...
Fix transpose doc
7 years ago
Luo Tao
761b329793
unify the indentation of license
7 years ago
wanghaoshuang
835a4d17f2
Change 'python' code block to 'text' block
7 years ago
wanghaoshuang
b6f638f052
Fix doc of transpose_op
7 years ago
ying
515e44e5f5
fix doc.
7 years ago
Yu Yang
e445b3ff20
Move framework.proto to proto namespace ( #6718 )
...
* Move framework.proto to proto namespace
* Fix compile
* Fix compile
* Fix Compile
7 years ago
QI JUN
61ec0b9516
Refine device context ( #6433 )
...
There are mainly following fixes:
- take `DeviceContext` as the template parameter of math functors and OpKernel instead of `Place`
- remove `eigen_device` interface in base class `DeviceContext`
- remove `GetEigenDevice` interface in `ExecutionContext` and base class `DeviceContext`
- remove unused `platform::EigenDeviceConverter`
- rename `REGISTER_OP_GPU_KERNEL` to `REGISTER_OP_CUDA_KERNEL`
- rename `USE_GPU_ONLY_OP` to `USE_CUDA_ONLY_OP`
7 years ago
kavyasrinet
e65ab795af
Fixing documentations for few more operators ( #5374 )
...
* Doc fix for smooth L1 loss
* Adding doc for softmax_op
* Added doc for softmax_with_cross_entropy
* Adding documentation for transpose_op
* small change to restart TeamCity CI
8 years ago
Yu Yang
73a8b78a72
Correct OpWithKernel's infershape ( #4847 )
...
They are public now
8 years ago
qiaolongfei
c0a34e1c64
rename InferShapeContextBase to InferShapeContext
8 years ago
Qiao Longfei
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
dangqingqing
6e2782e958
update to develop branch.
8 years ago
xzl
0cd9b8c0aa
modify the input\output name to X\Out
8 years ago
xzl
9de45e113a
fixed bug when dims.size == 1, modify the variable naming, add judgement when input_grad is null
8 years ago
xzl
5ede6fd434
delete cuda impl, complete comments, modify variable naming
8 years ago
xzl
55991822a0
modify GetAttr to Attr
8 years ago
xzl
d6651b9b8e
fixed bug of the gpu impl
8 years ago
xzl
17b4b980e7
add the transpose op
8 years ago