Searched refs:ExtendedInferenceContext (Results 1 – 3 of 3) sorted by relevance
35 class ExtendedInferenceContext {37 ExtendedInferenceContext( in ExtendedInferenceContext() function65 TF_DISALLOW_COPY_AND_ASSIGN(ExtendedInferenceContext);122 ExtendedInferenceContext* GetExtendedContext(const Node* node) const { in GetExtendedContext()290 ExtendedInferenceContext* ec,301 absl::flat_hash_map<const Node*, std::unique_ptr<ExtendedInferenceContext>,
261 std::unique_ptr<ExtendedInferenceContext> ec( in AddNodeInternal()262 new ExtendedInferenceContext(std::move(ic), node)); in AddNodeInternal()312 ExtendedInferenceContext* node_ext_context = it->second.get(); in UpdateNode()670 ExtendedInferenceContext* ec, in RunShapeFn()
1051 ExtendedInferenceContext* shape_context = in InferInputType()1098 ExtendedInferenceContext* shape_context = in InferOutputType()