Searched refs:ExtendedInferenceContext (Results 1 – 3 of 3) sorted by relevance
34 class ExtendedInferenceContext {36 ExtendedInferenceContext( in ExtendedInferenceContext() function64 TF_DISALLOW_COPY_AND_ASSIGN(ExtendedInferenceContext);121 ExtendedInferenceContext* GetExtendedContext(const Node* node) const { in GetExtendedContext()184 ExtendedInferenceContext* outer_context);229 ExtendedInferenceContext* ec);239 std::unordered_map<const Node*, std::unique_ptr<ExtendedInferenceContext>>
142 ExtendedInferenceContext* outer_context) { in InferShapesForFunction()240 std::unique_ptr<ExtendedInferenceContext> ec( in AddNode()241 new ExtendedInferenceContext(std::move(ic), node)); in AddNode()291 ExtendedInferenceContext* node_ext_context = it->second.get(); in UpdateNode()641 ExtendedInferenceContext* ec) { in RunShapeFn()
858 ExtendedInferenceContext* shape_context = in InferInputType()869 ExtendedInferenceContext* shape_context = in InferOutputType()