Searched defs:inference_context (Results 1 – 3 of 3) sorted by relevance
110 std::unique_ptr<InferenceContext> inference_context; in Invoke() local
545 shape_inference::InferenceContext* const inference_context, in InferenceShape()575 shape_inference::InferenceContext* const inference_context, in SetInferenceShape()
634 std::unique_ptr<InferenceContext> inference_context; member