Searched refs:kGradientOp (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 27 const char* const kGradientOp = "SymbolicGradient"; variable
73 ctx, lib->Instantiate(kGradientOp, AttrSlice(&def().attr()), &handle_), in ComputeAsync()
110 REGISTER_XLA_OP(Name(kGradientOp), SymbolicGradientOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 36 static const char* const kGradientOp = FunctionLibraryDefinition::kGradientOp; variable
249 ctx, lib->Instantiate(kGradientOp, AttrSlice(def()), &handle), done); in ComputeAsync()
286 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_CPU),
288 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_GPU),
291 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_SYCL),
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | gradients.cc | 39 static const char* const kGradientOp = "SymbolicGradient"; variable
108 ndef.set_op(kGradientOp); in AddSymGrad()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 56 static constexpr const char* const kGradientOp = variable
57 FunctionLibraryDefinition::kGradientOp;
703 if (function_name == kGradientOp) { in Instantiate()
709 if (func.name() == kGradientOp) { in Instantiate()
1681 clone->type_string() == kGradientOp) { in InlineFunctionBody()
1787 node.type_string() == FunctionLibraryDefinition::kGradientOp || in IsFunctionCall()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.cc | 211 func->set_name(FunctionLibraryDefinition::kGradientOp); in GetFunctionNameAndAttr()
|
| D | tf2xla_util.cc | 571 if (node_def.op() == FunctionLibraryDefinition::kGradientOp) { in HasAssociatedFunction()
600 } else if (node.type_string() == FunctionLibraryDefinition::kGradientOp) { in GetAssociatedFunctions()
|
| D | xla_compiler.cc | 948 if (op == FunctionLibraryDefinition::kGradientOp || flib_def.Find(op)) { in ValidateFunctionDef()
995 if (node->type_string() == FunctionLibraryDefinition::kGradientOp) { in ValidateGraph()
|
| D | functionalize_while.cc | 265 if (node.op() == FunctionLibraryDefinition::kGradientOp) { in AddMissingFunctionDef()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 406 static constexpr const char* const kGradientOp = "SymbolicGradient"; variable
|
| D | function_test.cc | 1332 ndef.set_op(FunctionLibraryDefinition::kGradientOp); in TEST()
|
| D | function.cc | 1290 if (ndef.op() != kGradientOp) { in GetAttrImpl()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1291 } else if (n->type_string() == FunctionLibraryDefinition::kGradientOp) { in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
|