| 62 | } |
| 63 | |
| 64 | void VarDesc::SetTensorDescNum(size_t num) { |
| 65 | switch (desc_.type().type()) { |
| 66 | case proto::VarType::READER: { |
| 67 | auto *dense_tensors_ptr = |
| 68 | desc_.mutable_type()->mutable_reader()->mutable_dense_tensor(); |
| 69 | dense_tensors_ptr->Clear(); |
| 70 | for (size_t i = 0; i < num; ++i) { |
| 71 | dense_tensors_ptr->Add(); |
| 72 | } |
| 73 | return; |
| 74 | } break; |
| 75 | default: |
| 76 | PADDLE_THROW( |
| 77 | common::errors::Unavailable("Setting 'sub_tensor_number' is not " |
| 78 | "supported by the %s type variable.", |
| 79 | this->Name())); |
| 80 | } |
| 81 | need_updated_ = true; |
| 82 | } |
| 83 | |
| 84 | size_t VarDesc::GetTensorDescNum() const { |
| 85 | switch (desc_.type().type()) { |