Get the name of OneDNN op from original TensorFlow op We prefix 'OneDNN' to the original op to get OneDNN op.

Adds {name, value} pair to hyperparameters. Args: name: Name of the hyperparameter. value: Value of the hyperparameter. Can be one of

Create a test session with master target set to the testing cluster. Creates a test session that connects to the local testing cluster. The s

The function to predict.

Return the unique identifier for this HloBuffer.

rbegin/rend for iterating over all nodes in reverse.

Allocate the second output tensor that will contain the OneDNN shape serialized

Returns a deep copy of node (excluding some fields, see copy_clean).

Removes the last layer in the model. Raises: TypeError: if there are no layers in the model.

The number of dimensions in this shape, or None if unknown.

xla::int64 is not the same type as tensorflow::protobuf_int64 in open-source. Wrapper function that gives an int64 array slice view of a repeated int6

Returns a shape based on `self` with at least the given rank. Args: rank: An integer. Returns: A shape that is at least as speci

Creates an `Iterator` for enumerating the elements of this dataset. Note: The returned iterator will be in an uninitialized state, and you mu

Dequeues and concatenates `n` elements from this queue. This operation concatenates queue-element component tensors along the 0th dimension t

Returns the next record (key, value pair) produced by the reader. The multiple reader instances are all configured to `read()` from the filen

Runs the core tests. Args: ds_fn: 0-argument function that returns a Dataset. num_outputs: Total number of outputs expected from this

Apply gradients to variables. This is the second part of `minimize()`. It returns an `Operation` that applies gradients. Args: gra

Returns the list of global variables created by the Template.

Creates an `Iterator` for enumerating the elements of this dataset. Note: The returned iterator will be initialized automatically. A "one-sho

\brief Copy the other tensor into this tensor and reshape it. This tensor shares other's underlying storage. Returns `true` iff `other.shape()` has t

Returns a list of integers or `None` for each dimension. Returns: A list of integers or `None` for each dimension. Raises: Value

Creates an `Operation` in this graph. This is a low-level interface for creating an `Operation`. Most programs will not call this method dire

Add a sequence of trackable values.

Load the MetaGraphDef graph and restore variable values into the session. Args: sess: tf.compat.v1.Session to restore variable values.

-- ACCESSORS -------------------------------------------------------------- The class provides a value() accessor returning the stored ValueType value

Trains a model given training data `x` predictions and `y` labels. Args: x: Matrix of shape [n_samples, n_features...] or the dictionary of

Returns a new dimension of the given size. The returned value is owned by this context.

Returns the object referred to by `obj`, as an `Operation` or `Tensor`. This function validates that `obj` represents an element of this grap

Asserts that ProtoEq says a != b.

Returns `ModelFnOps` that a model_fn can return. Please note that, + Exactly one of `logits` and `logits_input` must be provided. + All a

r"""Gather slices from `params` into a Tensor with shape specified by `indices`. See tf.gather_nd for details. Args: indices: A `Tenso

Raises an exception unless a non-trivial restoration has completed.

Find dumped tensor data by a certain predicate. Args: predicate: A callable that takes two input arguments: ```python def

Looks up "name". Args: name: a string specifying the registry key for the candidate. Returns: Registered object if found Rais

Publicly accessible method for determining the current backend. Only exists for API compatibility with multi-backend Keras. Returns: The s

Matches nested structures. Recursively matches shape and values of `expected` and `actual`. Handles scalars, numpy arrays and other python se