experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/core/kernels
History
Bernardo Henz 6dc2a1becf Adding dependencies for layer normalization
2020-08-19 13:05:58 -03: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
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
adjust_contrast_op_test.cc
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
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
Merge pull request #29169 from ROCmSoftwarePlatform:google_upstream_adjust_hue_saturation_op
2019-06-04 15:49:18 -07:00
adjust_hue_op.cc
…
adjust_hue_op.h
…
adjust_saturation_op_gpu.cu.cc
…
adjust_saturation_op.cc
…
adjust_saturation_op.h
Add ROCm support for adjust_hue and adjust_saturation op
2019-05-30 15:47:43 +00:00
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
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
Add broadcasting support to tf.matmul.
2019-03-29 13:52:19 -07:00
batch_norm_op_gpu.cu.cc
Adding ROCm support for the batch_norm op
2019-05-29 15:41:21 +00:00
batch_norm_op_test.cc
…
batch_norm_op.cc
…
batch_norm_op.h
…
batch_util.h
…
batchtospace_op.cc
Adding ROCm support for batch_to_space op
2019-07-03 14:14:40 +00:00
bcast_ops.cc
Use absl instead of deprecated str_util
2019-05-21 08:10:10 -07:00
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
Add more benchmarks to BroadcastTo test
2019-10-03 12:24:53 -07:00
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
added __restrict__ to all possible __global__ kernels
2019-08-01 15:25:07 +09:00
bucketize_op.cc
Enable GPU tests for bucketize op and fix a hard crash on empty input tensors.
2019-09-09 20:49:25 -07:00
bucketize_op.h
…
BUILD
Adding dependencies for layer normalization
2020-08-19 13:05:58 -03: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
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_float.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_half.cc
…
cast_op_impl_int8.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_int16.cc
…
cast_op_impl_int32.cc
…
cast_op_impl_int64.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_uint8.cc
…
cast_op_impl_uint16.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_uint32.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
cast_op_impl_uint64.cc
…
cast_op_impl.h
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
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
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
Move clustering ops to core.
2019-01-10 17:39:50 -08:00
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
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
colorspace_op.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
colorspace_op.h
…
compare_and_bitpack_op_gpu.cu.cc
Merge pull request #31223 from ThisIsIsaac:restrict
2019-08-28 08:28:47 -07:00
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
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
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
Adds a lowering from Case to _SwitchN+Merge.
2019-05-30 18:28:08 -07:00
conv_2d_gpu_double.cu.cc
…
conv_2d_gpu_float.cu.cc
…
conv_2d_gpu_half.cu.cc
clean-up unused functors
2019-09-06 23:36:33 -07:00
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
Use Eigen::CuboidConvolutionBackwardKernel in Conv3DBackpropFilter.
2018-09-12 14:21:00 -07:00
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
Merge pull request #24011 from Intel-tensorflow:feature/daoxin/cropAndResizeParallize
2019-03-20 23:18:54 -07:00
cross_op_gpu.cu.cc
Adding ROCm support to cross_op
2019-05-30 14:29:45 +00:00
cross_op_test.cc
…
cross_op.cc
Adding ROCm support to cross_op
2019-05-30 14:29:45 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for remaining cwise ops and tests
2019-06-03 21:09:03 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
cwise_op_gpu_digamma.cu.cc
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Let SelectV2 handle dimensions up to 8. This matches recent changes to Where.
2019-06-26 15:34:43 -07:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Update cwise_op_gpu_xlogy.cu.cc
2019-08-15 10:49:49 -05:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
Remove ErfinvGrad and NdtriGrad since gradients are already registerd in math_grad.cc
2019-10-03 10:47:01 -07:00
cwise_op_ndtri_float.cu.cc
Remove ErfinvGrad and NdtriGrad since gradients are already registerd in math_grad.cc
2019-10-03 10:47:01 -07:00
cwise_op_ndtri.cc
Remove ErfinvGrad and NdtriGrad since gradients are already registerd in math_grad.cc
2019-10-03 10:47:01 -07:00
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
Add ROCm support for most cwise ops
2019-05-31 19:13:19 +00:00
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
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
Updated ReaderInterface and subclasses to use tstring.
2019-08-27 20:01:38 -07:00
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
Remove THIRD_PARTY_ from #include guards
2018-01-24 14:31:28 -08:00
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
Merge pull request #31223 from ThisIsIsaac:restrict
2019-08-28 08:28:47 -07:00
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
Add benchmark for the SpatialConvolution specialized gemm_pack_rhs.
2018-11-02 12:14:43 -07:00
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
Update kernels and related libs to use tstring.
2019-08-27 18:22:08 -07:00
encode_proto_op.cc
Update kernels and related libs to use tstring.
2019-08-27 18:22:08 -07:00
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
Remove THIRD_PARTY_ from #include guards
2018-01-24 14:31:28 -08:00
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
Add complex128 support to RFFT, RFFT2D, RFFT3D, IRFFT, IRFFT2D, and IRFFT3D.
2019-09-12 15:07:02 -07:00
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
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
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
Updated ReaderInterface and subclasses to use tstring.
2019-08-27 20:01:38 -07:00
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
Split gather_nd CPU functors into multiple files for faster compile times.
2016-08-02 16:17:47 -07:00
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
Move "third_party/tensorflow/core/util/util.h" import from gather_nd_op.cc to
2019-04-10 03:09:56 -07:00
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
Migrated a subset of kernels to use tstring.
2019-08-14 09:29:06 -07:00
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
Adds a GuaranteeConstOp.
2017-12-01 16:29:19 -08:00
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
Segmentation Fault 11 thrown by tensorflow::HistogramFixedWidthFunctor::Compute in histogram_op.cc
2019-07-11 11:47:41 -07:00
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
Merge pull request #29346 from guoyinchen:aar-build-error-clang3.8
2019-06-03 23:45:30 -07:00
immutable_constant_op_test.cc
…
immutable_constant_op.cc
…
immutable_constant_op.h
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
[Intel MKL] Conditionally enabling bfloat16
2019-09-30 13:37:34 -07:00
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
Add support for AssertOp in XLA
2019-05-20 13:30:08 -07:00
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
Add TensorFlow op to compute the LU decomposition of a matrix.
2018-12-05 16:50:20 -08:00
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
Adding ROCm support for the "matmul" op
2019-05-13 18:33:22 +00:00
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
Add a mutex in cuda_solvers GetrfImpl. (#14492)
2017-12-28 17:38:27 -08:00
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
Add support for complex in matrix_solve_ls_op.
2017-08-18 13:29:08 -07:00
matrix_solve_ls_op_double.cc
Add support for complex in matrix_solve_ls_op.
2017-08-18 13:29:08 -07:00
matrix_solve_ls_op_float.cc
…
matrix_solve_ls_op_impl.h
Fix include guards
2018-08-16 18:29:13 -07:00
matrix_solve_op.cc
Don't crash in matrix_solve on empty matrices.
2019-05-04 11:34:45 -07:00
matrix_square_root_op.cc
Merge pull request #20611 from jonathanwyatt16:matrix_square_root
2018-10-22 11:09:50 -07:00
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
Remove THIRD_PARTY_ from #include guards
2018-01-24 14:31:28 -08:00
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
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
mfcc_op_test.cc
Remove using directives. Test appertaining to ops have been moved into namespace tensorflow::ops; all other tests now use explicit using-declarations.
2018-01-02 10:49:13 -08:00
mfcc_op.cc
Add Mfcc op to TensorFlow for speech feature generation
2017-04-21 11:24:49 -07:00
mfcc_test.cc
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
mfcc.cc
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
mfcc.h
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
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
Shard CPU implementation for mirror_pad to improve compile times
2016-08-16 14:17:23 -07:00
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
[INTEL_MKL] Changed 4 unit tests because of the node label changes recently introduced by graph rewrite changes
2019-06-28 18:21:34 -07:00
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
Update tensorflow/core/kernels/mkl_requantize_ops_test.cc
2019-02-08 14:04:53 -08:00
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
Merge pull request #30893 from samikama:GPUNMSFixes
2019-09-09 18:13:12 -07:00
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
Migrated a subset of kernels to use tstring.
2019-08-14 09:29:06 -07:00
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
Merge pull request #31223 from ThisIsIsaac:restrict
2019-08-28 08:28:47 -07:00
population_count_op.cc
Adding ROCM support for the population_count op
2019-06-10 15:39:12 +00:00
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
Fix include guards
2018-08-16 18:29:13 -07:00
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
Remove unused protobuf header inclusions
2017-06-06 03:15:27 -07:00
quantize_down_and_shrink_range.cc
Arm32/64 kernel optimizations:
2016-10-27 15:32:39 -07:00
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
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
quantized_bias_add_op_test.cc
…
quantized_bias_add_op.cc
…
quantized_concat_op_test.cc
Sort header inclusions; define EIGEN_USE_THREADS where headers depend on it.
2017-06-07 07:41:38 -07:00
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
Updated the majority of string tensor accessors to use tstring type.
2019-08-07 11:46:35 -07:00
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
Update RaggedTensors to support int32 row_splits.
2019-04-24 19:39:31 -07:00
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
Add random binomial sampling op.
2019-04-09 13:20:18 -07:00
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
Replace std::pow with MathUtil::IPow when the exponent is a non-negative integer.
2017-12-20 11:42:37 -08:00
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
Add tf.random_poisson(shape, lam) to tf core.
2017-02-07 18:26:46 -08:00
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
Updated ReaderInterface and subclasses to use tstring.
2019-08-27 20:01:38 -07:00
record_input_op.cc
Update kernels and related libs to use tstring.
2019-08-27 18:22:08 -07:00
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
Adding ROCm support for reduction ops
2019-07-03 12:44:32 +00:00
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
Merge pull request #26229 from Intel-tensorflow:requantize_support_int8
2019-03-20 12:38:39 -07:00
reshape_op.cc
Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using them.
2019-04-24 17:19:21 -07:00
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
fix C++ header guards.
2018-08-21 16:22:05 -07:00
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
Use absl instead of deprecated str_util
2019-05-21 08:10:10 -07:00
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
Adding ROCm support for scatter ops
2019-06-10 14:51:44 +00:00
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
Changes to scatter_nd ops
2016-11-08 16:12:19 -08:00
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
Branch 183429339 (#16469)
2018-01-26 13:32:16 -08:00
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
Merge pull request #31223 from ThisIsIsaac:restrict
2019-08-28 08:28:47 -07:00
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
Use the safe sparse tensor API that returns errors rather than crashing
2018-07-16 12:10:09 -07:00
sparse_reshape_op.cc
Rename tensorflow::Reshape() to tensorflow::ReshapeSparseTensor().
2020-04-08 10:47:41 -07:00
sparse_slice_grad_op.cc
Use the safe sparse tensor API that returns errors rather than crashing
2018-07-16 12:10:09 -07:00
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
Migrated a subset of kernels to use tstring.
2019-08-14 09:29:06 -07:00
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
Split topk GPU code into multiple files
2019-01-08 03:07:45 -08:00
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
Merge pull request #31223 from ThisIsIsaac:restrict
2019-08-28 08:28:47 -07:00
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
Simple benchmark of resource manager overhead.
2017-02-28 09:56:07 -08:00
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
Allow tf.compat.v2.where to support Tensors with rank up to 8.
2019-06-20 17:31:07 -07:00
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
…