GetInferenceContext

Applicability

Product

Supported or Not

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference products

Atlas training products

Header File

#include <graph/ct_infer_shape_context.h>

Function Usage

Obtains the InferenceContext pointer.

Prototype

1
ge::InferenceContext *GetInferenceContext() const

Parameters

None

Returns

InferenceContext pointer.

For details about the definition of the InferenceContext type, see InferenceContext.

Constraints

None

Examples

1
2
3
4
ge::graphStatus InferShapeForXXX(CtInferShapeContext *context) {
  const auto &read_inference_context = ct_context->GetInferenceContext();
  // ...
}