ngraph.opset3.ops

Factory functions for all ngraph ops.

Module Attributes

Functions

as_node(input_value)

Return input values as nodes.

as_nodes(*input_values)

Return input values as nodes.

assert_list_of_ints(value_list, message)

Verify that the provided value is an iterable of integers.

assign(new_value, variable_id[, name])

Return a node which produces the Assign operation.

binary_op(node_factory_function)

Convert the first two input values to Constant Nodes if numeric values are detected.

broadcast(data, target_shape[, ...])

Create a node which broadcasts the input node's values along specified axes to a desired shape.

bucketize(data, buckets[, output_type, ...])

Return a node which produces the Bucketize operation.

check_valid_attributes(op_name, attributes, ...)

Perform attributes validation according to specified type, value criteria.

cum_sum(arg, axis[, exclusive, reverse, name])

Construct a cumulative summation operation.

embedding_bag_offsets_sum(emb_table, ...[, ...])

Return a node which performs sums of bags of embeddings without the intermediate embeddings.

embedding_bag_packed_sum(emb_table, indices)

Return an EmbeddingBagPackedSum node.

embedding_segments_sum(emb_table, indices, ...)

Return an EmbeddingSegmentsSum node.

extract_image_patches(image, sizes, strides, ...)

Return a node which produces the ExtractImagePatches operation.

get_dtype(ngraph_type)

Return a numpy.dtype for an ngraph element type.

get_element_type(data_type)

Return an ngraph element type for a Python type or numpy.dtype.

get_element_type_str(data_type)

Return an ngraph element type string representation for a Python type or numpy dtype.

gru_cell(X, initial_hidden_state, W, R, B, ...)

Perform GRUCell operation on the tensor from input node.

is_non_negative_value(x)

Determine whether the specified x is non-negative value.

is_positive_value(x)

Determine whether the specified x is positive value.

make_constant_node(value[, dtype])

Return an ngraph Constant node with the specified value.

nameable_op(node_factory_function)

Set the name to the ngraph operator returned by the wrapped function.

non_max_suppression(boxes, scores[, ...])

Return a node which performs NonMaxSuppression.

non_zero(data[, output_type, name])

Return the indices of the elements that are non-zero.

read_value(init_value, variable_id[, name])

Return a node which produces the Assign operation.

rnn_cell(X, initial_hidden_state, W, R, B, ...)

Perform RNNCell operation on tensor from input node.

roi_align(data, rois, batch_indices, ...[, name])

Return a node which performs ROIAlign.

scatter_elements_update(data, indices, ...)

Return a node which produces a ScatterElementsUpdate operation.

scatter_update(data, indices, updates, axis)

Return a node which produces a ScatterUpdate operation.

shape_of(data[, output_type, name])

Return a node which produces a tensor containing the shape of its input data.

shuffle_channels(data, axis, group[, name])

Perform permutation on data in the channel dimension of the input tensor.

topk(data, k, axis, mode, sort[, ...])

Return a node which performs TopK.

unary_op(node_factory_function)

Convert the first input value to a Constant Node if a numeric value is detected.

Classes

Constant

ngraph.impl.op.Constant wraps ngraph::op::Constant

GraphBody(parameters, results)

Class containing graph parameters and results.

Node

ngraph.impl.Node wraps ngraph::Node

NodeFactory([opset_version])

Factory front-end to create node objects.

Parameter

ngraph.impl.op.Parameter wraps ngraph::op::Parameter

Shape

ngraph.impl.Shape wraps ngraph::Shape

TensorIteratorBodyOutputDesc(body_value_idx, ...)

Represents an output from a specific iteration.

TensorIteratorConcatOutputDesc(...)

Represents an output produced by concatenation of output from each iteration.

TensorIteratorInvariantInputDesc(input_idx, ...)

Represents a TensorIterator graph body input that has invariant value during iteration.

TensorIteratorMergedInputDesc(input_idx, ...)

Represents a TensorIterator graph body input with initial value in the first iteration.

TensorIteratorSliceInputDesc(input_idx, ...)

Represents a TensorIterator graph body input formed from slices of TensorIterator input.

partial

partial(func, *args, **keywords) - new function with partial application of the given arguments and keywords.

Exceptions

Modules