1. 28 11月, 2017 2 次提交
    • W
      Fix double type error while using eigen api · ab9d59c5
      wanghaoshuang 提交于
      ab9d59c5
    • W
      Refine code and comments · ea7359c6
      wanghaoshuang 提交于
      1. Remove checking for num_neg_samples.
      2. Fix dims of Output(Cost) and Input(Bias).
      3. Renamed num_sampled_classes to num_neg_samples.
      4. Add TODO for add more distribution sampler.
      5. Init grad_data of bias by zero.
      6. Refine comments.
      7. Register a kernel for type double.
      ea7359c6
  2. 16 11月, 2017 1 次提交
  3. 08 11月, 2017 1 次提交
    • W
      Add nce op · 3d276277
      wanghaoshuang 提交于
      1. Add nce forward and backward kernel for CPU
      3d276277