Any idea on generating and saving a tensorflow model that has gradient nodes like below and the example file on the repo
node {
name: “training/LossScaleOptimizer/gradients/res5c_branch2b_1/Conv2D_grad/Conv2DBackpropInput”
op: “Conv2DBackpropInput”
input: “ConstantFolding/training/LossScaleOptimizer/gradients/res5c_branch2b_1/Conv2D_grad/ShapeN-matshapes-0”
input: “res5c_branch2b_1/Conv2D/Cast”
input: “training/LossScaleOptimizer/gradients/bn5c_branch2b_1/FusedBatchNormV2_grad/FusedBatchNormGradV2”
device: “/job:localhost/replica:0/task:0/device:GPU:0”
attr {
key: “T”
value {
type: DT_HALF
}
}
attr {
key: “data_format”
value {
s: “NCHW”
}
}
attr {
key: “dilations”
value {
list {
i: 1
i: 1
i: 1
i: 1
}
}
}