Searched refs:SymbolicGradientBuilder (Results 1 – 2 of 2) sorted by relevance
48 class SymbolicGradientBuilder { class50 SymbolicGradientBuilder(const Scope& scope,137 TF_DISALLOW_COPY_AND_ASSIGN(SymbolicGradientBuilder);140 SymbolicGradientBuilder::SymbolicGradientBuilder( in SymbolicGradientBuilder() function in tensorflow::__anon2882cf250111::SymbolicGradientBuilder151 Status SymbolicGradientBuilder::BackpropAlongEdge(const Output& dst_grad, in BackpropAlongEdge()167 std::vector<bool> SymbolicGradientBuilder::GetReachableNodes() { in GetReachableNodes()191 std::unordered_set<int> SymbolicGradientBuilder::GetStopBackpropNodes( in GetStopBackpropNodes()249 Status SymbolicGradientBuilder::Initialize() { in Initialize()342 Status SymbolicGradientBuilder::SumGradients(const Output& src, Output* grad) { in SumGradients()373 bool SymbolicGradientBuilder::IsPrimitiveOpWithNoGrad(const string& opname) { in IsPrimitiveOpWithNoGrad()[all …]
151 class SymbolicGradientBuilder { class153 SymbolicGradientBuilder(gtl::ArraySlice<NodeOut> y_node_outputs,203 TF_DISALLOW_COPY_AND_ASSIGN(SymbolicGradientBuilder);206 SymbolicGradientBuilder::SymbolicGradientBuilder( in SymbolicGradientBuilder() function in tensorflow::SymbolicGradientBuilder225 void SymbolicGradientBuilder::BackpropAlongEdge(const NodeOut& dst_grad, in BackpropAlongEdge()238 void SymbolicGradientBuilder::BackpropZerosAlongEdge(const NodeOut& src) { in BackpropZerosAlongEdge()248 void SymbolicGradientBuilder::InitBackprop() { in InitBackprop()294 NodeOut SymbolicGradientBuilder::SumGradients(const NodeOut& src) { in SumGradients()333 Status SymbolicGradientBuilder::Compute() { in Compute()403 SymbolicGradientBuilder builder(y_node_outputs, x_node_outputs, in AddSymbolicGradients()