Searched refs:gbody (Results 1 – 2 of 2) sorted by relevance
1395 void Copy(FunctionBody* gbody);1400 void SymbolicGradientHelper::Copy(FunctionBody* gbody) { in Copy() argument1402 gbody->graph = new Graph(src.op_registry()); in Copy()1403 Graph* dst = gbody->graph; in Copy()1409 *(gbody->fdef.mutable_attr()) = fbody_->fdef.attr(); in Copy()1427 gbody->arg_types = fbody_->arg_types; in Copy()1429 gbody->arg_nodes.push_back(node_map[fbody_->arg_nodes[i]->id()]); in Copy()1434 gbody->ret_types = fbody_->ret_types; in Copy()1436 gbody->ret_nodes.push_back(node_map[fbody_->ret_nodes[i]->id()]); in Copy()1441 FunctionBody* gbody = new FunctionBody; in Compute() local[all …]
361 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody)); in GetGradBody() local362 CHECK_NOTNULL(gbody); in GetGradBody()364 CopyGraph(*gbody->graph, ret.get()); in GetGradBody()