InstanceOutputShape

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/Library File

  • Header file: #include <ge/compliant_node_builder.h>
  • Library files: libeager_style_graph_builder_base.so and libeager_style_graph_builder_base_static.a

Function Usage

Sets the output shape.

Prototype

1
CompliantNodeBuilder &InstanceOutputShape(const char_t *name, const std::vector<int64_t> &shape)

Parameters

Parameter

Input/Output

Description

name

Input

Output name.

shape

Input

Shape vector.

Returns

Parameter

Type

Description

-

CompliantNodeBuilder &

Reference to the current builder object, which supports chain calls.

Constraints

None