Searched refs:shape_refiner_ (Results 1 – 1 of 1) sorted by relevance
421 std::unique_ptr<ShapeRefiner> shape_refiner_ = nullptr; member in tensorflow::__anon83888ea00111::ImporterBase707 shape_refiner_ = absl::make_unique<ShapeRefiner>(graph_->versions(), in AddNodesToShapeRefiner()712 shape_refiner_->set_require_shape_inference_fns(false); in AddNodesToShapeRefiner()713 shape_refiner_->set_function_library_for_shape_inference(&graph_flib_); in AddNodesToShapeRefiner()760 Status status = shape_refiner_->AddNode(placeholder_node); in AddNodesToShapeRefiner()783 Status status = shape_refiner_->AddNode(node); in AddNodesToShapeRefiner()792 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner()827 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner()903 auto* shape_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner()913 shape_refiner_->UpdateNode(node, /*relax=*/false, &inferred); in AddNodesToShapeRefiner()[all …]