Returns a `ConcreteFunction` specialized to inputs and execution context. Args: *args: inputs to specialize on. **kwargs: inputs to s

r"""Computes the absolute value of a tensor. Given a tensor of integer or floating-point values, this operation returns a tensor of the same type

Outputs random values from a normal distribution. Args: shape: A 1-D integer Tensor or Python array. The shape of the output tensor

Asserts that two floats are near each other. Checks that |f1 - f2| < err and asserts a test failure if not. Args: f1: A float valu

A create function. This will never actually get called in this test, it's just nice to know that it compiles.

Returns the index of this allocation.

Get common shape dim, DCHECKing that they all agree.

Gets an existing variable with this name or create a new one.

Returns the parameter instruction for the given parameter number.

Adds a value to this variable. This is essentially a shortcut for `assign_add(self, delta)`. Args: delta: A `Tensor`. The value to ad

Returns the total number of elements in the array.

A deterministic truncated normal if seed is passed.

Run the test, and return 0 on success, 2 otherwise.

Whether this buffer contains an array.

Returns a context manager to enclose code expected to raise an exception. If the exception is an OpError, the op stack is also included in the me

Build tensor data spreading the range [min_value, max_value).

Returns the `Operation` with the given `name`. This method may be called concurrently from multiple threads. Args: name: The name of t

Calculate the GPU launch config we should use for a kernel launch. This is assuming the kernel is quite simple and will largely be memory-limited. REQ

Returns the maximum representable value in this data type. Raises: TypeError: if this is a non-numeric, unordered, or quantized type.

Restore a training checkpoint. Restores this `Checkpoint` and any objects it depends on. Either assigns values immediately if variables to r

Resets the RefPtr from a pointer. This is an ownership transfer operation, and the caller has to own a valid reference to ptr (unless ptr is nullptr).

Saves a training checkpoint and provides basic checkpoint management. The saved checkpoint includes variables created by this object and any

Copy a subgraph. Args: sgv: the source subgraph-view. This argument is converted to a subgraph using the same rules than the function sub

Returns index of node in GraphDef/GraphView.

A TensorFlow session object which will execute the `run`.

Overrides the shape for this variable. Args: shape: the `TensorShape` representing the overridden shape.

Verifies the shape or the shape and the layout of an HLO instruction against the provided shape object.

Verifies that 'op' produces 'expected' when fed input 'inp' . Args: op: operator to test inp: numpy input array to use as input to 'o

The device for SaveSpec Tensors.

Raises an exception if `other` is not compatible with this Dimension. Args: other: Another Dimension. Raises: ValueError: If `se

Returns a `Tensor` for the given key. A `str` key is used to access a base feature (not-transformed). When a `FeatureColumn` is passed, the t

The index of the DT_RESOURCE input to the _XlaCompile/_XlaRun operator. Note that the indices can be different between _XlaCompile and _XlaRun.

Load new value into this variable. Writes new value to variable's memory. Doesn't add ops to the graph. This convenience method requires a s

Return a dense (batch) matrix representing this operator.

Return pointer to variable length section.

Write `value` into index `index` of the TensorArray. Args: index: 0-D. int32 scalar with the index to write to. value: N-D. Tensor

Accumulate a new instance of ProfileDatum. Args: profile_datum: (`ProfileDatum`) an instance of `ProfileDatum` to accumulate to thi