1. 25 12月, 2017 1 次提交
    • Q
      Impl kernel hint (#6883) · af0c4c45
      Qiao Longfei 提交于
      * init kernel hint
      
      * fix typo
      
      * rm unused code
      
      * add include in op_kernel.h
      
      * restore op_kernel since it will be moved to op_kernel_type
      
      * change force_cpu to use_cpu
      
      * fix compilation
      af0c4c45
  2. 20 12月, 2017 1 次提交
  3. 12 12月, 2017 1 次提交
    • Q
      Refine device context (#6433) · 61ec0b95
      QI JUN 提交于
      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`
      61ec0b95
  4. 08 11月, 2017 1 次提交
    • Y
      Polish OpWithKernel · bbdac7f7
      Yu Yang 提交于
      * 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.
      bbdac7f7
  5. 07 11月, 2017 2 次提交
  6. 04 11月, 2017 1 次提交
  7. 01 11月, 2017 1 次提交
    • Refine evaluator op types (#5208) · 8013328e
      武毅 提交于
      * refine evaluator op types
      
      * update
      
      * follow comments
      
      * update
      
      * fix v2 mnist case
      
      * fix v2 mnist case
      
      * update
      
      * update
      8013328e
  8. 31 10月, 2017 2 次提交
    • Refine evaluator op types (#5208) · 878dd88f
      武毅 提交于
      * refine evaluator op types
      
      * update
      
      * follow comments
      
      * update
      
      * fix v2 mnist case
      
      * fix v2 mnist case
      
      * update
      
      * update
      878dd88f
    • F
      Add accuracy layer (#4958) · f122a5da
      fengjiayi 提交于
      * Complete accuray layer
      
      * Fix error
      
      * Fix error
      
      * Add 'accuracy' to __all__
      
      * update
      
      * Fix Type error
      
      * Fix error
      
      * Refine unit tests
      
      * Fix an unit test error
      f122a5da
  9. 29 10月, 2017 1 次提交
    • Y
      Polish Accuracy Op (#5191) · 46a13e37
      Yu Yang 提交于
      * Accuracy does not support float/double, only support integers
      * Polish error message when an operator does not support some device.
      46a13e37
  10. 20 10月, 2017 1 次提交
  11. 17 10月, 2017 1 次提交
  12. 07 10月, 2017 1 次提交
  13. 27 9月, 2017 1 次提交
    • Q
      Refactoring InferShape (#3946) · 9a9d50a6
      Qiao Longfei 提交于
      * 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
      9a9d50a6
  14. 21 9月, 2017 1 次提交
  15. 20 9月, 2017 2 次提交
  16. 15 9月, 2017 1 次提交
  17. 14 9月, 2017 2 次提交
    • D
      Fix accuracy_op. · d757c1cd
      dangqingqing 提交于
      d757c1cd
    • Accuracy op (#3907) · 2d623364
      武毅 提交于
      * init add
      
      * add topk op
      
      * someupdate
      
      * fix style check
      
      * add test py file
      
      * update top k cuda kernel
      
      * follow comments
      
      * remove debug print
      
      * accuracy_op
      
      * fix casting error
      
      * fix casting error
      
      * fix casting error
      
      * fix rename bug...
      
      * make it smaller
      
      * update cast
      2d623364