| 242 | } |
| 243 | |
| 244 | Status InferenceContext::ExpandOutputs(int new_output_size) { |
| 245 | if (new_output_size < outputs_.size()) { |
| 246 | return errors::InvalidArgument("Trying to reduce number of outputs of op."); |
| 247 | } |
| 248 | outputs_.resize(new_output_size, nullptr); |
| 249 | output_handle_shapes_and_types_.resize(new_output_size); |
| 250 | return Status::OK(); |
| 251 | } |
| 252 | |
| 253 | void InferenceContext::PostInputInit( |
| 254 | std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_data) { |
no test coverage detected