CreateInputs

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

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

Function Usage

Creates a specified number of input nodes.

Prototype

  • Create a specified number of input nodes and return an array, which is applicable to structured binding during compilation.
    template <size_t inputs_num>
    std::array<EsTensorHolder, inputs_num> CreateInputs(size_t start_index = 0)
  • Create a specified number of input nodes and return a container, which is applicable to dynamic binding at run time.
    1
    std::vector<EsTensorHolder> CreateInputs(size_t inputs_num, size_t start_index = 0)
    

Parameters

Parameter

Input/Output

Description

inputs_num

Input

Number of input nodes.

start_index

Input

Input node start index. If the value is not 0, other input nodes have been created. The input node index should start from 0 and increase consecutively.

Returns

Parameter

Type

Description

-

std::array<EsTensorHolder, inputs_num>

Array of created input tensor holders.

-

std::vector<EsTensorHolder>

Container of the input tensor holder.

Restrictions

None

Example

  • Create a specified number of input nodes and return an array, which is applicable to structured binding during compilation.
    //1. Create a graph builder (EsGraphBuilder).
    EsGraphBuilder builder("graph_name");
    // 2. Add two input nodes and use the structured binding of C++17 to bind data0 and data1.
    EsTensorHolder [data0, data1] = builder.CreateInputs<2>();
  • Create a specified number of input nodes and return a container, which is applicable to dynamic binding at run time.
    1
    2
    3
    4
    //1. Create a graph builder (EsGraphBuilder).
    EsGraphBuilder builder("graph_name");
    // 2. Add two input nodes.
    std::vector<EsTensorHolder> inputs = builder.CreateInputs(2);