experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/core/kernels
History
Reuben Morais ca5d9fdf5c Add sigmoid and tanh to deepspeech_cwise_ops
2021-01-02 17:17:29 +00:00
..
batching_util
Add TraceMeProducer/Consumer for SharedBatchScheduler.
2020-06-17 15:27:09 -07:00
boosted_trees
Add Buckettize, SparseCross and BoostedTreesBucketize to flex whitelist
2020-05-28 17:29:20 -07:00
cubin_headers
Add support for windows.
2020-06-22 01:43:15 -07:00
data
Update "master" to "dispatch"/"dispatcher" in tf.data service terminology.
2020-07-16 13:07:28 -07:00
fuzzing
Move identity_fuzz to new location. Add linkstatic=1 to fuzz target.
2020-06-08 16:43:06 -07:00
hexagon
Fix issues caught by -Wtautological-constant-compare
2020-06-19 15:34:06 -07:00
neon
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
rnn
minor spelling tweaks
2020-02-27 15:42:16 +09:00
sparse
[ROCm] Fix for ROCm CSB breakage on 200507
2020-05-07 20:55:49 +00:00
special_math
Add Bessel functions to the public api:
2020-06-17 21:06:09 -07:00
spectrogram_test_data
…
tensor_forest
…
adjust_contrast_op_benchmark_test.cc
…
adjust_contrast_op_gpu.cu.cc
…
adjust_contrast_op_test.cc
…
adjust_contrast_op.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
adjust_contrast_op.h
…
adjust_hsv_gpu.cu.h
…
adjust_hue_op_gpu.cu.cc
…
adjust_hue_op.cc
…
adjust_hue_op.h
…
adjust_saturation_op_gpu.cu.cc
…
adjust_saturation_op.cc
…
adjust_saturation_op.h
…
aggregate_ops_cpu.h
…
aggregate_ops_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
aggregate_ops.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
aggregate_ops.h
Reduce aggregate_ops' dependencies by moving OP_REQUIRES macro definitions to a
2020-02-18 14:19:01 -08:00
argmax_op_gpu.cu.cc
Support argmin/argmax of boolean tensors on CPUs and GPUs.
2020-04-14 13:37:21 -07:00
argmax_op.cc
Support argmin/argmax of boolean tensors on CPUs and GPUs.
2020-04-14 13:37:21 -07:00
argmax_op.h
Internal change
2020-01-29 14:37:24 -08:00
as_string_op.cc
…
assign_op.h
…
attention_ops.cc
Use ExtractGlimpseV2 and ExtractGlimpse to make sure C++ kernel is backward compatible
2020-05-12 09:31:29 -07:00
avgpooling_op_gpu.cu.cc
…
avgpooling_op.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
avgpooling_op.h
…
banded_triangular_solve_op_test.cc
Add banded triangular solve op.
2020-06-18 10:03:35 -07:00
banded_triangular_solve_op.cc
Add banded triangular solve op.
2020-06-18 10:03:35 -07:00
barrier_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
base64_ops.cc
…
basic_ops_benchmark_test.cc
…
batch_kernels.cc
A refactor to add helper functions {Split, Concat}, which do Tensor dtype deduction automatically.
2020-06-16 10:14:28 -07:00
batch_matmul_op_complex.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
batch_matmul_op_impl.h
Fix bug in complex einsum grad.
2020-05-05 19:46:15 -07:00
batch_matmul_op_real.cc
…
batch_matmul_op_test.cc
…
batch_norm_op_gpu.cu.cc
…
batch_norm_op_test.cc
…
batch_norm_op.cc
…
batch_norm_op.h
…
batch_util.h
…
batchtospace_op.cc
…
bcast_ops.cc
…
betainc_op_gpu.cu.cc
…
betainc_op.cc
…
betainc_op.h
…
bias_op_gpu.cu.cc
Add int32 type support for BiasAdd. CPU support already exists, so adding GPU support only.
2019-10-31 15:58:08 -07:00
bias_op_gpu.h
…
bias_op_test.cc
Remove redundant reshape from BiasOp
2019-10-17 21:35:51 -07:00
bias_op.cc
Add int32 type support for BiasAdd. CPU support already exists, so adding GPU support only.
2019-10-31 15:58:08 -07:00
bias_op.h
Remove redundant reshape from BiasOp
2019-10-17 21:35:51 -07:00
bincount_op_gpu.cu.cc
Add bincount to support dense/sparse/ragged inputs.
2020-05-06 14:00:20 -07:00
bincount_op_test.cc
…
bincount_op.cc
Update attr name for Dense version Bincount.
2020-05-13 16:14:22 -07:00
bincount_op.h
Add bincount to support dense/sparse/ragged inputs.
2020-05-06 14:00:20 -07:00
broadcast_to_op_gpu.cu.cc
Register int64 kernels for BroadcastTo
2020-04-28 12:19:44 -07:00
broadcast_to_op_test.cc
…
broadcast_to_op.cc
Register int64 kernels for BroadcastTo
2020-04-28 12:19:44 -07:00
broadcast_to_op.h
…
bucketize_op_gpu.cu.cc
…
bucketize_op.cc
…
bucketize_op.h
…
BUILD
Add sigmoid and tanh to deepspeech_cwise_ops
2021-01-02 17:17:29 +00:00
build_defs.bzl
Use a MLIR generated kernel for tanh for the GPU backend (behind build flag).
2020-06-10 07:09:56 -07:00
candidate_sampler_ops.cc
…
cast_op_gpu.cu.cc
…
cast_op_impl_bfloat.cc
…
cast_op_impl_bool.cc
…
cast_op_impl_complex64.cc
…
cast_op_impl_complex128.cc
…
cast_op_impl_double.cc
…
cast_op_impl_float.cc
…
cast_op_impl_half.cc
…
cast_op_impl_int8.cc
…
cast_op_impl_int16.cc
…
cast_op_impl_int32.cc
…
cast_op_impl_int64.cc
…
cast_op_impl_uint8.cc
…
cast_op_impl_uint16.cc
…
cast_op_impl_uint32.cc
…
cast_op_impl_uint64.cc
…
cast_op_impl.h
…
cast_op_test.cc
Fixing bug in cast_op test: A case study in the danger of default arguments.
2020-04-13 17:41:59 -07:00
cast_op.cc
[CastOp] Avoid creating and destroying a temporary tensor in the common case.
2020-03-05 11:16:13 -08:00
cast_op.h
…
check_numerics_op_gpu.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
check_numerics_op.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
cholesky_grad.cc
…
cholesky_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
clustering_ops_test.cc
…
clustering_ops.cc
…
collective_nccl_broadcaster.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
collective_nccl_broadcaster.h
…
collective_nccl_gatherer.cc
Unblock dependencies in RingReducer.
2019-10-21 19:28:33 -07:00
collective_nccl_gatherer.h
…
collective_nccl_reducer.cc
Unblock dependencies in RingReducer.
2019-10-21 19:28:33 -07:00
collective_nccl_reducer.h
…
collective_nccl_test.cc
[NFC] Provide synchronous versions of CopyDeviceTensorToCPU and CopyCPUTensorToDevice
2020-04-17 12:27:29 -07:00
collective_nccl.cc
…
collective_nccl.h
…
collective_ops.cc
Add timeout to collective ops to detect deadlocks.
2020-05-29 15:37:42 -07:00
colorspace_op_gpu.cu.cc
…
colorspace_op_test.cc
…
colorspace_op.cc
…
colorspace_op.h
…
compare_and_bitpack_op_gpu.cu.cc
…
compare_and_bitpack_op.cc
…
compare_and_bitpack_op.h
…
concat_lib_cpu.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
concat_lib_cpu.h
Internal change
2019-10-25 21:39:13 -07:00
concat_lib_gpu_impl.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
concat_lib_gpu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
concat_lib_gpu.h
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
concat_lib.h
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
concat_op_test.cc
Fixes/updates for tensorflow::tstring migration.
2019-11-26 23:35:38 -08:00
concat_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
conditional_accumulator_base_op.cc
…
conditional_accumulator_base_op.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
conditional_accumulator_base.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
conditional_accumulator_base.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
conditional_accumulator_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
conditional_accumulator.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
constant_op_gpu.cu.cc
…
constant_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
constant_op.cc
Mozilla TensorFlow r2.3
2020-08-06 19:19:26 +02:00
constant_op.h
Internal change
2020-01-10 17:44:33 -08:00
control_flow_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
control_flow_ops.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
control_flow_ops.h
…
conv_2d_gpu_double.cu.cc
…
conv_2d_gpu_float.cu.cc
…
conv_2d_gpu_half.cu.cc
…
conv_2d_gpu_int_spatial_convolution_backward.cu.cc
Split up conv_2d kernels even further
2020-02-06 00:34:58 -08:00
conv_2d_gpu_int_spatial_convolution.cu.cc
Split up conv_2d kernels even further
2020-02-06 00:34:58 -08:00
conv_2d_gpu_int.cu.cc
Split up conv_2d kernels even further
2020-02-06 00:34:58 -08:00
conv_2d_gpu_uint8.cu.cc
…
conv_2d_gpu_uint16.cu.cc
…
conv_2d_gpu_uint32.cu.cc
…
conv_2d_gpu_uint64.cu.cc
…
conv_2d_gpu.h
Merge pull request #39166 from kaixih:pr_vectorize_transpose
2020-05-26 18:50:26 -07:00
conv_2d.h
Added int32 supports for Conv2D and Conv2DBackpropInput ops.
2019-10-21 17:27:50 -07:00
conv_3d.h
…
conv_grad_filter_ops_benchmark_test.cc
Add benchmarks for Conv2D filter gradient with strides not equal to one
2019-11-27 16:50:27 -08:00
conv_grad_filter_ops.cc
PR #36267: [ROCm] Reverting ROCm to use MIOpen Find Mode APIs (be default) for convolution
2020-04-08 00:44:58 -07:00
conv_grad_input_ops_benchmark_test.cc
Add benchmarks for Conv2D input gradient with strides not equal to one
2019-11-27 16:00:31 -08:00
conv_grad_input_ops.cc
Allow the compiler to vectorize the loop.
2020-06-17 13:10:12 -07:00
conv_grad_ops_3d.cc
Merge pull request #40399 from kaixih:pr_cudnn_conv3d_ndhwc
2020-06-17 13:38:36 -07:00
conv_grad_ops.h
Move conv grad shape related utils to a separate lib
2019-10-30 13:58:39 -07:00
conv_grad_shape_utils.cc
Take #2: Improve Conv2DBackpropInput to take input_sizes as a 2D shape.
2020-03-26 16:33:53 -07:00
conv_grad_shape_utils.h
Take #2: Improve Conv2DBackpropInput to take input_sizes as a 2D shape.
2020-03-26 16:33:53 -07:00
conv_ops_3d.cc
Merge pull request #40399 from kaixih:pr_cudnn_conv3d_ndhwc
2020-06-17 13:38:36 -07:00
conv_ops_3d.h
Supporting conv3d as a TFRT forwarded kernel.
2020-03-11 14:28:30 -07:00
conv_ops_benchmark_test.cc
Add Conv2D benchmarks for NASNet shapes
2019-12-10 18:22:35 -08:00
conv_ops_fused_double.cc
…
conv_ops_fused_float.cc
…
conv_ops_fused_half.cc
…
conv_ops_fused_image_transform.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
conv_ops_fused_impl.h
This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor.
2020-04-06 11:16:51 -07:00
conv_ops_gpu.h
Merge pull request #32302 from houtoms:pr_cudnn_ctc_loss
2020-01-17 20:32:44 -08:00
conv_ops_test.cc
PR #39548: [INTEL MKL] Fix conv_ops_test and remapper_test
2020-05-21 14:37:31 -07:00
conv_ops_using_gemm.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
conv_ops.cc
PR #36267: [ROCm] Reverting ROCm to use MIOpen Find Mode APIs (be default) for convolution
2020-04-08 00:44:58 -07:00
conv_ops.h
…
count_ops_test.cc
Add tf.sparse.bincount op.
2020-04-30 17:50:34 -07:00
count_ops.cc
Automated g4 rollback of changelist 311477582.
2020-05-14 08:42:30 -07:00
count_up_to_op.cc
…
crop_and_resize_op_benchmark_test.cc
…
crop_and_resize_op_gpu.cu.cc
…
crop_and_resize_op_test.cc
Explicitly capture variables in lambdas.
2020-01-10 14:42:49 -08:00
crop_and_resize_op.cc
Remove unused unique_tensor_references library.
2020-03-12 16:35:29 -07:00
crop_and_resize_op.h
…
cross_op_gpu.cu.cc
…
cross_op_test.cc
…
cross_op.cc
…
cross_op.h
…
ctc_decoder_ops.cc
Stops calling std::copy_n over zero-sided vec, which indeed dereferences null.
2020-06-15 18:46:51 -07:00
ctc_loss_op.cc
[ROCm] Adding ROCm support for CTC Loss
2020-01-27 18:50:24 +00:00
cuda_solvers.cc
This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor.
2020-04-06 11:16:51 -07:00
cuda_solvers.h
Remove unused unique_tensor_references library.
2020-03-12 16:35:29 -07:00
cuda_sparse.cc
Merge pull request #38802 from nluehr:cusparse_remove_deprecated
2020-05-06 11:05:02 -07:00
cuda_sparse.h
[ROCm] Fix for ROCm CSB breakage on 200507
2020-05-07 20:55:49 +00:00
cudnn_pooling_gpu.cc
…
cudnn_pooling_gpu.h
…
cudnn_rnn_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
cwise_op_abs.cc
…
cwise_op_acos.cc
…
cwise_op_acosh.cc
…
cwise_op_add_1.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
cwise_op_add_2.cc
Support uint32 in tf.add.
2020-04-07 18:30:13 -07:00
cwise_op_arg.cc
…
cwise_op_asin.cc
…
cwise_op_asinh.cc
…
cwise_op_atan2.cc
…
cwise_op_atan.cc
…
cwise_op_atanh.cc
…
cwise_op_bitwise_and.cc
…
cwise_op_bitwise_or.cc
…
cwise_op_bitwise_xor.cc
…
cwise_op_ceil.cc
Add int32 support to floor, ceil & rint
2019-12-04 18:47:55 -08:00
cwise_op_clip_gpu.cu.cc
…
cwise_op_clip.cc
…
cwise_op_clip.h
…
cwise_op_complex.cc
…
cwise_op_conj.cc
…
cwise_op_cos.cc
…
cwise_op_cosh.cc
…
cwise_op_digamma.cc
…
cwise_op_div.cc
Enabling several cwise op kernels for ROCm
2020-01-16 22:12:08 -08:00
cwise_op_equal_to_1.cc
Add uint16, uint32, uint64 support for tf.math.equal
2020-04-06 22:42:34 +00:00
cwise_op_equal_to_2.cc
…
cwise_op_erf.cc
…
cwise_op_erfc.cc
…
cwise_op_exp.cc
…
cwise_op_expm1.cc
…
cwise_op_floor_div.cc
…
cwise_op_floor_mod.cc
Add uint64 support to DynamicPartition and FloorMod on CPU.
2020-03-17 09:34:04 -07:00
cwise_op_floor.cc
Add int32 support to floor, ceil & rint
2019-12-04 18:47:55 -08:00
cwise_op_gpu_abs.cu.cc
…
cwise_op_gpu_acos.cu.cc
…
cwise_op_gpu_acosh.cu.cc
…
cwise_op_gpu_add.cu.cc
Support uint32 in tf.add.
2020-04-07 18:30:13 -07:00
cwise_op_gpu_arg.cu.cc
Enabling several cwise op kernels for ROCm
2020-01-16 22:12:08 -08:00
cwise_op_gpu_asin.cu.cc
…
cwise_op_gpu_asinh.cu.cc
…
cwise_op_gpu_atan2.cu.cc
…
cwise_op_gpu_atan.cu.cc
…
cwise_op_gpu_atanh.cu.cc
…
cwise_op_gpu_bitwise_and.cu.cc
…
cwise_op_gpu_bitwise_or.cu.cc
…
cwise_op_gpu_bitwise_xor.cu.cc
…
cwise_op_gpu_ceil.cu.cc
…
cwise_op_gpu_complex.cu.cc
…
cwise_op_gpu_conj.cu.cc
…
cwise_op_gpu_cos.cu.cc
…
cwise_op_gpu_cosh.cu.cc
…
cwise_op_gpu_digamma.cu.cc
…
cwise_op_gpu_div.cu.cc
Enabling several cwise op kernels for ROCm
2020-01-16 22:12:08 -08:00
cwise_op_gpu_equal_to.cu.cc
…
cwise_op_gpu_erf.cu.cc
…
cwise_op_gpu_erfc.cu.cc
…
cwise_op_gpu_exp.cu.cc
…
cwise_op_gpu_expm1.cu.cc
…
cwise_op_gpu_floor_div.cu.cc
…
cwise_op_gpu_floor_mod.cu.cc
…
cwise_op_gpu_floor.cu.cc
…
cwise_op_gpu_greater_equal.cu.cc
…
cwise_op_gpu_greater.cu.cc
…
cwise_op_gpu_igammas_double.cu.cc
…
cwise_op_gpu_igammas_float.cu.cc
…
cwise_op_gpu_imag.cu.cc
…
cwise_op_gpu_inverse.cu.cc
…
cwise_op_gpu_invert.cu.cc
…
cwise_op_gpu_isfinite.cu.cc
…
cwise_op_gpu_isinf.cu.cc
…
cwise_op_gpu_isnan.cu.cc
…
cwise_op_gpu_left_shift.cu.cc
…
cwise_op_gpu_less_equal.cu.cc
…
cwise_op_gpu_less.cu.cc
…
cwise_op_gpu_lgamma.cu.cc
…
cwise_op_gpu_log1p.cu.cc
…
cwise_op_gpu_log.cu.cc
…
cwise_op_gpu_logical_and.cu.cc
…
cwise_op_gpu_logical_not.cu.cc
…
cwise_op_gpu_logical_or.cu.cc
…
cwise_op_gpu_maximum.cu.cc
Improves tf-numpy's clip to pass testClipStaticBounds, and adds uint8 kernels to tf.minimum and tf.maximum.
2020-04-22 13:22:24 -07:00
cwise_op_gpu_minimum.cu.cc
Improves tf-numpy's clip to pass testClipStaticBounds, and adds uint8 kernels to tf.minimum and tf.maximum.
2020-04-22 13:22:24 -07:00
cwise_op_gpu_mod.cu.cc
…
cwise_op_gpu_mul.cu.cc
…
cwise_op_gpu_neg.cu.cc
Expand dtype support for Neg
2020-06-18 20:31:51 -07:00
cwise_op_gpu_not_equal_to.cu.cc
…
cwise_op_gpu_pow.cu.cc
…
cwise_op_gpu_random_grad.cu.cc
…
cwise_op_gpu_real.cu.cc
…
cwise_op_gpu_right_shift.cu.cc
…
cwise_op_gpu_rint.cu.cc
…
cwise_op_gpu_round.cu.cc
…
cwise_op_gpu_rsqrt.cu.cc
…
cwise_op_gpu_select.cu.cc
…
cwise_op_gpu_sigmoid.cu.cc
…
cwise_op_gpu_sign.cu.cc
…
cwise_op_gpu_sin.cu.cc
…
cwise_op_gpu_sinh.cu.cc
…
cwise_op_gpu_sqrt.cu.cc
…
cwise_op_gpu_square.cu.cc
…
cwise_op_gpu_squared_difference.cu.cc
…
cwise_op_gpu_sub.cu.cc
Adding these ops to enable the use of uint32 for indices in sparse matrices, since int32 has issues on GPU. Adding support for uint32 for:
2020-03-03 14:24:41 -08:00
cwise_op_gpu_tan.cu.cc
…
cwise_op_gpu_tanh.cu.cc
Add support for f16 generated tanh kernel.
2020-06-15 10:12:37 -07:00
cwise_op_gpu_xdivy.cu.cc
…
cwise_op_gpu_xlog1py.cu.cc
Add tf.math.xlog1py, a safe way to compute x * log1p(y)
2020-01-09 14:28:37 -08:00
cwise_op_gpu_xlogy.cu.cc
…
cwise_op_gpu_zeta.cu.cc
…
cwise_op_greater_equal.cc
…
cwise_op_greater.cc
…
cwise_op_igammas.cc
…
cwise_op_imag.cc
…
cwise_op_invert.cc
…
cwise_op_isfinite.cc
…
cwise_op_isinf.cc
…
cwise_op_isnan.cc
…
cwise_op_left_shift.cc
…
cwise_op_less_equal.cc
…
cwise_op_less.cc
…
cwise_op_lgamma.cc
…
cwise_op_log1p.cc
…
cwise_op_log.cc
…
cwise_op_logical_and.cc
…
cwise_op_logical_not.cc
…
cwise_op_logical_or.cc
…
cwise_op_maximum.cc
Improves tf-numpy's clip to pass testClipStaticBounds, and adds uint8 kernels to tf.minimum and tf.maximum.
2020-04-22 13:22:24 -07:00
cwise_op_minimum.cc
Improves tf-numpy's clip to pass testClipStaticBounds, and adds uint8 kernels to tf.minimum and tf.maximum.
2020-04-22 13:22:24 -07:00
cwise_op_mod.cc
…
cwise_op_mul_1.cc
Add a CPU bfloat16 kernel for MulNoNan
2020-05-01 16:37:16 -07:00
cwise_op_mul_2.cc
…
cwise_op_ndtri_double.cu.cc
…
cwise_op_ndtri_float.cu.cc
…
cwise_op_ndtri.cc
…
cwise_op_neg_1.cc
Expand dtype support for Neg
2020-06-18 20:31:51 -07:00
cwise_op_neg_2.cc
Expand dtype support for Neg
2020-06-18 20:31:51 -07:00
cwise_op_not_equal_to_1.cc
Merge pull request #38288 from yongtang:26069-equal-dtype
2020-04-21 11:00:34 -07:00
cwise_op_not_equal_to_2.cc
…
cwise_op_pow.cc
…
cwise_op_random_grad.cc
…
cwise_op_real.cc
…
cwise_op_reciprocal.cc
Addressing review comments - round 2
2019-10-14 19:02:45 -07:00
cwise_op_right_shift.cc
…
cwise_op_rint.cc
Add int32 support to floor, ceil & rint
2019-12-04 18:47:55 -08:00
cwise_op_round.cc
…
cwise_op_rsqrt.cc
…
cwise_op_select.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
cwise_op_sigmoid.cc
…
cwise_op_sign.cc
…
cwise_op_sin.cc
…
cwise_op_sinh.cc
…
cwise_op_sqrt.cc
…
cwise_op_square.cc
…
cwise_op_squared_difference.cc
…
cwise_op_sub.cc
Adding these ops to enable the use of uint32 for indices in sparse matrices, since int32 has issues on GPU. Adding support for uint32 for:
2020-03-03 14:24:41 -08:00
cwise_op_tan.cc
…
cwise_op_tanh.cc
Add support for f16 generated tanh kernel.
2020-06-15 10:12:37 -07:00
cwise_op_xdivy.cc
…
cwise_op_xlog1py.cc
Add tf.math.xlog1py, a safe way to compute x * log1p(y)
2020-01-09 14:28:37 -08:00
cwise_op_xlogy.cc
Enabling several cwise op kernels for ROCm
2020-01-16 22:12:08 -08:00
cwise_op_zeta.cc
…
cwise_ops_common.cc
…
cwise_ops_common.h
Fix typo
2020-03-11 10:38:38 +00:00
cwise_ops_gpu_common.cu.h
Add _USE_MATH_DEFINES to cwise_ops_gpu_common.cu.h to fix Windows GPU build.
2019-12-19 11:07:54 -08:00
cwise_ops_gpu_gradients.cu.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
cwise_ops_gradients.h
[ROCm] Enabling vectorization/packet support for some cwise ops on the ROCm platform
2020-01-06 21:53:49 +00:00
cwise_ops_sycl_common.h
…
cwise_ops_test.cc
Enabling several cwise op kernels for ROCm
2020-01-16 22:12:08 -08:00
cwise_ops.h
Add Bessel functions to the public api:
2020-06-17 21:06:09 -07:00
data_format_ops_gpu.cu.cc
…
data_format_ops.cc
DataFormatVecPermute accepts a vector of size 2.
2020-05-20 16:01:33 -07:00
data_format_ops.h
…
debug_ops_gpu.cu.cc
[tfdbg] Fix DebugNumericSummaryV2OP kernel mode=CURT_HEALTH breakage on Windows GPU builds
2019-12-20 08:53:47 -08:00
debug_ops_test.cc
[tfdbg] Remove use of dirent.h in debug_ops_test.cc
2020-02-02 19:14:32 -08:00
debug_ops.cc
[tfdbg] enable_dump_debug_info(): Support CONCISE_HEALTH and SHAPE modes
2019-12-17 17:43:32 -08:00
debug_ops.h
[tfdbg2] Add tfdbg_run_id to metadata of data dumps
2020-06-15 13:52:52 -07:00
decode_bmp_op.cc
…
decode_compressed_op.cc
…
decode_csv_op.cc
…
decode_image_op.cc
change tf.image.decode_image to tf.io.decode_image
2020-03-13 11:59:24 +08:00
decode_padded_raw_op.cc
…
decode_proto_op.cc
Fixes/updates for tensorflow::tstring migration.
2019-11-26 23:35:38 -08:00
decode_raw_op.cc
[tstring] Avoid temporary std::string creation in DecodeRawOp.
2020-01-23 14:48:29 -08:00
decode_wav_op_test.cc
…
decode_wav_op.cc
…
deep_conv2d_test.cc
…
deep_conv2d.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
deep_conv2d.h
…
dense_update_functor_gpu.cu.cc
Adding these ops to enable the use of uint32 for indices in sparse matrices, since int32 has issues on GPU. Adding support for uint32 for:
2020-03-03 14:24:41 -08:00
dense_update_functor.cc
…
dense_update_functor.h
…
dense_update_ops.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
depthtospace_op_gpu.cu.cc
…
depthtospace_op.cc
…
depthtospace_op.h
…
depthwise_conv_grad_op.cc
Merge pull request #40303 from kaixih:test_nhwc_depwise_pr
2020-06-10 21:26:08 -07:00
depthwise_conv_op_gpu_double.cu.cc
…
depthwise_conv_op_gpu_float.cu.cc
…
depthwise_conv_op_gpu_half.cu.cc
…
depthwise_conv_op_gpu.h
fix LaunchDepthwiseConvBackpropFilterOp
2020-06-16 23:38:53 +08:00
depthwise_conv_op.cc
Merge pull request #40303 from kaixih:test_nhwc_depwise_pr
2020-06-10 21:26:08 -07:00
depthwise_conv_op.h
Add explicit padding to depthwise conv2d.
2020-03-24 12:28:49 -07:00
depthwise_conv_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
dequantize_op_test.cc
Support Dequantize to bfloat16.
2020-01-14 11:58:53 -08:00
dequantize_op.cc
fix dequantize op regression issue
2020-05-08 07:34:54 +08:00
deserialize_sparse_string_op.cc
Rename tensorflow::Reshape() to tensorflow::ReshapeSparseTensor().
2020-04-08 10:47:41 -07:00
deserialize_sparse_variant_op.cc
…
determinant_op_gpu.cu.cc
…
determinant_op.cc
…
determinant_op.h
…
diag_op_gpu.cu.cc
Register Eigen::half kernels for Diag and DiagPart
2020-05-04 18:57:36 -07:00
diag_op_test.cc
…
diag_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
diag_op.h
…
dilation_ops_gpu.cu.cc
…
dilation_ops.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
dilation_ops.h
…
draw_bounding_box_op.cc
…
dynamic_partition_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
dynamic_partition_op_test.cc
…
dynamic_partition_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
dynamic_stitch_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
dynamic_stitch_op_test.cc
…
dynamic_stitch_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
edit_distance_op.cc
…
eig_op_complex64.cc
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
eig_op_complex128.cc
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
eig_op_double.cc
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
eig_op_float.cc
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
eig_op_impl.h
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
eigen_activations_test.cc
…
eigen_activations.h
…
eigen_attention_test.cc
…
eigen_attention.h
Merge pull request #38549 from yongtang:12829-extract_glimpse
2020-05-21 12:14:19 -07:00
eigen_backward_cuboid_convolutions_test.cc
Split eigen_backward_spatial_convolutions_test into two.
2020-01-24 16:07:33 -08:00
eigen_backward_cuboid_convolutions.h
…
eigen_backward_spatial_convolutions_test.cc
Split eigen_backward_spatial_convolutions_test into two.
2020-01-24 16:07:33 -08:00
eigen_backward_spatial_convolutions.h
…
eigen_benchmark_cpu_test.cc
…
eigen_benchmark.h
…
eigen_contraction_kernel.cc
Internal Copybara change.
2020-03-20 12:06:08 -07:00
eigen_contraction_kernel.h
Suppress Clang error: unused variable 'row_base' when building with MSAN instrumentation
2020-06-19 13:17:03 -07:00
eigen_convolution_helpers.h
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
eigen_cuboid_convolution.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
eigen_mkldnn_contraction_kernel_test.cc
Update Eigen to c906e1142b6b.
2019-10-08 19:11:15 -07:00
eigen_pooling_test.cc
…
eigen_pooling.h
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
eigen_spatial_convolutions_test.cc
Fix bug in padding detection in contraction packing
2020-01-09 18:53:24 -08:00
eigen_spatial_convolutions-inl.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
eigen_spatial_convolutions.h
Optimize col-major packing for PADDING_VALID convolutions
2020-01-09 11:36:57 -08:00
eigen_volume_patch.h
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
einsum_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
einsum_op_impl_bfloat16.cc
…
einsum_op_impl_complex64.cc
…
einsum_op_impl_complex128.cc
…
einsum_op_impl_double.cc
…
einsum_op_impl_float.cc
…
einsum_op_impl_half.cc
…
einsum_op_impl_int32.cc
…
einsum_op_impl_int64.cc
…
einsum_op_impl.h
Fix bug in complex einsum grad.
2020-05-05 19:46:15 -07:00
einsum_op.h
…
encode_jpeg_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
encode_jpeg_op.cc
…
encode_png_op.cc
…
encode_proto_op.cc
…
encode_wav_op_test.cc
…
encode_wav_op.cc
…
example_parsing_ops_test.cc
[ParseExample] Reduce variance in benchmark.
2020-02-26 17:09:20 -08:00
example_parsing_ops.cc
Remove all uses of TENSORFLOW_LITE_PROTOS.
2020-05-06 02:43:41 -07:00
extract_image_patches_op_gpu.cu.cc
Add complex number support for tf.extract_image_patches
2020-01-17 21:39:58 +00:00
extract_image_patches_op.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
extract_image_patches_op.h
…
extract_jpeg_shape_op.cc
…
extract_volume_patches_op_gpu.cu.cc
…
extract_volume_patches_op.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
extract_volume_patches_op.h
…
eye_functor_gpu.cu.cc
…
eye_functor.h
…
fact_op.cc
…
fake_quant_ops_functor.h
…
fake_quant_ops_gpu.cu.cc
…
fake_quant_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
fake_quant_ops.cc
PR #36526: Removed NDIMS template arg from BinaryElementWiseOp::Operate
2020-02-26 20:38:14 -08:00
fft_ops.cc
…
fifo_queue_op.cc
Add an internal GPU-compatible version of FIFOQueue.
2020-01-06 10:10:52 -08:00
fifo_queue.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
fifo_queue.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
fill_functor.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
fill_functor.cu.cc
…
fill_functor.h
…
fingerprint_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
fingerprint_op.cc
…
fixed_length_record_reader_op.cc
…
fractional_avg_pool_op.cc
…
fractional_max_pool_op.cc
…
fractional_pool_common.cc
…
fractional_pool_common.h
…
function_ops.cc
[ArgOp] Use CanConsumeArg() and ConsumeArg() when possible.
2020-05-06 09:19:01 -07:00
function_ops.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
functional_ops.cc
Add a function to dynamic choose and execute the proper implementation based on underlying device placement
2020-05-28 09:54:31 -07:00
fused_batch_norm_ex_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
fused_batch_norm_op_test.cc
Automated rollback of commit 80acd88cd43f09a1a2980792e3955f2ce5147bfd
2020-02-18 13:51:51 -08:00
fused_batch_norm_op.cc
Support computing exponential running mean and variance in fused_batch_norm.
2020-02-26 14:31:32 -08:00
fused_batch_norm_op.cu.cc
Remove vestigial code from cuDNN 4 in stream executor.
2020-02-06 10:15:11 -08:00
fused_batch_norm_op.h
Remove vestigial code from cuDNN 4 in stream executor.
2020-02-06 10:15:11 -08:00
fused_eigen_output_kernels.cc
…
fused_eigen_output_kernels.h
…
gather_functor_batched_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_functor_batched_gpu.cu.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
gather_functor_batched.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_functor_batched.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
gather_functor_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_functor_gpu.cu.h
adjust gather ops launch config. for NCF model, this means ~20% gain. (due to grid size from 80->160 on volta).
2020-05-19 16:12:30 -07:00
gather_functor.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_functor.h
Fix HandleCopies so it no longer invokes UB when the params or out TensorMaps are empty with no backing data.
2020-05-28 17:01:07 -07:00
gather_nd_op_cpu_impl_0.cc
…
gather_nd_op_cpu_impl_1.cc
…
gather_nd_op_cpu_impl_2.cc
…
gather_nd_op_cpu_impl_3.cc
…
gather_nd_op_cpu_impl_4.cc
…
gather_nd_op_cpu_impl_5.cc
…
gather_nd_op_cpu_impl_6.cc
…
gather_nd_op_cpu_impl_7.cc
…
gather_nd_op_cpu_impl.h
…
gather_nd_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_nd_op_test.cc
…
gather_nd_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_nd_op.h
…
gather_op_test.cc
Fix parameter check for batchdim in Gather.
2020-05-19 15:58:33 -07:00
gather_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
gemm_functors.h
…
generate_box_proposals_op.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
generate_vocab_remapping_op.cc
…
gpu_device_array_gpu.h
…
gpu_device_array.h
Remove unused unique_tensor_references library.
2020-03-12 16:35:29 -07:00
gpu_prim.h
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
gpu_utils.cc
PR #36267: [ROCm] Reverting ROCm to use MIOpen Find Mode APIs (be default) for convolution
2020-04-08 00:44:58 -07:00
gpu_utils.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
guarantee_const_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
guarantee_const_op.cc
…
hinge-loss.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
histogram_op_gpu.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
histogram_op.cc
…
histogram_op.h
…
host_constant_op.cc
…
host_constant_op.h
Internal change
2020-01-10 17:44:33 -08:00
i_remote_fused_graph_executor.h
…
i_remote_fused_graph_ops_definitions.cc
…
i_remote_fused_graph_ops_definitions.h
…
identity_n_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
identity_n_op.cc
…
identity_n_op.h
…
identity_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
identity_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
identity_op.h
…
identity_reader_op.cc
…
image_ops_gpu.cu.cc
Image Translation Keras layer for preprocessing.
2020-01-29 18:58:47 -08:00
image_ops.cc
Export fill mode for image_projective_transform
2020-03-12 16:54:14 -07:00
image_ops.h
Export fill mode for image_projective_transform
2020-03-12 16:54:14 -07:00
image_resizer_state.h
…
immutable_constant_op_test.cc
…
immutable_constant_op.cc
…
immutable_constant_op.h
…
in_topk_op_gpu.cu.cc
Fix in_top_k on GPU with empty input.
2020-03-27 09:12:21 -07:00
in_topk_op_test.cc
…
in_topk_op.cc
…
in_topk_op.h
…
initializable_lookup_table.cc
Simplify and optimize fastpath for initializable lookuptable.
2019-12-13 01:10:21 -08:00
initializable_lookup_table.h
Simplify and optimize fastpath for initializable lookuptable.
2019-12-13 01:10:21 -08:00
inplace_ops_functor_gpu.cu.cc
…
inplace_ops_functor.h
…
inplace_ops.cc
Make inplace_op behavior consistent on CPU/GPU, for empty tensors.
2020-01-28 12:36:00 -08:00
kernel_platform_strings.h
…
l2loss_op_gpu.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
l2loss_op.cc
…
l2loss_op.h
…
linalg_ops_common.cc
Return if an error status was set in LinearAlgebraOp::AnalyzeInput() or LinearAlgebraOp::PrepareOutputs(), instead of potentially crashing.
2019-11-11 15:51:45 -08:00
linalg_ops_common.h
Merge pull request #28340 from Randl:eig
2019-10-14 20:47:54 -07:00
list_kernels.cc
Define TensorList class in a separate library
2020-01-24 17:37:07 -08:00
list_kernels.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
list_kernels.h
Don't concatenate empty tensors
2020-05-07 17:32:16 -07:00
listdiff_op.cc
…
lmdb_reader_op.cc
…
load_and_remap_matrix_op.cc
…
logging_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
logging_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
logging_ops.h
…
logistic-loss.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
lookup_table_init_op.cc
Allowing tf.lookup tables to be initialized from a Dataset
2020-03-02 15:41:42 -08:00
lookup_table_init_op.h
…
lookup_table_op.cc
add TF_LOCKS_EXCLUDED to MutableDenseHashTable::MemoryUsed
2020-05-27 13:27:18 +08:00
lookup_table_op.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
lookup_util.cc
Allowing tf.lookup tables to be initialized from a Dataset
2020-03-02 15:41:42 -08:00
lookup_util.h
Allowing tf.lookup tables to be initialized from a Dataset
2020-03-02 15:41:42 -08:00
loss_test.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
loss.h
…
lrn_op_test.cc
Fix bad index type for Eigen::DSizes.
2019-10-17 21:14:22 -07:00
lrn_op.cc
…
lu_op_gpu.cu.cc
…
lu_op.cc
…
map_stage_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
matching_files_op.cc
…
matmul_op_fused.cc
…
matmul_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
matmul_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matmul_op.h
…
matrix_band_part_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_band_part_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_band_part_op.h
…
matrix_diag_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_diag_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_diag_op.h
Add Matrix{Diag,SetDiag,DiagPart}V3 ops with alignment options.
2019-11-14 16:02:19 -08:00
matrix_exponential_op.cc
…
matrix_inverse_op.cc
…
matrix_logarithm_op.cc
…
matrix_set_diag_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_set_diag_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_set_diag_op.h
Add Matrix{Diag,SetDiag,DiagPart}V3 ops with alignment options.
2019-11-14 16:02:19 -08:00
matrix_solve_ls_op_complex64.cc
…
matrix_solve_ls_op_complex128.cc
…
matrix_solve_ls_op_double.cc
…
matrix_solve_ls_op_float.cc
…
matrix_solve_ls_op_impl.h
…
matrix_solve_op.cc
…
matrix_square_root_op.cc
…
matrix_triangular_solve_op_complex.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
matrix_triangular_solve_op_impl.h
[ROCm] Fix for a bug in the ROCm implementation for matrix_triangular_solve op
2020-02-12 15:14:49 +00:00
matrix_triangular_solve_op_real.cc
Add Broadcasted Matrix Triangular Solve.
2020-01-27 20:43:43 -08:00
matrix_triangular_solve_op_test.cc
Add Broadcasted Matrix Triangular Solve.
2020-01-27 20:43:43 -08:00
maxpooling_op_gpu.cu.cc
…
maxpooling_op_gpu.h
…
maxpooling_op.cc
…
maxpooling_op.h
…
merge_v2_checkpoints_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
meta_support.cc
…
meta_support.h
…
mfcc_dct_test.cc
…
mfcc_dct.cc
…
mfcc_dct.h
…
mfcc_mel_filterbank_test.cc
…
mfcc_mel_filterbank.cc
Merge pull request #36405 from gaurav1086:center_frequencies_index_access_check
2020-02-26 09:28:31 -08:00
mfcc_mel_filterbank.h
…
mfcc_op_test.cc
…
mfcc_op.cc
…
mfcc_test.cc
…
mfcc.cc
…
mfcc.h
…
mirror_pad_op_benchmark_test.cc
change MirrorPad packet region
2019-11-27 08:52:42 +08:00
mirror_pad_op_cpu_impl_1.cc
…
mirror_pad_op_cpu_impl_2.cc
…
mirror_pad_op_cpu_impl_3.cc
…
mirror_pad_op_cpu_impl_4.cc
…
mirror_pad_op_cpu_impl_5.cc
…
mirror_pad_op_cpu_impl.h
…
mirror_pad_op_gpu.cu.cc
…
mirror_pad_op_test.cc
Explicitly capture variables in lambdas.
2020-01-10 14:42:49 -08:00
mirror_pad_op.cc
…
mirror_pad_op.h
Merge pull request #34628 from Intel-tensorflow:yang/mirrorpad
2020-01-02 17:22:31 -08:00
mkl_aggregate_ops.cc
[Intel MKL] Adding DNNL ops (part 2) supporting threadpool work
2020-06-15 13:22:19 -07:00
mkl_avgpooling_op.cc
Merge pull request #39519 from Intel-tensorflow:sriniva2/threadpool_pooling
2020-05-27 12:16:52 -07:00
mkl_batch_matmul_op.cc
Merge pull request #40128 from Intel-tensorflow:sriniva2/tp_batch_matmul
2020-06-08 12:31:59 -07:00
mkl_concat_op.cc
[Intel MKL] Adding DNNL ops (part 2) supporting threadpool work
2020-06-15 13:22:19 -07:00
mkl_conv_grad_filter_ops.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_conv_grad_input_ops.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_conv_ops_test.cc
Addressing review comments
2020-06-10 09:15:38 -07:00
mkl_conv_ops.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_conv_ops.h
Bug Fix : resolve cast error by adding cleaner type for MKL
2020-03-21 18:13:50 +01:00
mkl_cwise_ops_common.cc
…
mkl_dequantize_op_test.cc
Add two newlines.
2020-03-01 14:49:16 -08:00
mkl_dequantize_op.cc
[Intel MKL] Adding DNNL ops (part 2) supporting threadpool work
2020-06-15 13:22:19 -07:00
mkl_fused_batch_norm_op_test.cc
Minor follow-up fixes for PR #37176.
2020-03-02 10:44:26 -08:00
mkl_fused_batch_norm_op.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_fused_ops_test.cc
Merge test cases and fix code style
2020-04-15 22:55:44 +08:00
mkl_identity_op.cc
Added changes for DNN 0.9 to softmax, identity_op, and lrn ops.
2020-02-10 15:40:01 -08:00
mkl_input_conversion_op.cc
Boiler plate code for dnnl threadpool and mkl_conv_ops.cc
2020-04-05 21:24:01 -07:00
mkl_lrn_op.cc
[Intel MKL] Adding DNNL ops (part 2) supporting threadpool work
2020-06-15 13:22:19 -07:00
mkl_matmul_op_fused.cc
Merge pull request #40254 from Intel-tensorflow:sriniva2/tp_qmatmul
2020-06-08 16:16:14 -07:00
mkl_matmul_op.cc
Merge pull request #40254 from Intel-tensorflow:sriniva2/tp_qmatmul
2020-06-08 16:16:14 -07:00
mkl_matmul_ops_common.h
matmul,qmatmul and fusedops support for threadpool api.
2020-06-07 15:48:07 -07:00
mkl_maxpooling_op.cc
Merge pull request #39519 from Intel-tensorflow:sriniva2/threadpool_pooling
2020-05-27 12:16:52 -07:00
mkl_pooling_ops_common.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_pooling_ops_common.h
Merge pull request #39519 from Intel-tensorflow:sriniva2/threadpool_pooling
2020-05-27 12:16:52 -07:00
mkl_qmatmul_op_test.cc
Merge pull request #35285 from kiszk:spelling_tweaks_core
2020-02-28 10:58:32 -08:00
mkl_qmatmul_op.cc
Merge pull request #39893 from Intel-tensorflow:sshiddib/dnnl_threadpool_openmp
2020-06-09 11:02:45 -07:00
mkl_quantize_op_test.cc
[INTEL MKL] Enabled MIN_FIRST support and primitive caching for MKL-DNN Quantize OP
2019-10-28 12:54:08 +08:00
mkl_quantize_op.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_quantized_concat_op_test.cc
…
mkl_quantized_conv_ops_perchannel_test.cc
…
mkl_quantized_conv_ops_test.cc
minor formatting changes.
2019-10-08 08:56:40 -07:00
mkl_quantized_conv_ops.h
Update tensorflow/core/kernels/mkl_quantized_conv_ops.h
2020-06-08 10:30:03 -07:00
mkl_quantized_pooling_ops_test.cc
…
mkl_relu_op_test.cc
Merge pull request #40598 from Intel-tensorflow:nhasabni/mkl_tanh_buildfix
2020-06-20 14:02:17 -07:00
mkl_relu_op.cc
Merge pull request #40598 from Intel-tensorflow:nhasabni/mkl_tanh_buildfix
2020-06-20 14:02:17 -07:00
mkl_requantization_range_per_channel_op.cc
Merge pull request #39893 from Intel-tensorflow:sshiddib/dnnl_threadpool_openmp
2020-06-09 11:02:45 -07:00
mkl_requantize_ops_test.cc
…
mkl_requantize_per_channel_op.cc
Threadpool api support for misc ops.
2020-05-12 13:18:53 -07:00
mkl_reshape_op.cc
Boiler plate code for dnnl threadpool and mkl_conv_ops.cc
2020-04-05 21:24:01 -07:00
mkl_slice_op.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_softmax_op.cc
[Intel MKL] Adding DNNL ops (part 1) supporting threadpool work
2020-06-15 13:14:53 -07:00
mkl_tfconv_op.h
Boiler plate code for dnnl threadpool and mkl_conv_ops.cc
2020-04-05 21:24:01 -07:00
mkl_tmp_bf16_ops.cc
remove duplicate registration for softmax bf16 op
2020-06-05 16:09:52 -07:00
mkl_transpose_op.cc
[Intel MKL] Adding DNNL ops (part 2) supporting threadpool work
2020-06-15 13:22:19 -07:00
mlir_generated_cwise_op_gpu_tanh.cu.cc
Also set the unroll factor in the C++ integration code for tanh.
2020-06-18 05:32:31 -07:00
mlir_generated_op_gpu_tanh_test.cc
Add a small test to cover the mlir generated Tanh GPU kernel.
2020-06-19 01:52:25 -07:00
multinomial_op_gpu.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
multinomial_op_test.cc
…
multinomial_op.cc
…
multinomial_op.h
…
mutex_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
nccl_ops.cc
…
nextafter_op_gpu.cu.cc
…
nextafter_op.cc
Define missing EIGEN_USE_THREADS in Tensorflow kernels.
2019-10-28 09:49:58 -07:00
nextafter_op.h
…
nn_ops_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
no_op.cc
…
no_op.h
…
non_max_suppression_op_benchmark_test.cc
refine the cost in NMS and add more test
2020-02-24 21:17:21 +08:00
non_max_suppression_op_gpu_test.cc
Make //tensorflow/core/kernels:non_max_suppression_op_gpu_test IWYU clean.
2020-01-28 17:20:20 -08:00
non_max_suppression_op_test.cc
Internal change
2020-02-26 16:38:57 -08:00
non_max_suppression_op.cc
Internal change
2020-02-26 16:38:57 -08:00
non_max_suppression_op.cu.cc
Remove duplicate macros
2020-06-09 05:13:35 +00:00
non_max_suppression_op.h
…
nth_element_op.cc
…
nth_element_op.h
…
one_hot_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
one_hot_op_test.cc
Optimize OneHot op on CPU
2019-10-22 12:49:09 -07:00
one_hot_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
one_hot_op.h
Optimize OneHot op on CPU
2019-10-22 12:49:09 -07:00
ops_testutil_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ops_testutil.cc
Optimize (legacy, reference-typed) VariableOp::Compute().
2020-03-27 14:48:18 -07:00
ops_testutil.h
Merge pull request #36852 from tfeher:allow_build_at_runtime
2020-02-20 13:35:32 -08:00
ops_util_test.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
ops_util.h
…
pack_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
pad_op_gpu.cu.cc
…
pad_op.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
pad_op.h
…
padding_fifo_queue_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
padding_fifo_queue.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
padding_fifo_queue.h
…
parameterized_truncated_normal_op_gpu.cu.cc
…
parameterized_truncated_normal_op_test.cc
…
parameterized_truncated_normal_op.cc
Add StatelessParameterizedTruncatedNormal sampler.
2020-06-18 10:28:38 -07:00
parameterized_truncated_normal_op.h
Add StatelessParameterizedTruncatedNormal sampler.
2020-06-18 10:28:38 -07:00
parse_tensor_op.cc
…
parse_tensor_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
partitioned_function_ops.cc
Add ability for functions to share rendezvous
2020-06-08 11:47:46 -07:00
partitioned_function_ops.h
Add ability for functions to share rendezvous
2020-06-08 11:47:46 -07:00
poisson-loss.h
…
pooling_ops_3d_gpu.cu.cc
…
pooling_ops_3d_gpu.h
…
pooling_ops_3d_sycl.h
…
pooling_ops_3d.cc
Don't crash in 3D pooling ops with empty batch size on GPU.
2020-05-26 17:33:38 -07:00
pooling_ops_3d.h
…
pooling_ops_common_gpu.h
…
pooling_ops_common.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
pooling_ops_common.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
population_count_op_gpu.cu.cc
…
population_count_op.cc
…
population_count_op.h
…
priority_queue_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
priority_queue.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
priority_queue.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
qr_op_complex64.cc
…
qr_op_complex128.cc
…
qr_op_double.cc
…
qr_op_float.cc
…
qr_op_impl.h
…
quantization_utils_test.cc
Internal change
2020-03-05 11:36:56 -08:00
quantization_utils.cc
…
quantization_utils.h
segrating changes 2
2020-06-15 21:18:56 +00:00
quantize_and_dequantize_op_gpu.cu.cc
Temporary rollback of QuantizeAndDequantizeV2Grad.
2020-02-21 14:06:30 -08:00
quantize_and_dequantize_op_test.cc
Temporary rollback of QuantizeAndDequantizeV2Grad.
2020-02-21 14:06:30 -08:00
quantize_and_dequantize_op.cc
Temporary rollback of QuantizeAndDequantizeV2Grad.
2020-02-21 14:06:30 -08:00
quantize_and_dequantize_op.h
Temporary rollback of QuantizeAndDequantizeV2Grad.
2020-02-21 14:06:30 -08:00
quantize_down_and_shrink_range_op_test.cc
…
quantize_down_and_shrink_range.cc
…
quantize_op_test.cc
Remove uses of rand_r in quantize/dequantize op_test.
2020-01-11 14:35:01 -08:00
quantize_op.cc
Rename scalar_round_op_google -> scalar_round_half_to_even_op to document its semantics.
2019-12-17 12:47:10 -08:00
quantized_activation_ops_test.cc
…
quantized_activation_ops.cc
…
quantized_add_op_test.cc
Internal change
2020-03-05 11:36:56 -08:00
quantized_add_op.cc
…
quantized_batch_norm_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
quantized_batch_norm_op.cc
…
quantized_bias_add_op_test.cc
…
quantized_bias_add_op.cc
…
quantized_concat_op_test.cc
…
quantized_concat_op.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
quantized_conv_ops_test.cc
…
quantized_conv_ops.cc
Splitting out common_shape_fns.cc:
2020-03-03 14:28:49 -08:00
quantized_instance_norm_test.cc
Internal change
2020-03-05 11:36:56 -08:00
quantized_instance_norm.cc
…
quantized_matmul_op_test.cc
…
quantized_matmul_op.cc
…
quantized_mul_op_test.cc
Internal change
2020-03-05 11:36:56 -08:00
quantized_mul_op.cc
…
quantized_pooling_ops_test.cc
…
quantized_pooling_ops.cc
…
quantized_reshape_op_test.cc
…
quantized_reshape_op.cc
…
quantized_resize_bilinear_op_test.cc
[Build cleanup] Split "core_cpu_impl" into fine-grained targets (4/n).
2020-04-28 09:53:59 -07:00
quantized_resize_bilinear_op.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
queue_base.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
queue_base.h
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
queue_op.cc
…
queue_op.h
…
queue_ops.cc
…
ragged_cross_op.cc
Add tf.ragged.cross and tf.ragged.cross_hashed operations. These ops generate feature crosses from a list of input tensors (similar to tf.sparse.cross), and return the result as a RaggedTensor. Inputs may be tf.RaggedTensors, tf.Tensors, or tf.SparseTensors.
2020-03-02 18:13:45 -08:00
ragged_gather_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_gather_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
ragged_range_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_range_op.cc
…
ragged_tensor_from_variant_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_tensor_from_variant_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
ragged_tensor_to_sparse_kernel_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_tensor_to_sparse_kernel.cc
…
ragged_tensor_to_tensor_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_tensor_to_tensor_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
ragged_tensor_to_variant_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
ragged_tensor_to_variant_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
random_binomial_op_test.cc
…
random_binomial_op.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
random_binomial_op.h
…
random_crop_op.cc
…
random_op_cpu.h
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
random_op_gpu.cu.cc
Modifies tf.random.stateless_uniform to support unbounded int sampling (when minval=maxval=None). Adds a StatelessRandomUniformFullInt op to support this.
2020-02-13 09:00:36 -08:00
random_op_gpu.h
…
random_op_test.cc
…
random_op.cc
Change log to log1p in Random Gamma sampler.
2020-04-22 11:07:43 -07:00
random_op.h
…
random_poisson_op_test.cc
…
random_poisson_op.cc
Adds a tf.random.stateless_poisson sampler for CPU.
2020-02-07 10:05:56 -08:00
random_poisson_op.h
Adds a tf.random.stateless_poisson sampler for CPU.
2020-02-07 10:05:56 -08:00
random_shuffle_op.cc
…
random_shuffle_queue_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
range_sampler_test.cc
…
range_sampler.cc
…
range_sampler.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
reader_ops.cc
…
record_input_op.cc
…
record_yielder.cc
…
record_yielder.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
reduce_join_op.cc
…
reduction_gpu_kernels.cu.h
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
reduction_ops_all.cc
…
reduction_ops_any.cc
…
reduction_ops_common_gpu.h
…
reduction_ops_common.cc
…
reduction_ops_common.h
…
reduction_ops_euclidean.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reduction_ops_gpu_bool.cu.cc
…
reduction_ops_gpu_complex64.cu.cc
…
reduction_ops_gpu_complex128.cu.cc
…
reduction_ops_gpu_double.cu.cc
…
reduction_ops_gpu_float.cu.cc
…
reduction_ops_gpu_int.cu.cc
…
reduction_ops_half_mean_sum.cu.cc
…
reduction_ops_half_prod_max_min.cu.cc
…
reduction_ops_max.cc
…
reduction_ops_mean.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reduction_ops_min.cc
…
reduction_ops_prod.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reduction_ops_sum.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reduction_ops_test.cc
…
reduction_ops.h
Add a bfloat16 sum reducer that uses float32 accumulators. Fix existing tests.
2020-06-01 10:21:39 -07:00
redux_functor.h
Fix fp16 FusedBatchNorm CPU crash if batch dimension is 1.
2020-06-19 10:38:53 -07:00
reference_gemm.h
…
regex_full_match_op.cc
…
regex_replace_op_test.cc
…
regex_replace_op.cc
…
relu_op_functor.h
working around a known gcc/hcc interface kernel args passing bug
2020-02-14 21:37:30 +00:00
relu_op_gpu.cu.cc
Merge pull request #40552 from kaixih:relu_grad_vect_pr
2020-06-18 16:10:59 -07:00
relu_op.cc
Merge pull request #35965 from ROCmSoftwarePlatform:google-upstream-relu-half
2020-04-07 00:54:59 -07:00
relu_op.h
PR #36526: Removed NDIMS template arg from BinaryElementWiseOp::Operate
2020-02-26 20:38:14 -08:00
remote_fused_graph_execute_op_test_utils.cc
…
remote_fused_graph_execute_op_test_utils.h
…
remote_fused_graph_execute_op_test.cc
…
remote_fused_graph_execute_op.cc
…
remote_fused_graph_execute_utils_test.cc
Remove unused variables.
2019-12-19 21:23:42 -08:00
remote_fused_graph_execute_utils.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
remote_fused_graph_execute_utils.h
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
remote_fused_graph_rewriter_transform_test.cc
…
remote_fused_graph_rewriter_transform.cc
…
requantization_range_op_test.cc
…
requantization_range_op.cc
…
requantize_op_test.cc
…
requantize.cc
…
reshape_op.cc
…
reshape_op.h
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
reshape_util.cc
Rename tensorflow::Reshape() to tensorflow::ReshapeSparseTensor().
2020-04-08 10:47:41 -07:00
reshape_util.h
Rename tensorflow::Reshape() to tensorflow::ReshapeSparseTensor().
2020-04-08 10:47:41 -07:00
resize_area_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
resize_area_op.cc
Fix 64-bit integer portability problems in TensorFlow kernels.
2020-01-21 19:14:20 -08:00
resize_bicubic_op_test.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
resize_bicubic_op.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
resize_bilinear_op_gpu.cu.cc
…
resize_bilinear_op_test.cc
Optimize bilinear resizing using vectorization.
2020-06-07 23:42:57 -07:00
resize_bilinear_op.cc
Optimize bilinear resizing using vectorization.
2020-06-07 23:42:57 -07:00
resize_bilinear_op.h
…
resize_nearest_neighbor_op_gpu.cu.cc
…
resize_nearest_neighbor_op_test.cc
…
resize_nearest_neighbor_op.cc
Merge pull request #34212 from Intel-tensorflow:Leslie_fang/resize_nearest_neighbor_op
2019-11-22 16:17:40 -08:00
resize_nearest_neighbor_op.h
add the cost estimator
2019-11-20 21:46:23 +08:00
resize_op_benchmark_test.cc
remove ParallelExecute and use d.pallelFor instead of Shard
2019-11-20 20:30:38 +08:00
resource_variable_ops.cc
[Cleanup] Remove allowed_devices of ResourceHandle since it's no longer used.
2020-06-22 13:18:43 -07:00
resource_variable_ops.h
[Cleanup] Remove allowed_devices of ResourceHandle since it's no longer used.
2020-06-22 13:18:43 -07:00
restore_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
restore_op.cc
…
restore_v2_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
reverse_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reverse_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
reverse_op.cc
Fix tensorflow::errors:* calls, which use StrCat instead of StrFormat
2020-06-11 00:59:42 -07:00
reverse_op.h
…
reverse_sequence_op_gpu.cu.cc
…
reverse_sequence_op.cc
Fix argument check tests to work in eager mode
2020-05-18 15:53:45 -07:00
reverse_sequence_op.h
…
rocm_solvers.cc
Add Broadcasted Matrix Triangular Solve.
2020-01-27 20:43:43 -08:00
rocm_solvers.h
Remove unused unique_tensor_references library.
2020-03-12 16:35:29 -07:00
rocm_sparse.cc
Adding ROCm support for the GpuSparse API (TF wrapper for cuSPARSE/hipSPARSE)
2019-12-13 15:24:42 +00:00
roll_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
roll_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
roll_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
roll_op.h
…
rpc_op.cc
Remove unused header include gtl/stl_util
2019-10-31 15:13:12 -07:00
sample_distorted_bounding_box_op.cc
…
sampling_kernels_test.cc
…
sampling_kernels.cc
…
sampling_kernels.h
…
save_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
save_op.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
save_restore_tensor.cc
…
save_restore_tensor.h
…
save_restore_v2_ops.cc
…
save_v2_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
scale_and_translate_op_test.cc
Adding BUILD file under tensorflow/core/lib/png.
2019-11-05 15:38:19 -08:00
scale_and_translate_op.cc
…
scale_and_translate_op.h
…
scan_ops_gpu_double.cu.cc
…
scan_ops_gpu_float.cu.cc
…
scan_ops_gpu_half.cu.cc
…
scan_ops_gpu_int.cu.cc
…
scan_ops_gpu.h
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
scan_ops_test.cc
…
scan_ops.cc
…
scan_ops.h
Modified cumulative_logsumexp to improve handling of -inf elements.
2020-04-20 14:26:19 -07:00
scatter_functor_gpu.cu.cc
…
scatter_functor_gpu.cu.h
…
scatter_functor.cc
…
scatter_functor.h
Fixes/updates for tensorflow::tstring migration.
2019-11-26 23:35:38 -08:00
scatter_nd_op_cpu_impl_0.cc
…
scatter_nd_op_cpu_impl_1.cc
…
scatter_nd_op_cpu_impl_2.cc
…
scatter_nd_op_cpu_impl_3.cc
…
scatter_nd_op_cpu_impl_4.cc
…
scatter_nd_op_cpu_impl_5.cc
…
scatter_nd_op_cpu_impl_6.cc
…
scatter_nd_op_cpu_impl_7.cc
…
scatter_nd_op_cpu_impl.h
Add Registration for non-stateful scatter_nd_min and scatter_nd_max.
2020-06-03 11:05:54 -07:00
scatter_nd_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
scatter_nd_op_test.cc
…
scatter_nd_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
scatter_nd_op.h
Add Registration for non-stateful scatter_nd_min and scatter_nd_max.
2020-06-03 11:05:54 -07:00
scatter_op_gpu.cu.cc
…
scatter_op_test.cc
Merge pull request #33115 from Intel-tensorflow:sriniva2/scatter_opt
2019-10-11 14:04:57 -07:00
scatter_op.cc
[core] scatter_op: Remove duplicate check for IsScalar()
2020-01-29 20:45:15 -05:00
scoped_allocator_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
scoped_allocator_ops.cc
…
sdca_internal.cc
…
sdca_internal.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
sdca_ops_test.cc
…
sdca_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
searchsorted_op_gpu.cu.cc
Fixed a crash when calling tf.searchsorted with zero-sized values.
2020-03-06 16:30:32 -08:00
searchsorted_op.cc
remove out of date TODO
2019-11-26 10:06:23 -08:00
searchsorted_op.h
…
segment_reduction_ops_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
segment_reduction_ops_impl_1.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
segment_reduction_ops_impl_2.cc
…
segment_reduction_ops_impl_3.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
segment_reduction_ops_impl_4.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
segment_reduction_ops_impl_5.cc
Add bfloat16 support for SparseSegmentMean*/SparseSegmentSqrtN*
2020-05-27 14:26:19 -07:00
segment_reduction_ops_impl.h
Add bfloat16 support for SparseSegmentMean*/SparseSegmentSqrtN*
2020-05-27 14:26:19 -07:00
segment_reduction_ops_test.cc
…
segment_reduction_ops.h
…
self_adjoint_eig_op.cc
…
self_adjoint_eig_v2_op_complex64.cc
…
self_adjoint_eig_v2_op_complex128.cc
…
self_adjoint_eig_v2_op_double.cc
…
self_adjoint_eig_v2_op_float.cc
…
self_adjoint_eig_v2_op_gpu.cc
…
self_adjoint_eig_v2_op_impl.h
…
sendrecv_ops_test.cc
…
sendrecv_ops.cc
Server-side cancellation support for distributed function execution.
2020-05-29 13:11:45 -07:00
sendrecv_ops.h
give sendop/recvop some temporary attribute for tracing. so we can annotate the memcpy device events better.
2020-02-20 10:32:29 -08:00
sequence_ops_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
sequence_ops.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
serialize_sparse_op.cc
[tf.data] Further optimizations for SerializeManySparseOp.
2020-01-23 09:00:57 -08:00
session_ops.cc
…
set_kernels.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
shape_ops_test.cc
[ExpandDimsOp] Micro-optimizations for tf.expand_dims().
2020-04-27 09:23:22 -07:00
shape_ops.cc
…
shape_ops.h
[ExpandDimsOp] Micro-optimizations for tf.expand_dims().
2020-04-27 09:23:22 -07:00
slice_op_cpu_impl_1.cc
…
slice_op_cpu_impl_2.cc
…
slice_op_cpu_impl_3.cc
…
slice_op_cpu_impl_4.cc
…
slice_op_cpu_impl_5.cc
…
slice_op_cpu_impl_6.cc
…
slice_op_cpu_impl_7.cc
…
slice_op_cpu_impl_8.cc
…
slice_op_cpu_impl.h
…
slice_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
slice_op_test.cc
[Slice] Reduce variance in benchmark.
2020-03-03 10:45:10 -08:00
slice_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
slice_op.h
…
smooth-hinge-loss.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
snapshot_op_gpu.cu.cc
…
snapshot_op.cc
…
snapshot_op.h
…
sobol_op.cc
Add tf.math.sobol_sample for generating Sobol sequences (CPU implementation only).
2019-12-19 03:49:17 -08:00
softmax_op_functor.h
…
softmax_op_gpu.cu.cc
Add missing softmax kernels
2020-06-03 18:59:16 -07:00
softmax_op.cc
Add missing softmax kernels
2020-06-03 18:59:16 -07:00
softplus_op_gpu.cu.cc
…
softplus_op.cc
PR #36526: Removed NDIMS template arg from BinaryElementWiseOp::Operate
2020-02-26 20:38:14 -08:00
softplus_op.h
*** Reason for rollback ***
2020-05-01 17:01:28 -07:00
softsign_op_gpu.cu.cc
…
softsign_op.cc
PR #36526: Removed NDIMS template arg from BinaryElementWiseOp::Operate
2020-02-26 20:38:14 -08:00
softsign_op.h
…
spacetobatch_benchmark_test.cc
…
spacetobatch_functor_gpu.cu.cc
…
spacetobatch_functor.cc
…
spacetobatch_functor.h
Merge pull request #37507 from t-ubukata:typo
2020-03-20 11:48:19 -07:00
spacetobatch_op.cc
…
spacetodepth_op_gpu.cu.cc
…
spacetodepth_op.cc
…
spacetodepth_op.h
…
sparse_add_grad_op.cc
…
sparse_add_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
sparse_add_op.cc
Fixing nondeterministic nullptr access.
2020-06-05 22:45:43 -07:00
sparse_concat_op.cc
…
sparse_conditional_accumulator_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
sparse_conditional_accumulator.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
sparse_cross_op.cc
Add SparseCrossV2 which supports strong_hash with salt, and fingerprint doens't
2020-05-18 17:28:14 -07:00
sparse_dense_binary_op_shared_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
sparse_dense_binary_op_shared.cc
…
sparse_fill_empty_rows_op.cc
[OpKernel] Implement OpKernelContext::output_required(int).
2020-02-25 21:20:37 -08:00
sparse_matmul_op_test.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
sparse_matmul_op.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
sparse_matmul_op.h
…
sparse_reduce_op.cc
Fix 64-bit integer portability problems in TensorFlow kernels.
2020-01-21 19:14:20 -08:00
sparse_reduce_sum_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
sparse_reorder_op.cc
…
sparse_reshape_op.cc
Rename tensorflow::Reshape() to tensorflow::ReshapeSparseTensor().
2020-04-08 10:47:41 -07:00
sparse_slice_grad_op.cc
…
sparse_slice_op.cc
…
sparse_softmax_op.cc
…
sparse_sparse_binary_op_shared.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
sparse_split_op.cc
…
sparse_tensor_dense_add_op.cc
…
sparse_tensor_dense_add_op.h
…
sparse_tensor_dense_matmul_op_gpu.cu.cc
…
sparse_tensor_dense_matmul_op_test.cc
…
sparse_tensor_dense_matmul_op.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
sparse_tensor_dense_matmul_op.h
…
sparse_tensors_map_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
sparse_to_dense_op_test.cc
[SparseToDense] Fix benchmark reporting for the SparseToDense op.
2020-02-19 09:30:01 -08:00
sparse_to_dense_op.cc
[SparseTensor] Optimize the tf.sparse.to_dense() implementation.
2020-02-19 15:52:10 -08:00
sparse_utils_test.cc
Making GetStartIndicesOfEachDenseRow produce leading zeros in case if several beginning empty rows are empty.
2020-04-22 01:30:52 -07:00
sparse_utils.cc
Making GetStartIndicesOfEachDenseRow produce leading zeros in case if several beginning empty rows are empty.
2020-04-22 01:30:52 -07:00
sparse_utils.h
Making GetStartIndicesOfEachDenseRow produce leading zeros in case if several beginning empty rows are empty.
2020-04-22 01:30:52 -07:00
sparse_xent_op_gpu.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
sparse_xent_op_test.cc
…
sparse_xent_op.cc
…
sparse_xent_op.h
…
spectrogram_convert_test_data.cc
…
spectrogram_op_test.cc
fix support of multichaannel data in audio_spectrogram
2020-03-16 10:26:05 -07:00
spectrogram_op.cc
fix support of multichaannel data in audio_spectrogram
2020-03-16 10:26:05 -07:00
spectrogram_test_utils.cc
…
spectrogram_test_utils.h
…
spectrogram_test.cc
Make use of GetDataDependencyFilepath and JoinPath to build paths which will
2020-02-21 09:57:35 -08:00
spectrogram.cc
fix support of multichaannel data in audio_spectrogram
2020-03-16 10:26:05 -07:00
spectrogram.h
fix support of multichaannel data in audio_spectrogram
2020-03-16 10:26:05 -07:00
split_lib_cpu.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
split_lib_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
split_lib_gpu.h
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
split_lib.h
…
split_op_test.cc
…
split_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
split_v_op_test.cc
…
split_v_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
squared-loss.h
…
stack_ops.cc
…
stack.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
stack.h
…
stage_op.cc
…
stateful_random_ops_cpu_gpu.h
Fix for build without ROCm or CUDA
2020-01-17 03:18:04 -08:00
stateful_random_ops_gpu.cu.cc
Fix for spurious whitespaces
2020-01-03 19:08:01 -08:00
stateful_random_ops.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
stateful_random_ops.h
…
stateless_random_ops.cc
Change log to log1p in Random Gamma sampler.
2020-04-22 11:07:43 -07:00
stateless_random_ops.h
…
strided_slice_op_define_grad.cc
…
strided_slice_op_gpu_bool.cu.cc
…
strided_slice_op_gpu_complex.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
strided_slice_op_gpu_impl.h
…
strided_slice_op_gpu_int.cu.cc
Register uint8, int16, and uint16 for strided slice on GPU, such that it supports the same types as on CPU.
2020-05-27 15:35:39 -07:00
strided_slice_op_gpu_number_types.cu.cc
…
strided_slice_op_impl.h
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
strided_slice_op_inst_0.cc
…
strided_slice_op_inst_1.cc
…
strided_slice_op_inst_2.cc
…
strided_slice_op_inst_3.cc
…
strided_slice_op_inst_4.cc
…
strided_slice_op_inst_5.cc
…
strided_slice_op_inst_6.cc
…
strided_slice_op_inst_7.cc
…
strided_slice_op_inst_8.cc
…
strided_slice_op_test.cc
…
strided_slice_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
strided_slice_op.h
Fixes/updates for tensorflow::tstring migration.
2019-11-26 23:35:38 -08:00
string_format_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
string_format_op.cc
…
string_join_op.cc
…
string_length_op.cc
…
string_lower_op.cc
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
string_ngrams_op_test.cc
…
string_ngrams_op.cc
Fix bug in tf.strings.ngrams when data is a ragged tensor with no values. (The row splits vector was not getting zeroed out.)
2020-01-23 09:23:39 -08:00
string_split_op_test.cc
…
string_split_op.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
string_strip_op.cc
…
string_to_hash_bucket_op.cc
…
string_to_hash_bucket_op.h
…
string_to_number_op.cc
…
string_upper_op.cc
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
string_util.cc
…
string_util.h
…
substr_op_test.cc
…
substr_op.cc
…
summary_audio_op_test.cc
…
summary_audio_op.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
summary_image_op_test.cc
…
summary_image_op.cc
Clean up OpKernel 1.
2020-01-29 15:49:57 -08:00
summary_interface.h
…
summary_kernels.cc
…
summary_op_test.cc
…
summary_op.cc
Remove unnecessary string copy.
2020-06-18 16:35:39 -07:00
summary_tensor_op_test.cc
…
summary_tensor_op.cc
Use ParseFromTString/ParseFromArray for tstring.
2020-02-04 14:18:15 -08:00
svd_op_complex64.cc
…
svd_op_complex128.cc
…
svd_op_double.cc
…
svd_op_float.cc
…
svd_op_gpu.cu.cc
Correctly handle empty matrices in tf.linalg.svd.
2020-05-13 15:05:03 -07:00
svd_op_impl.h
Correctly handle empty matrices in tf.linalg.svd.
2020-05-13 15:05:03 -07:00
tensor_array_ops.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
tensor_array.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
tensor_array.h
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
tensor_cord_test.cc
Fixes/updates for tensorflow::tstring migration.
2019-11-26 23:35:38 -08:00
tensor_cord.cc
TensorCord Implementation.
2019-10-25 16:02:26 -07:00
tensor_cord.h
Avoid designated initializers, as MSVC complains about them.
2020-01-16 09:58:55 -08:00
tensor_flag_utils_test.cc
Fix 64-bit integer portability problems in TensorFlow kernels.
2020-01-21 19:14:20 -08:00
tensor_flag_utils.cc
…
tensor_flag_utils.h
…
tensor_list.cc
Define TensorList class in a separate library
2020-01-24 17:37:07 -08:00
tensor_list.h
Define TensorList class in a separate library
2020-01-24 17:37:07 -08:00
text_line_reader_op.cc
…
tf_record_reader_op.cc
…
tile_functor_cpu_bfloat16.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_bool.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_complex64.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_complex128.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_double.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_float.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_half.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_int8.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_int16.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_int32.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_int64.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_tstring.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_uint8.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_cpu_uint32.cc
Add uint32/uint64 support for tf.tile
2020-05-11 17:46:29 +00:00
tile_functor_cpu_uint64.cc
Add uint32/uint64 support for tf.tile
2020-05-11 17:46:29 +00:00
tile_functor_cpu_variant.cc
Allow tile op to work on variant dtype.
2020-05-19 16:55:56 -07:00
tile_functor_cpu.h
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor_gpu_bool.cu.cc
…
tile_functor_gpu_complex64.cu.cc
…
tile_functor_gpu_complex128.cu.cc
…
tile_functor_gpu_double.cu.cc
…
tile_functor_gpu_float.cu.cc
…
tile_functor_gpu_half.cu.cc
…
tile_functor_gpu_int16.cu.cc
…
tile_functor_gpu_int32.cu.cc
…
tile_functor_gpu_int64.cu.cc
…
tile_functor_gpu.h
…
tile_functor_sycl.cc
[Build time] Split "tile_functor_cpu.cc" into a shared header and type-specific .cc files.
2020-04-02 17:09:49 -07:00
tile_functor.h
…
tile_ops_cpu_impl_1.cc
…
tile_ops_cpu_impl_2.cc
…
tile_ops_cpu_impl_3.cc
…
tile_ops_cpu_impl_4.cc
…
tile_ops_cpu_impl_5.cc
…
tile_ops_cpu_impl_6.cc
…
tile_ops_cpu_impl_7.cc
…
tile_ops_cpu_impl.h
…
tile_ops_gpu_impl_1.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_2.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_3.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_4.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_5.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_6.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_7.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl_8.cu.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_gpu_impl.h
minor spelling tweaks
2020-02-27 15:42:16 +09:00
tile_ops_impl.h
…
tile_ops.cc
Merge pull request #39424 from yongtang:39405-tile-uint32-uint64
2020-05-26 16:43:07 -07:00
topk_op_gpu_double.cu.cc
…
topk_op_gpu_float.cu.cc
…
topk_op_gpu_half.cu.cc
…
topk_op_gpu_int8.cu.cc
…
topk_op_gpu_int16.cu.cc
…
topk_op_gpu_int32.cu.cc
…
topk_op_gpu_int64.cu.cc
…
topk_op_gpu_uint8.cu.cc
…
topk_op_gpu_uint16.cu.cc
Adding these ops to enable the use of uint32 for indices in sparse matrices, since int32 has issues on GPU. Adding support for uint32 for:
2020-03-03 14:24:41 -08:00
topk_op_gpu.h
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
topk_op.cc
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-20 15:17:39 -07:00
topk_op.h
…
training_op_helpers.cc
…
training_op_helpers.h
…
training_ops_gpu.cu.cc
Merge pull request #35924 from ROCmSoftwarePlatform:google_upstream_training_ops
2020-04-14 12:00:17 -07:00
training_ops_test.cc
…
training_ops.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
training_ops.h
Added CPU and GPU implementations of a modified version of FTRL, triggered using an attribute, that multiplies the "linear" accumulator by the learning rate. This enables a learning rate of zero to work correctly.
2020-04-07 12:11:41 -07:00
transpose_functor_cpu.cc
…
transpose_functor_gpu.cu.cc
…
transpose_functor.h
Handle DT_UINT32 & DT_UINT64 in transpose
2020-06-20 14:15:24 -07:00
transpose_op.cc
Allow tf.math.invert_permutation to broadcast.
2020-01-10 23:37:25 -08:00
transpose_op.h
…
transpose_util_test.cc
…
tridiagonal_matmul_op_gpu.cu.cc
…
tridiagonal_matmul_op.cc
…
tridiagonal_solve_op_gpu.cu.cc
Remove usages of cusparse gtsv*
2020-03-23 11:03:31 -07:00
tridiagonal_solve_op.cc
…
typed_conditional_accumulator_base.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
typed_queue.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
unary_ops_composition_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
unary_ops_composition.cc
Change some const declarations to constexpr
2020-04-20 14:59:17 -07:00
unicode_ops.cc
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
unicode_script_op.cc
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
unique_op_test.cc
Rolling forward "[tf.unique] Optimize the hash table implementation in UniqueOp::Compute()."
2020-04-22 09:27:19 -07:00
unique_op.cc
Rolling forward "[tf.unique] Optimize the hash table implementation in UniqueOp::Compute()."
2020-04-22 09:27:19 -07:00
unpack_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
unravel_index_op.cc
Merge pull request #40214 from yongtang:40204-tf.unravel_index
2020-06-09 13:32:49 -07:00
unsorted_segment_join_op.cc
minor spelling tweaks
2020-02-27 15:42:16 +09:00
variable_ops_test.cc
…
variable_ops.cc
Rollforward: Add DT_BOOL support to GPU variable ops
2020-06-19 18:06:12 -07:00
variable_ops.h
Optimize (legacy, reference-typed) VariableOp::Compute().
2020-03-27 14:48:18 -07:00
where_op_gpu_impl_1.cu.cc
…
where_op_gpu_impl_2.cu.cc
…
where_op_gpu_impl_3.cu.cc
…
where_op_gpu_impl_4.cu.cc
…
where_op_gpu_impl_5.cu.cc
…
where_op_gpu_impl_6.cu.cc
…
where_op_gpu_impl_7.cu.cc
…
where_op_gpu_impl_8.cu.cc
…
where_op_gpu.cu.h
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
where_op.cc
[WhereCPUOp] Optimize the tensor-wrapping of the num_true scalar.
2020-02-24 11:00:44 -08:00
where_op.h
[WhereCPUOp] Optimize the tensor-wrapping of the num_true scalar.
2020-02-24 11:00:44 -08:00
whole_file_read_ops.cc
…
winograd_transform.h
…
word2vec_kernels.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
xent_op_gpu.cu.cc
…
xent_op_test.cc
…
xent_op.cc
…
xent_op.h
…
xsmm_conv2d_test.cc
Merge pull request #24706 from hfp:upstream
2019-11-19 12:31:39 -08:00
xsmm_conv2d.cc
Merge pull request #24706 from hfp:upstream
2019-11-19 12:31:39 -08:00
xsmm_conv2d.h
…