Searched refs:REGISTER_GRADIENT_OP (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/cc/gradients/ |
D | array_grad.cc | 57 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 …]
|
D | math_grad.cc | 63 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 …]
|
D | nn_grad.cc | 48 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 …]
|
D | image_grad.cc | 46 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);
|
D | data_flow_grad.cc | 107 REGISTER_GRADIENT_OP("DynamicPartition", DynamicPartitionGrad); 150 REGISTER_GRADIENT_OP("DynamicStitch", DynamicStitchGrad); 151 REGISTER_GRADIENT_OP("ParallelDynamicStitch", DynamicStitchGrad);
|
D | README.md | 22 REGISTER_GRADIENT_OP("OpName", OpNameGrad);
|
/external/tensorflow/tensorflow/cc/framework/ |
D | grad_op_registry.h | 60 #define REGISTER_GRADIENT_OP(name, fn) \ macro
|