STT-tensorflow

distributions

Merge pull request #36617 from marload:master

2020-02-13 08:08:40 -08:00

linalg

Add banded triangular solve op.

2020-06-18 10:03:35 -07:00

losses

Follow up to 0e1f3de50a. This change enables keras tensor tests related to updates/losses.

2020-06-09 01:13:30 -07:00

numpy_ops

[TF-numpy] Adds __rmatmul__ method to ndarray.

2020-06-22 18:16:03 -07:00

parallel_for

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

ragged

Merge pull request #40240 from cclauss:Fix-SyntaxWarnings

2020-06-18 11:48:25 -07:00

signal

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

structured

Internal visibility change

2020-06-15 09:46:32 -07:00

accumulate_n_benchmark.py

Inlined tensor_shape.{scalar,vector,matrix}

2019-07-23 15:33:44 -07:00

array_grad.py

Add additional constraint to place identity under CPU

2020-05-22 19:12:20 +00:00

array_ops_test.py

array_ops.meshgrid should correctly handle no/empty args

2020-04-28 14:04:16 -07:00

array_ops.py

Merge pull request #40474 from guillaumekln:reverse-sequence-doc-remove-deprecation-warnings

2020-06-16 11:45:30 -07:00

batch_norm_benchmark.py

Apply tf1-tf2 renames to tensorflow/python/ops docstrings and comments.

2019-04-18 16:11:35 -07:00

batch_ops_test.py

Move python batch op from contrib to core.

2019-04-09 13:38:12 -07:00

batch_ops.py

…

bincount_ops_test.py

use bincount_v2 in tf.math.bincount, support axis and binary_output, and support int64.

2020-05-20 15:55:06 -07:00

bincount_ops.py

use bincount_v2 in tf.math.bincount, support axis and binary_output, and support int64.

2020-05-20 15:55:06 -07:00

bitwise_ops_test.py

…

bitwise_ops.py

…

boosted_trees_ops.py

Best feature split operates on a list of tensors.

2019-12-26 17:27:29 -08:00

candidate_sampling_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

check_ops.py

Improve the documentation of tf.debugging.assert_shapes()

2020-06-21 19:54:30 -07:00

clip_ops_test.py

Use arithmetic instead of tf.where to set NaN for tf.clip_by_global_norm failure.

2020-04-13 05:39:43 -07:00

clip_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

clustering_ops_test.py

…

clustering_ops.py

Use arg 'axis' instead of deprecated arg 'dim'

2019-09-16 22:35:50 -04:00

collective_ops_benchmark.py

Small adjustments on import spacing.

2019-12-18 20:32:12 -08:00

collective_ops_gpu_test.py

Disable collective_ops_gpu_test on single GPU and enable on multiple GPUs.

2020-01-10 12:51:56 -08:00

collective_ops_test.py

Remove run_deprecated_v1 annotations from collective ops tests.

2020-06-19 13:22:16 -07:00

collective_ops_xla_test.py

Error out from ScopedAllocatorOptimizer when input shape is not fully known.

2020-04-23 14:04:42 -07:00

collective_ops.py

Add timeout to collective ops to detect deadlocks.

2020-05-29 15:37:42 -07:00

concat_benchmark.py

More conversions of flags library to argparse.

2017-01-11 15:08:53 -08:00

cond_v2.py

No lowering on gradient case op when input is DeviceIndex op

2020-06-02 22:20:41 -07:00

confusion_matrix.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

control_flow_grad.py

More tolerant control flow grad code. It currently fails sometimes inside tf.function

2020-06-01 15:20:48 -07:00

control_flow_ops_benchmark.py

…

control_flow_ops_test.py

Add DeviceIndex xla op.

2020-06-22 15:06:38 -07:00

control_flow_ops.py

Rollback of rollback: [TF/XLA] Only force retracing for non-unique XLA context ID for TPUReplicatedContext

2020-06-18 11:07:20 -07:00

control_flow_state.py

Replace usage of Graph._building_function with a public one

2020-02-07 13:49:05 -08:00

control_flow_util_v2.py

Add a function to dynamic choose and execute the proper implementation based on underlying device placement

2020-05-28 09:54:31 -07:00

control_flow_util.py

minor spelling tweaks

2020-02-11 15:09:21 +09:00

control_flow_v2_disable_test.py

Enable v2 control flow inside graph_mode in TF 2.0.

2019-08-02 20:04:13 -07:00

control_flow_v2_enable_test.py

Enable v2 control flow inside graph_mode in TF 2.0.

2019-08-02 20:04:13 -07:00

control_flow_v2_func_graphs.py

Do not propagate parent name_scope in v2 control flow when inside v1 graph since that would break existing TF2 estimator models.

2020-03-19 12:46:36 -07:00

control_flow_v2_toggles_test.py

Add an experimental API to control behavior of outputting all intermediates when using v2 control flow inside Keras models in graph.

2019-08-02 15:48:06 -07:00

control_flow_v2_toggles.py

[Tensorflow Metrics] Add tracking for enable_control_flow() and disable_control_flow() usage.

2020-04-02 16:34:16 -07:00

conv2d_benchmark.py

…

critical_section_ops.py

[TF CriticalSection] Make deadlock detection friendly for non-distributed eager execution.

2019-08-29 11:04:17 -07:00

ctc_ops.py

Merge pull request #40012 from kyscg:ctc

2020-06-02 11:55:24 -07:00

cudnn_rnn_grad.py

…

custom_gradient.py

Adds custom_grad and vjp to tf_numpy/extensions and trax/math.

2020-06-02 18:40:42 -07:00

data_flow_grad.py

Changed gradient of dynamic_partition to use parallel_dynamic_stitch.

2019-08-29 12:37:32 -07:00

data_flow_ops.py

minor spelling tweaks

2020-02-11 15:09:21 +09:00

default_gradient.py

Support taking gradients of tf.cond and tf.while_loop using LookupTable.

2019-10-28 11:06:28 -07:00

dequantize_op_test.py

Add narrow_range to DequantizeOp

2019-09-30 19:26:31 -07:00

embedding_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

functional_ops_test.py

Make tf.If work with ConcreteFunction.

2020-05-29 15:32:59 -07:00

functional_ops.py

Make tf.If work with ConcreteFunction.

2020-05-29 15:32:59 -07:00

gradient_checker_test.py

Change XLA's default to disable cpu_fast_math options with the exception of min_max behavior.

2020-05-15 13:11:47 -07:00

gradient_checker_v2_test.py

Removes the special case for bfloat16 in gradient_checker_v2, which causes an infinite loop and doesn't help with anything (if the function returns bfloat16, the precision has already been lost).

2020-06-01 15:48:09 -07:00

gradient_checker_v2.py

Removes the special case for bfloat16 in gradient_checker_v2, which causes an infinite loop and doesn't help with anything (if the function returns bfloat16, the precision has already been lost).

2020-06-01 15:48:09 -07:00

gradient_checker.py

Deprecates gradient_checker V1

2019-06-12 13:22:32 -07:00

gradients_impl.py

Modernize tf.gradients docstring.

2020-04-25 10:32:07 -07:00

gradients_test.py

Handle the case in unconnected gradients where an op has multiple outputs and only some of them having incoming grads.

2020-06-03 17:25:26 -07:00

gradients_util.py

Handle the case in unconnected gradients where an op has multiple outputs and only some of them having incoming grads.

2020-06-03 17:25:26 -07:00

gradients.py

Forwardprop: add a public API

2019-10-09 19:40:16 -07:00

histogram_ops_test.py

Internal change

2020-05-06 09:44:17 -07:00

histogram_ops.py

Merge pull request #38958 from yongtang:histogram-doc

2020-05-26 12:05:27 -07:00

image_grad_test.py

Merge pull request #32374 from autoih:t_raise

2019-10-15 12:35:31 -07:00

image_grad.py

Move rgb_to_hsv example.

2020-03-02 11:44:15 -08:00

image_ops_impl.py

Shift forward-compat window for NMS padded to fix TFLite conversion in 2.3

2020-07-06 16:10:41 -07:00

image_ops_test.py

Enable tf.image.ssim to run on tensor-like inputs and add SSIM name space.

2020-06-22 10:41:49 -07:00

image_ops.py

Export fill mode for image_projective_transform

2020-03-12 16:54:14 -07:00

init_ops_test.py

[ROCm] Skipping subtests that check support for the Cholesky and QR ops

2019-07-08 17:34:11 +00:00

init_ops_v2_test.py

Merge pull request #30497 from ROCmSoftwarePlatform:google_upstream_skip_cholesky_qr_subtests

2019-07-09 12:02:34 -07:00

init_ops_v2.py

Make Keras maintain its own initializer classes.

2020-03-26 11:29:12 -07:00

init_ops.py

Fix more docstrings to be compatible with the doctest format.

2019-09-05 12:20:29 -07:00

initializers_ns.py

Add test for API generated with tf_export calls to api_compatibility_test.py.

2018-03-13 17:06:54 -07:00

inplace_ops.py

Deprecate alias_inplace_{update,add,sub} ops.

2019-06-04 12:22:09 -07:00

io_ops.py

…

linalg_grad.py

Add banded triangular solve op.

2020-06-18 10:03:35 -07:00

linalg_ops_impl.py

…

linalg_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

list_ops.py

Handle case when delement is None in PopBackGrad.

2020-03-30 16:22:09 -07:00

logging_ops.py

Enable type annotations for python/ops.

2020-06-18 20:06:47 -07:00

lookup_ops.py

Add SaveableObjects to SavedModel.

2020-07-01 14:28:41 -07:00

manip_grad.py

…

manip_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

map_fn.py

Fix formatting

2020-06-17 07:26:51 -07:00

math_grad_test.py

Add tf.math.xlog1py, a safe way to compute x * log1p(y)

2020-01-09 14:28:37 -08:00

math_grad.py

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

math_ops_linspace_test.py

Move linspace tests to their own file.

2020-05-29 14:26:50 -07:00

math_ops_test.py

Make a couple of remaining tests in math_ops_test.py run in eager mode.

2020-06-22 17:16:11 -07:00

math_ops.py

Export tf.math.reduce_all first

2020-06-12 15:21:49 +02:00

matmul_benchmark.py

…

metrics_impl.py

minor spelling tweaks

2020-02-11 15:09:21 +09:00

metrics.py

…

nccl_ops_test.py

Small adjustments on import spacing.

2019-12-18 20:32:12 -08:00

nccl_ops.py

…

nn_batchnorm_test.py

Fix nn_batchnorm_test test. Remove squeeze.

2019-08-02 14:14:30 -07:00

nn_fused_batchnorm_test.py

Fix fp16 FusedBatchNorm CPU crash if batch dimension is 1.

2020-06-19 10:38:53 -07:00

nn_grad_test.py

Add a bfloat16 sum reducer that uses float32 accumulators. Fix existing tests.

2020-06-01 10:21:39 -07:00

nn_grad.py

Add explicit padding to depthwise conv2d.

2020-03-24 12:28:49 -07:00

nn_impl.py

Follow up to 0e1f3de50a. This change enables keras tensor tests related to updates/losses.

2020-06-09 01:13:30 -07:00

nn_loss_scaling_utilities_test.py

Drop experimental and v2 qualifiers from Strategy experimental_run_v2 method.

2020-03-12 10:26:26 -07:00

nn_ops.py

Reduce Conv2D.__call__ eager overhead by ~20%, Conv3D.__call__ by ~35%, and

2020-06-12 14:54:58 -07:00

nn_test.py

Enable XLA:CPU fast math for min/max by default to be similar to TF's behavior.

2020-06-17 01:27:08 -07:00

nn_xent_test.py

…

nn.py

…

numerics.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

op_selector_test.py

minor spelling tweaks

2020-02-11 15:09:21 +09:00

op_selector.py

Replace list comprehension with generator expressions.

2019-12-16 12:26:12 -08:00

optional_grad.py

cond_v2: use optional tensors instead of FakeParams.

2018-11-28 16:54:09 -08:00

parsing_config.py

Remove circular dependencies by removing submodule imports from ragged package.

2020-03-06 18:26:12 -08:00

parsing_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

partitioned_variables.py

…

proto_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

quantized_conv_ops_test.py

…

quantized_ops_test.py

Add 'narrow_range' and 'int axis' options to tf.quantize and_tf.dequantize to match tf.quantize_and_dequantize

2019-09-06 17:04:14 -07:00

random_grad.py

Add StatelessParameterizedTruncatedNormal sampler.

2020-06-18 10:28:38 -07:00

random_ops.py

Merge pull request #40240 from cclauss:Fix-SyntaxWarnings

2020-06-18 11:48:25 -07:00

raw_ops_test.py

…

resource_variable_ops.py

Make "map_resources" overridable by subclass of Trackable.

2020-06-18 15:59:43 -07:00

resources.py

…

rnn_cell_impl.py

Move legacy TF RNN cells to Keras and export there.

2020-03-23 14:54:51 -07:00

rnn_cell_wrapper_impl.py

Move legacy TF RNN cells to Keras and export there.

2020-03-23 14:54:51 -07:00

rnn_cell.py

…

rnn_grad_test.py

Fix: fixed LSTMBlockCell cuda kernel

2020-02-12 19:36:04 +03:00

rnn_grad.py

Added BlockLSTMV2 and BlockLSTMGradV2 with IFCO layout

2019-08-14 21:34:19 -07:00

rnn.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

script_ops_test.py

Merge pull request #35973 from fsx950223:script-fix

2020-01-22 07:10:12 -08:00

script_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

sdca_ops.py

…

session_ops.py

Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information.

2020-01-29 18:29:05 -08:00

sets_impl.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

sets.py

…

sobol_ops_test.py

Change the default dtype for SobolSample op to float32.

2020-01-28 02:26:47 -08:00

sort_ops_test.py

…

sort_ops.py

tf.numpy: Change a bunch of ops to handle unknown shapes.

2020-06-15 23:04:26 -07:00

sparse_grad.py

…

sparse_ops_test.py

Merge pull request #34624 from noble-ai:dense_sparse_matmul

2020-01-16 17:37:53 -08:00

sparse_ops.py

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

special_math_ops_test.py

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

special_math_ops.py

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

split_benchmark.py

Remove hourglass imports from even more tests

2016-12-16 18:26:25 -08:00

standard_ops.py

Export tftrt python symbols.

2019-11-07 14:07:32 -08:00

state_grad.py

…

state_ops.py

Exposes the already-implemented variable scatter_div, scatter_mul, scatter_min, and scatter_max operations in the variables API. Also fixes some documentation, and adds tests that directly check the var.scatter_* methods.

2019-06-04 20:05:39 -07:00

stateful_random_ops_test.py

Call uniform_full_int in tf.random.Generator.uniform when dtype is integer and both minval and maxval are None, to align with stateless_random_uniform.

2020-03-06 16:05:21 -08:00

stateful_random_ops.py

Added a sub-class of tf.random.Generator in keras/layers/preprocessing/image_preprocessing.py to temporarily allow creating generators inside distribution strategies (with all replicas generating the same numbers).

2020-06-09 16:44:47 -07:00

stateless_random_ops.py

Add StatelessParameterizedTruncatedNormal sampler.

2020-06-18 10:28:38 -07:00

string_ops.py

Add dispatch support to more Python APIs.

2020-05-15 11:03:18 -07:00

summary_op_util.py

Add skip_on_eager flag to ops.name_scope

2019-11-11 05:08:32 -08:00

summary_ops_v2.py

Exposes should_record_summaries() and optimizes for static record_if values.

2020-04-22 13:52:03 -07:00

template.py

minor spelling tweaks

2020-02-11 15:09:21 +09:00

tensor_array_grad.py

Fix race condition in TensorArrayWrite grad.

2020-03-28 10:06:18 -07:00

tensor_array_ops_test.py

Add test case for TensorArray concat.

2019-12-05 17:35:04 -08:00

tensor_array_ops.py

Fix the formatting of TensorArray. __doc__ indents anything after the first line doc. Dedent the rest of the doc after the first line so that it is formatted properly on the site.

2020-05-20 15:17:32 -07:00

tensor_forest_ops.py

…

transpose_benchmark.py

Fix transpose bug for large dimension.

2017-10-06 09:38:26 -07:00

unconnected_gradients.py

…

variable_scope.py

Make core.Tensor the base type for Tensor and replace the register_dense_tensor_like with direct subclassing.

2020-05-12 14:57:45 -07:00

variable_spec_test.py

Add Variable encoding so that functions with variable arguments can be saved to SavedModel.

2019-10-31 16:41:04 -07:00

variables.py

Add Registration for non-stateful scatter_nd_min and scatter_nd_max.

2020-06-03 11:05:54 -07:00

weights_broadcast_ops.py

Branch 154885009 (#9604 )

2017-05-03 12:25:10 -07:00

while_v2_indexed_slices_rewriter.py

Make while_v2_indexed_slices_writers compatible with tensor equality.

2020-03-07 19:12:51 -08:00

while_v2.py

Calculate only necessary gradients of captures in while loop

2020-04-27 16:40:13 -07:00