cond_op modify

update-doc-pybind
zchen0211 8 years ago
parent 35cc956aac
commit c557402855

@ -61,12 +61,16 @@ class CondOp : public framework::OperatorBase {
*/ */
void InferShape(const framework::Scope& scope) const override; void InferShape(const framework::Scope& scope) const override;
// Set True Block /*
* Set True Block
*/
void set_truenet(std::unique_ptr<OperatorBase>&& net) { void set_truenet(std::unique_ptr<OperatorBase>&& net) {
sub_net_op_[0] = std::move(net); sub_net_op_[0] = std::move(net);
} }
// Set False Block /*
* Set False Block
*/
void set_falsenet(std::unique_ptr<OperatorBase>&& net) { void set_falsenet(std::unique_ptr<OperatorBase>&& net) {
sub_net_op_[1] = std::move(net); sub_net_op_[1] = std::move(net);
} }

Loading…
Cancel
Save