|
|
|
@ -93,9 +93,9 @@ execution.
|
|
|
|
|
|
|
|
|
|
class GetPlacesInferVarType : public framework::VarTypeInference {
|
|
|
|
|
public:
|
|
|
|
|
void operator()(framework::InferVarTypeContext &ctx) const override {
|
|
|
|
|
for (auto &o_name : ctx.Output("Out")) {
|
|
|
|
|
ctx.SetType(o_name, framework::proto::VarType::PLACE_LIST);
|
|
|
|
|
void operator()(framework::InferVarTypeContext *ctx) const override {
|
|
|
|
|
for (auto &o_name : ctx->Output("Out")) {
|
|
|
|
|
ctx->SetType(o_name, framework::proto::VarType::PLACE_LIST);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|