Searched refs:ExtendedInferenceContext (Results 1 – 2 of 2) sorted by relevance
35 class ExtendedInferenceContext {37 ExtendedInferenceContext( in ExtendedInferenceContext() function50 const std::unordered_map<string, std::unique_ptr<ExtendedInferenceContext>>&65 std::unordered_map<string, std::unique_ptr<ExtendedInferenceContext>> in set_nested_inferences()78 std::unordered_map<string, std::unique_ptr<ExtendedInferenceContext>>81 TF_DISALLOW_COPY_AND_ASSIGN(ExtendedInferenceContext);138 ExtendedInferenceContext* GetExtendedContext(const Node* node) const { in GetExtendedContext()216 ExtendedInferenceContext* outer_context);261 ExtendedInferenceContext* ec);271 std::unordered_map<const Node*, std::unique_ptr<ExtendedInferenceContext>>
134 ExtendedInferenceContext* outer_context) { in InferShapesForFunction()186 std::unordered_map<string, std::unique_ptr<ExtendedInferenceContext>> in InferShapesForFunction()267 std::unique_ptr<ExtendedInferenceContext> ec( in AddNode()268 new ExtendedInferenceContext(std::move(c), node)); in AddNode()318 ExtendedInferenceContext* node_ext_context = it->second.get(); in UpdateNode()623 ExtendedInferenceContext* ec) { in RunShapeFn()