| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| function_ops.cc | 27 const char* const kGradientOp = "SymbolicGradient";
73 ctx, lib->Instantiate(kGradientOp, AttrSlice(&def().attr()), &handle_),
110 REGISTER_XLA_OP(Name(kGradientOp), SymbolicGradientOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| function_ops.cc | 36 static const char* const kGradientOp = FunctionLibraryDefinition::kGradientOp;
249 ctx, lib->Instantiate(kGradientOp, AttrSlice(def()), &handle), done);
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/ |
| gradients.cc | 39 static const char* const kGradientOp = "SymbolicGradient";
108 ndef.set_op(kGradientOp);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| graph_compiler.cc | 211 func->set_name(FunctionLibraryDefinition::kGradientOp);
|
| tf2xla_util.cc | 571 if (node_def.op() == FunctionLibraryDefinition::kGradientOp) {
600 } else if (node.type_string() == FunctionLibraryDefinition::kGradientOp) {
|
| xla_compiler.cc | [all...] |
| functionalize_while.cc | 265 if (node.op() == FunctionLibraryDefinition::kGradientOp) {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 56 static constexpr const char* const kGradientOp =
57 FunctionLibraryDefinition::kGradientOp;
703 if (function_name == kGradientOp) {
709 if (func.name() == kGradientOp) {
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function.h | 406 static constexpr const char* const kGradientOp = "SymbolicGradient";
[all...] |
| function_test.cc | [all...] |
| function.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| extract_outside_compilation_pass.cc | [all...] |
