Code

Conv2d Per EG Gradient Weights#2

Code

Open

a-shan wants to merge 26 commits intonhynes:myelinnhynes/tvm:myelinfrom a-shan:myelinCopy head branch name to clipboard

+900-31

Commits

Commits on Oct 19, 2018

• Add variance op

  nhynes

  committed
  afb0552

  Copy full SHA for afb0552

• Add FExpandCompute

  nhynes

  committed
  d3189a4

  Copy full SHA for d3189a4

• Pull through hidden outputs

  nhynes

  committed
  f7b1a30

  Copy full SHA for f7b1a30

• Add batch norm training

  nhynes

  committed
  d06b5d5

  Copy full SHA for d06b5d5

• Take gradients of expanded graph

  nhynes

  committed
  0ff0829

  Copy full SHA for 0ff0829

• Use FExpandCompute for global_avgpool2d

  nhynes

  committed
  05589df

  Copy full SHA for 05589df

• Fix reduce import

  nhynes

  committed
  3292cc3

  Copy full SHA for 3292cc3

• Disable ExpandCompute shape check

  

  Larry

  authored and

  nhynes

  committed
  ae47140

  Copy full SHA for ae47140

• Fix batchnorm

  

  Larry

  authored and

  nhynes

  committed
  5e3c636

  Copy full SHA for 5e3c636

• Fix extra_outputs propogation

  Show description for 82c8451
  

  Larry

  authored and

  nhynes

  committed
  82c8451

  Copy full SHA for 82c8451

• Add conv2d_nchw_grad_weight

  Show description for c698252
  nhynes

  committed
  c698252

  Copy full SHA for c698252

• Add pad tag to topi pad

  nhynes

  committed
  3cf4f23

  Copy full SHA for 3cf4f23

• Add schedule for conv2d_grad_weight_nchw

  nhynes

  committed
  07fab00

  Copy full SHA for 07fab00

• Add nnvm conv2d_grad_weight

  nhynes

  committed
  3eeda85

  Copy full SHA for 3eeda85

• Fix conv2d gradient

  

  Larry

  authored and

  nhynes

  committed
  c5c47bb

  Copy full SHA for c5c47bb

• Change layout in conv2d FInferShape

  

  Larry

  authored and

  nhynes

  committed
  312990d

  Copy full SHA for 312990d

• Add custom conv2d_transpose scheduler

  

  Larry

  authored and

  nhynes

  committed
  69b2f72

  Copy full SHA for 69b2f72

• Fix extra_outputs propogation

  Show description for 84b2156
  

  Larry

  authored and

  nhynes

  committed
  84b2156

  Copy full SHA for 84b2156

• Fix transpose and broadcast_div gradient

  

  Larry

  authored and

  nhynes

  committed
  82cacf3

  Copy full SHA for 82cacf3

• Fix batchnorm training

  

  Larry

  authored and

  nhynes

  committed
  9c5f5bb

  Copy full SHA for 9c5f5bb

• Add InstanceNorm

  

  Larry

  authored and

  nhynes

  committed
  137ea12

  Copy full SHA for 137ea12

• Add ExpandCompute for mean

  nhynes

  committed
  d226609

  Copy full SHA for d226609

• Optional weight decay

  nhynes

  committed
  bdd15ea

  Copy full SHA for bdd15ea

• Only update BN moving stats during training

  nhynes

  committed
  0f8c0dc

  Copy full SHA for 0f8c0dc

• Export __tvm_startup

  nhynes

  committed
  edfc72e

  Copy full SHA for edfc72e

• Conv2d Per EG Gradient Weights

  a-shan

  committed
  acb40dc

  Copy full SHA for acb40dc