908d5b6ede
#6268 This CL does the following: (1) Adds TF_AddGradients function to C_API which adds gradient nodes for the specified inputs. (2) Adds internal constructor for Scope, need to create a scope from an existing graph in the c_api. (3) Adds constructor for AddSymbolicGradients that assumes OnesLike when grad_inputs aren't provided. (4) Improves error message when gradients aren't provided. Change: 153092774
49 lines
1.6 KiB
C++
49 lines
1.6 KiB
C++
/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==============================================================================*/
|
|
|
|
#include "tensorflow/cc/framework/grad_op_registry.h"
|
|
|
|
namespace tensorflow {
|
|
namespace ops {
|
|
|
|
// static
|
|
GradOpRegistry* GradOpRegistry::Global() {
|
|
static GradOpRegistry* grad_op_registry = new GradOpRegistry;
|
|
return grad_op_registry;
|
|
}
|
|
|
|
bool GradOpRegistry::Register(const string& op, GradFunc func) {
|
|
CHECK(registry_.insert({op, func}).second) << "Existing gradient for " << op;
|
|
return true;
|
|
}
|
|
|
|
Status GradOpRegistry::Lookup(const string& op, GradFunc* func) const {
|
|
auto iter = registry_.find(op);
|
|
if (iter == registry_.end()) {
|
|
const string error_msg =
|
|
"No gradient defined for op: " + op +
|
|
". Please see "
|
|
"https://www.tensorflow.org/code/"
|
|
"tensorflow/cc/gradients/README.md"
|
|
" for instructions on how to add C++ gradients.";
|
|
return errors::NotFound(error_msg);
|
|
}
|
|
*func = iter->second;
|
|
return Status::OK();
|
|
}
|
|
|
|
} // end namespace ops
|
|
} // namespace tensorflow
|