Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dmath_grad.cc54 REGISTER_OP_GRADIENT("Abs", AbsGrad);
63 REGISTER_OP_GRADIENT("Neg", NegGrad);
75 REGISTER_OP_GRADIENT("Inv", InvGrad);
76 REGISTER_OP_GRADIENT("Reciprocal", InvGrad);
88 REGISTER_OP_GRADIENT("Square", SquareGrad);
102 REGISTER_OP_GRADIENT("Sqrt", SqrtGrad);
117 REGISTER_OP_GRADIENT("Rsqrt", RsqrtGrad);
127 REGISTER_OP_GRADIENT("Exp", ExpGrad);
137 REGISTER_OP_GRADIENT("Expm1", Expm1Grad);
147 REGISTER_OP_GRADIENT("Log", LogGrad);
[all …]
Darray_grad.cc51 REGISTER_OP_GRADIENT("Reshape", ReshapeGrad);
52 REGISTER_OP_GRADIENT("ExpandDims", ReshapeGrad);
71 REGISTER_OP_GRADIENT("Squeeze", SqueezeGrad);
90 REGISTER_OP_GRADIENT("Identity", IdentityGrad);
116 REGISTER_OP_GRADIENT("Pack", PackGrad);
140 REGISTER_OP_GRADIENT("Unpack", UnpackGrad);
224 REGISTER_OP_GRADIENT("Concat", ConcatGrad);
225 REGISTER_OP_GRADIENT("ConcatV2", ConcatGradV2);
245 REGISTER_OP_GRADIENT("Split", SplitGrad);
266 REGISTER_OP_GRADIENT("SplitV", SplitVGrad);
[all …]
Dnn_grad.cc48 REGISTER_OP_GRADIENT("Softmax", SoftmaxGrad);
73 REGISTER_OP_GRADIENT("LogSoftmax", LogSoftmaxGrad);
91 REGISTER_OP_GRADIENT("Relu", ReluGrad);
109 REGISTER_OP_GRADIENT("Relu6", Relu6Grad);
138 REGISTER_OP_GRADIENT("CrossEntropy", CrossEntropyGrad);
174 REGISTER_OP_GRADIENT("Conv2D", Conv2DGrad);
205 REGISTER_OP_GRADIENT("MaxPool", MaxPoolGrad);
231 REGISTER_OP_GRADIENT("AvgPool", AvgPoolGrad);
262 REGISTER_OP_GRADIENT("MaxPoolGrad", MaxPoolGradGrad);
283 REGISTER_OP_GRADIENT("BiasAdd", BiasAddGrad);
Dfunctional_grad.cc56 REGISTER_OP_GRADIENT("MapAccumulate", MapAccumulateGrad);
Dresource_variable_ops.cc176 REGISTER_OP_GRADIENT("ReadVariableOp", ReadGrad);
/external/tensorflow/tensorflow/core/framework/
Dfunction.h1132 #define REGISTER_OP_GRADIENT(name, fn) \ macro
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc402 REGISTER_OP_GRADIENT("Supported", SupportedGrad);
416 REGISTER_OP_GRADIENT("Unsupported", UnsupportedGrad);