GetInferenceContext

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

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.

Restrictions

None

Examples

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