1. 13 Jul, 2020 1 commit
  2. 03 Jul, 2020 1 commit
    • Da Zheng's avatar
      [Feature] add sparse embedding. (#1497) · cadcc1c2
      Da Zheng authored
      * add sparse embedding.
      
      * fix
      
      * add test.
      
      * man fixes.
      
      * many fixes
      
      * fix sparse emb.
      
      * fix.
      
      * fix lint.
      
      * fix lint.
      
      * fix kvstore.
      
      * expose DistTensor.
      
      * test sparse embeddings.
      
      * add attach_grad to the backends.
      
      * remove part_id
      
      * fix.
      
      * move backward computation.
      
      * move more computation to backend.
      
      * fix a bug when applying learning rate.
      
      * fix a few things.
      
      * fix a few things.
      
      * add docstring
      
      * fix.
      
      * apply no_grad.
      
      * fix tests.
      
      * fix for other frameworks.
      
      * add examples in docstring.
      cadcc1c2
  3. 22 Jun, 2020 1 commit
    • Zihao Ye's avatar
      [kernel] New SpMM & SDDMM kernel on CPU and CUDA (#1644) · 071cba1f
      Zihao Ye authored
      * udp
      
      * simplify
      
      * sddmm dot cpu
      
      * upd
      
      * format
      
      * upd
      
      * compatible with MJ's PR
      
      * lint
      
      * upd
      
      * upd
      
      * upd
      
      * python end
      
      * upd
      
      * upd
      
      * lint
      
      * lint
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * lint
      
      * fix mxnet
      
      * upd
      
      * lint
      
      * use minjie's ptr
      
      * macro
      
      * upd
      
      * reorg
      
      * lint
      
      * fix corner cases
      
      * upd
      
      * enrich cpu docs
      
      * upd
      
      * upd
      
      * lint
      
      * lint
      
      * pylint
      
      * sx review
      
      * improve docstring
      
      * python doc
      
      * upd
      
      * restructure
      
      * lint
      
      * upd test
      
      * upd
      
      * pylint
      
      * fix corner cases and test
      071cba1f
  4. 19 Jan, 2020 1 commit
    • VoVAllen's avatar
      [Tensorflow] Several nn & example (#1191) · a00636a0
      VoVAllen authored
      * several nn example
      
      * appnp
      
      * fix lint
      
      * lint
      
      * add dgi
      
      * fix
      
      * fix
      
      * fix
      
      * fff
      
      * docs
      
      * 111
      
      * fix
      
      * change init
      
      * change result
      
      * tiaocan+1
      
      * fix
      
      * fix lint
      
      * fix
      
      * fix
      a00636a0
  5. 20 Dec, 2019 1 commit
    • VoVAllen's avatar
      [Backend] TF backend (#978) · d30a69bf
      VoVAllen authored
      * tf
      
      * add builtin support
      
      * fiix
      
      * pytest
      
      * fix
      
      * fix
      
      * fix some bugs
      
      * fix selecting
      
      * fix todo
      
      * fix test
      
      * fix test fail in tf
      
      * fix
      
      * fix
      
      * fix gather row
      
      * fix gather row
      
      * log backend
      
      * fix gather row
      
      * fix gather row
      
      * fix for pytorch
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix tests
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix convert
      
      * fix
      
      * fix
      
      * fix
      
      * fix inplace
      
      * add alignment setting
      
      * add debug option
      
      * Revert "add alignment setting"
      
      This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.
      
      * tf ci
      
      * fix lint
      
      * fix lint
      
      * add tfdlpack
      
      * fix type
      
      * add env
      
      * fix backend
      
      * fix
      
      * fix tests
      
      * remove one_hot
      
      * remove comment
      
      * remove comment
      
      * fix
      
      * use pip to install all
      
      * fix test
      
      * fix base
      
      * fix
      
      * fix
      
      * add skip
      
      * upgrade cmake
      
      * change version
      
      * change ci
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix seg fault
      
      * fix
      
      * fix python version
      
      * fix
      
      * try fix
      
      * fix
      
      * fix
      
      * tf takes longer time in ci
      
      * change py version
      
      * fix
      
      * fix
      
      * fix oom
      
      * change kg env
      
      * change kg env
      
      * 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊
      
      * 我再也不搞各种乱七八糟环境了……
      
      * use pytest
      
      * Chang image
      d30a69bf