Home
last modified time | relevance | path

Searched refs:REGISTER_GRADIENT_OP (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc57 REGISTER_GRADIENT_OP("Pack", PackGrad);
67 REGISTER_GRADIENT_OP("Unpack", UnpackGrad);
75 REGISTER_GRADIENT_OP("Identity", IdentityGrad);
83 REGISTER_GRADIENT_OP("RefIdentity", RefIdentityGrad);
91 REGISTER_GRADIENT_OP("QuantizeAndDequantize", QuantizeAndDequantizeGrad);
101 REGISTER_GRADIENT_OP("QuantizeAndDequantizeV2", QuantizeAndDequantizeV2Grad);
112 REGISTER_GRADIENT_OP("QuantizeAndDequantizeV3", QuantizeAndDequantizeV3Grad);
121 REGISTER_GRADIENT_OP("Split", SplitGrad);
139 REGISTER_GRADIENT_OP("Fill", FillGrad);
147 REGISTER_GRADIENT_OP("Diag", DiagGrad);
[all …]
Dmath_grad.cc63 REGISTER_GRADIENT_OP("Abs", AbsGrad);
72 REGISTER_GRADIENT_OP("Neg", NegGrad);
82 REGISTER_GRADIENT_OP("Inv", InvGrad);
83 REGISTER_GRADIENT_OP("Reciprocal", InvGrad);
96 REGISTER_GRADIENT_OP("Square", SquareGrad);
106 REGISTER_GRADIENT_OP("Sqrt", SqrtGrad);
116 REGISTER_GRADIENT_OP("Rsqrt", RsqrtGrad);
128 REGISTER_GRADIENT_OP("Exp", ExpGrad);
141 REGISTER_GRADIENT_OP("Expm1", Expm1Grad);
154 REGISTER_GRADIENT_OP("Log", LogGrad);
[all …]
Dnn_grad.cc48 REGISTER_GRADIENT_OP("Softmax", SoftmaxGrad);
113 REGISTER_GRADIENT_OP("SoftmaxCrossEntropyWithLogits",
126 REGISTER_GRADIENT_OP("LogSoftmax", LogSoftmaxGrad);
135 REGISTER_GRADIENT_OP("Relu", ReluGradHelper);
144 REGISTER_GRADIENT_OP("Relu6", Relu6GradHelper);
157 REGISTER_GRADIENT_OP("LeakyRelu", LeakyReluGradHelper);
171 REGISTER_GRADIENT_OP("LeakyReluGrad", LeakyReluGradGradHelper);
180 REGISTER_GRADIENT_OP("Elu", EluGradHelper);
189 REGISTER_GRADIENT_OP("Selu", SeluGradHelper);
197 REGISTER_GRADIENT_OP("L2Loss", L2LossGrad);
[all …]
Dimage_grad.cc46 REGISTER_GRADIENT_OP("ResizeNearestNeighbor", ResizeNearestNeighborGradHelper);
64 REGISTER_GRADIENT_OP("ResizeBilinear", ResizeBilinearGradHelper);
83 REGISTER_GRADIENT_OP("ResizeBicubic", ResizeBicubicGradHelper);
104 REGISTER_GRADIENT_OP("ScaleAndTranslate", ScaleAndTranslateGradHelper);
124 REGISTER_GRADIENT_OP("CropAndResize", CropAndResizeGradHelper);
Ddata_flow_grad.cc107 REGISTER_GRADIENT_OP("DynamicPartition", DynamicPartitionGrad);
150 REGISTER_GRADIENT_OP("DynamicStitch", DynamicStitchGrad);
151 REGISTER_GRADIENT_OP("ParallelDynamicStitch", DynamicStitchGrad);
DREADME.md22 REGISTER_GRADIENT_OP("OpName", OpNameGrad);
/external/tensorflow/tensorflow/cc/framework/
Dgrad_op_registry.h60 #define REGISTER_GRADIENT_OP(name, fn) \ macro