Searched refs:kGradientOp (Results 1 – 15 of 15) sorted by relevance
26 const char* const kGradientOp = "SymbolicGradient"; variable72 REGISTER_XLA_OP(Name(kGradientOp), AlwaysFailOp);
37 static const char* const kGradientOp = FunctionLibraryDefinition::kGradientOp; variable221 ctx, lib->Instantiate(kGradientOp, AttrSlice(def()), &handle), done); in ComputeAsync()259 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_CPU),261 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_GPU),
82 } else if (n->type_string() == FunctionLibraryDefinition::kGradientOp) { in RewriteFunctionCallNode()
40 static const char* const kGradientOp = "SymbolicGradient"; variable109 ndef.set_op(kGradientOp); in AddSymGrad()
60 static constexpr const char* const kGradientOp = variable61 FunctionLibraryDefinition::kGradientOp;
63 static constexpr const char* const kGradientOp = variable64 FunctionLibraryDefinition::kGradientOp;777 if (function_name == kGradientOp) { in Instantiate()783 if (func.name() == kGradientOp) { in Instantiate()
218 func->set_name(FunctionLibraryDefinition::kGradientOp); in GetFunctionNameAndAttr()
570 if (node_def.op() == FunctionLibraryDefinition::kGradientOp) { in HasAssociatedFunction()599 } else if (node.type_string() == FunctionLibraryDefinition::kGradientOp) { in GetAssociatedFunctions()
1193 if (op == FunctionLibraryDefinition::kGradientOp || flib_def.Find(op)) { in ValidateFunctionDef()1248 if (node->type_string() == FunctionLibraryDefinition::kGradientOp) { in ValidateGraph()
379 static constexpr const char* const kGradientOp = "SymbolicGradient"; variable
53 /* static */ constexpr const char* const FunctionLibraryDefinition::kGradientOp; member in tensorflow::FunctionLibraryDefinition1522 if (ndef.op() != kGradientOp) { in GetAttrImpl()
1338 ndef.set_op(FunctionLibraryDefinition::kGradientOp); in TEST()
2279 FunctionLibraryDefinition::kGradientOp); in LiftOutsideCompilationOnlyArgsFromCallNode()2280 func.set_name(FunctionLibraryDefinition::kGradientOp); in LiftOutsideCompilationOnlyArgsFromCallNode()2340 FunctionLibraryDefinition::kGradientOp); in LiftOutsideCompilationOnlyArgsFromCallNode()2446 FunctionLibraryDefinition::kGradientOp); in LiftOutsideCompilationOnlyArgs()
1822 TF_RET_CHECK(n->type_string() == FunctionLibraryDefinition::kGradientOp); in ExtractOutsideCompilationForFuncCallNode()1823 func.set_name(FunctionLibraryDefinition::kGradientOp); in ExtractOutsideCompilationForFuncCallNode()1827 if (func.name() == FunctionLibraryDefinition::kGradientOp) { in ExtractOutsideCompilationForFuncCallNode()
1022 if (node.type_string() == FunctionLibraryDefinition::kGradientOp) { in MakeFunctionBodyForInlining()