| 224 | string InferenceContext::op() const { return node_def_->op(); } |
| 225 | |
| 226 | void InferenceContext::PreInputInit( |
| 227 | const OpDef& op_def, const std::vector<const Tensor*>& input_tensors, |
| 228 | const std::vector<ShapeHandle>& input_tensors_as_shapes) { |
| 229 | input_tensors_ = input_tensors; |
| 230 | input_tensors_as_shapes_ = input_tensors_as_shapes; |
| 231 | |
| 232 | construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_, |
| 233 | &output_name_map_); |
| 234 | if (!construction_status_.ok()) return; |
| 235 | |
| 236 | int num_outputs = 0; |
| 237 | for (const auto& e : output_name_map_) { |
| 238 | num_outputs = std::max(num_outputs, e.second.second); |
| 239 | } |
| 240 | outputs_.assign(num_outputs, nullptr); |
| 241 | output_handle_shapes_and_types_.resize(num_outputs); |
| 242 | } |
| 243 | |
| 244 | Status InferenceContext::ExpandOutputs(int new_output_size) { |
| 245 | if (new_output_size < outputs_.size()) { |
nothing calls this directly
no test coverage detected