experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/core/kernels
History
A. Unique TensorFlower 9cb96ea8fd [MLIR][KernelGen] Add erf kernel and missing lowering for f16 type 
PiperOrigin-RevId: 352416184
Change-Id: I6c49a6e0da11d74380aacc8612e2588308672bb9
2021-01-18 08:24:43 -08:00
..
batching_util
Introduce param 'max_input_task_size' in ASBS::Queue<TaskType>::Options and set it in batch kernel.
2021-01-13 21:15:25 -08:00
boosted_trees
Merge pull request #37455 from bmc2-stripe:bmc/202003--leaf_ids
2020-11-10 17:16:10 -08:00
data
[tf.data service] Improve cancellation for tf.data service requests.
2021-01-14 10:26:18 -08:00
fuzzing
Lower memory threshold for parse_tensor_op_fuzz
2020-11-20 12:18:27 -08:00
hexagon
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
image
Merge pull request #45934 from Intel-tensorflow:sriniva2/cnmspq
2021-01-12 10:17:00 -08:00
linalg
Fix a misplaced comment.
2020-12-29 15:05:44 -08:00
mkl
Merge pull request #45817 from Intel-tensorflow:dnn0x_clean_matmul_fused_2
2021-01-07 22:41:57 -08:00
mlir_generated
[MLIR][KernelGen] Add erf kernel and missing lowering for f16 type
2021-01-18 08:24:43 -08:00
neon
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
risc
Add risc op register.
2020-12-11 00:20:27 -08:00
rnn
Remove depencencies on unnecessary aliases in tensorflow/core
2020-10-13 10:28:00 -07:00
sparse
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
special_math
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
spectrogram_test_data
Create BUILD files and corresponding targets for tensorflow/core/kernels/spectrogram_test_data/BUILD.
2020-11-03 16:09:26 -08:00
tensor_forest
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
aggregate_ops_cpu.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
aggregate_ops_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
aggregate_ops.cc
Remove OpKernel* templating from AddNOp.
2020-12-15 18:36:46 -08:00
aggregate_ops.h
Remove OpKernel* templating from AddNOp.
2020-12-15 18:36:46 -08:00
argmax_op_gpu.cu.cc
…
argmax_op.cc
…
argmax_op.h
…
as_string_op_test.cc
AddN for variants adds in a tree structure (pairwise summation)
2020-11-02 09:11:55 -08:00
as_string_op.cc
AddN for variants adds in a tree structure (pairwise summation)
2020-11-02 09:11:55 -08:00
assign_op.h
…
avgpooling_op_gpu.cu.cc
…
avgpooling_op.cc
Rollforward of 93c22a269bec68a3b4e93a2310c9282e22c6522a. Windows and RCOM builds have been fixed.
2020-08-31 14:33:25 -07:00
avgpooling_op.h
…
barrier_ops.cc
…
base64_ops.cc
…
basic_ops_benchmark_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
batch_kernels.cc
Introduce param 'max_input_task_size' in ASBS::Queue<TaskType>::Options and set it in batch kernel.
2021-01-13 21:15:25 -08:00
batch_norm_op_gpu.cu.cc
…
batch_norm_op_test.cc
…
batch_norm_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
batch_norm_op.h
…
batch_util.h
…
batchtospace_op.cc
…
bcast_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
betainc_op_gpu.cu.cc
…
betainc_op.cc
…
betainc_op.h
…
bias_op_gpu.cu.cc
…
bias_op_gpu.h
…
bias_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
bias_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
bias_op.h
…
bincount_op_gpu.cu.cc
Merge pull request #42431 from GHGmc2:master
2020-08-26 16:56:43 -07:00
bincount_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
bincount_op.cc
Fix tf.bincount errors in eager mode
2020-07-28 21:56:14 -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
…
broadcast_to_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
broadcast_to_op.cc
…
broadcast_to_op.h
…
bucketize_op_gpu.cu.cc
…
bucketize_op.cc
…
bucketize_op.h
…
BUILD
Add Tensor Map kernels to Flex
2021-01-14 20:32:48 -08:00
candidate_sampler_ops.cc
…
cast_op_gpu.cu.cc
…
cast_op_impl_bfloat.cc
…
cast_op_impl_bool.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_complex64.cc
…
cast_op_impl_complex128.cc
…
cast_op_impl_double.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_float.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_half.cc
…
cast_op_impl_int8.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_int16.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_int32.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_int64.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_uint8.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_uint16.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_uint32.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl_uint64.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_impl.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
cast_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cast_op.h
Merge pull request #40962 from Intel-tensorflow:yang/eigen-bf16
2020-07-14 17:21:38 -07:00
check_numerics_op_gpu.cu.cc
…
check_numerics_op.cc
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
clustering_ops_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
clustering_ops.cc
…
collective_nccl_broadcaster.cc
Make NcclManager part of CollectiveExecutorMgr
2020-09-17 14:35:16 -07:00
collective_nccl_broadcaster.h
…
collective_nccl_gatherer.cc
Make NcclManager part of CollectiveExecutorMgr
2020-09-17 14:35:16 -07:00
collective_nccl_gatherer.h
…
collective_nccl_reducer.cc
Make V2 collective ComputeAsync thread safe
2020-10-21 11:17:35 -07:00
collective_nccl_reducer.h
…
collective_nccl_test.cc
[rollback]Add ordering_token input to collective to leverage AutoControlDependency to order them
2020-10-21 19:43:47 -07:00
collective_nccl.cc
Make NcclManager part of CollectiveExecutorMgr
2020-09-17 14:35:16 -07:00
collective_nccl.h
Make NcclManager part of CollectiveExecutorMgr
2020-09-17 14:35:16 -07:00
collective_ops.cc
Add CollectiveBcastSend/RecvV2 which takes input tensors rather than attributes.
2020-12-17 10:45:19 -08:00
compare_and_bitpack_op_gpu.cu.cc
…
compare_and_bitpack_op.cc
…
compare_and_bitpack_op.h
…
concat_lib_cpu.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
concat_lib_cpu.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
concat_lib_gpu_impl.cu.cc
Improve GPU dtype coverage of concat
2020-09-24 00:52:50 -07:00
concat_lib_gpu.cc
Revert PR #43178: Make fast builds work with MSVC
2020-10-05 10:30:22 -07:00
concat_lib_gpu.h
Improve GPU dtype coverage of concat
2020-09-24 00:52:50 -07:00
concat_lib.h
Improve GPU dtype coverage of concat
2020-09-24 00:52:50 -07:00
concat_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
concat_op.cc
Improve GPU dtype coverage of concat
2020-09-24 00:52:50 -07:00
conditional_accumulator_base_op.cc
…
conditional_accumulator_base_op.h
…
conditional_accumulator_base.cc
…
conditional_accumulator_base.h
…
conditional_accumulator_op.cc
Move uses of tensorflow::MakeTypeIndex() to tensorflow::TypeIndex::Make.
2020-06-23 12:50:42 -07:00
conditional_accumulator.h
…
constant_op_gpu.cu.cc
…
constant_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
constant_op.cc
Add tf.qint32 support for tf.zeros
2021-01-10 05:01:12 +00:00
constant_op.h
…
control_flow_ops_test.cc
…
control_flow_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -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
…
conv_2d_gpu_int_spatial_convolution.cu.cc
…
conv_2d_gpu_int.cu.cc
…
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
PR #44471: [ROCm] Update to use ROCm 3.9 (when building TF with --config=rocm)
2020-11-10 00:57:26 -08:00
conv_2d.h
[ROCm] Fix for ROCm CSB Breakage - 200902 - 1
2020-09-04 00:41:31 +00:00
conv_3d.h
…
conv_grad_filter_ops_benchmark_test.cc
…
conv_grad_filter_ops.cc
Merge pull request #42897 from ROCmSoftwarePlatform:google_upstream_rocm_fix_200902_1
2020-09-25 11:54:03 -07:00
conv_grad_input_ops_benchmark_test.cc
…
conv_grad_input_ops_double.cc
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_input_ops_float.cc
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_input_ops_half.cc
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_input_ops_int32.cc
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_input_ops.cc
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_input_ops.h
Splitting up conv_grad_input_ops.cc to improve build times
2020-12-04 05:47:25 -08:00
conv_grad_ops_3d.cc
Merge pull request #42897 from ROCmSoftwarePlatform:google_upstream_rocm_fix_200902_1
2020-09-25 11:54:03 -07:00
conv_grad_ops.h
…
conv_grad_shape_utils.cc
Improve error message format when reporting batch dimension mismatch in conv2d.
2020-07-31 17:26:40 -07:00
conv_grad_shape_utils.h
…
conv_ops_3d.cc
Remove changes made to support TFRT-based OpKernel classes in Conv3d kernel.
2020-12-09 16:19:36 -08:00
conv_ops_benchmark_test.cc
Merge pull request #45856 from Intel-tensorflow:mabuzain/fixing-conv-ops-benchmark-test
2020-12-21 11:51:46 -08:00
conv_ops_fused_double.cc
…
conv_ops_fused_float.cc
…
conv_ops_fused_half.cc
…
conv_ops_fused_image_transform.cc
fix typos in core directory
2020-10-29 02:52:55 +03:00
conv_ops_fused_impl.h
Merge pull request #42897 from ROCmSoftwarePlatform:google_upstream_rocm_fix_200902_1
2020-09-25 11:54:03 -07:00
conv_ops_gpu.h
Rollback PR #43237: Integrate cuBLASLt API into backend
2020-10-28 17:04:53 -07:00
conv_ops_test.cc
Use TensorFloat-32 instead of TF32 in comments and symbols.
2020-09-08 17:38:28 -07:00
conv_ops_using_gemm.cc
Break up tensorflow/core/kernels/BUILD (part 3 of N):
2020-08-11 11:04:52 -07:00
conv_ops.cc
Rollback PR #43237: Integrate cuBLASLt API into backend
2020-10-28 17:04:53 -07:00
conv_ops.h
…
count_ops_test.cc
…
count_ops.cc
Fix multiple vulnerabilities in tf.raw_ops.*CountSparseOutput.
2020-09-18 18:20:14 -07:00
count_up_to_op.cc
…
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
…
cudnn_pooling_gpu.cc
Adding 3d Pooling using latest MIOpen API
2020-06-09 15:11:45 +00:00
cudnn_pooling_gpu.h
…
cudnn_rnn_ops.cc
Merge pull request #41832 from lissyx:issue41630
2020-08-07 10:08:12 -07:00
cwise_op_abs.cc
Add ComplexAbs kernels and tests.
2021-01-18 04:38:22 -08:00
cwise_op_acos.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_acosh.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_add_1.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_add_2.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_arg.cc
…
cwise_op_asin.cc
[MLIR][KernelGen] Add asin kernels and tests
2021-01-12 09:12:40 -08:00
cwise_op_asinh.cc
[MLIR][KernelGen] Add tf.Asinh kernels and complete their lowerings
2021-01-15 05:35:49 -08:00
cwise_op_atan2.cc
[MLIR][KernelGen] Add atan2 kernels
2021-01-11 03:47:49 -08:00
cwise_op_atan.cc
[MLIR][KernelGen] Add atan kernel and tests
2021-01-12 02:51:48 -08:00
cwise_op_atanh.cc
[MLIR][KernelGen] Add tf.Atanh kernels
2021-01-18 05:20:00 -08:00
cwise_op_bitwise_and.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_bitwise_or.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_bitwise_xor.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_ceil.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_clip_gpu.cu.cc
…
cwise_op_clip.cc
Add bfloat16 support to more cwise CPU ops
2020-08-06 09:48:01 -07:00
cwise_op_clip.h
…
cwise_op_complex.cc
Register MLIR generated kernel for Complex.
2021-01-15 00:11:01 -08:00
cwise_op_conj.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_cos.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_cosh.cc
[MLIR][KernelGen] Add cosh kernels and tests
2021-01-15 04:33:57 -08:00
cwise_op_digamma.cc
…
cwise_op_div.cc
[KERNEL_GEN] Add tf.TruncateDiv aliased kernel (tf.Div).
2021-01-15 13:52:01 -08:00
cwise_op_equal_to_1.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_equal_to_2.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_erf.cc
[MLIR][KernelGen] Add erf kernel and missing lowering for f16 type
2021-01-18 08:24:43 -08:00
cwise_op_erfc.cc
…
cwise_op_exp.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_expm1.cc
Add Expm1 Kernel and tests.
2021-01-15 02:40:50 -08:00
cwise_op_floor_div.cc
[KERNEL_GEN] Add kernel generation for FloorDiv.
2020-12-21 13:20:04 -08:00
cwise_op_floor_mod.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_floor.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_gpu_abs.cu.cc
Only instantiate abs kernel templates if needed.
2020-07-30 07:30:50 -07:00
cwise_op_gpu_acos.cu.cc
…
cwise_op_gpu_acosh.cu.cc
…
cwise_op_gpu_add.cu.cc
…
cwise_op_gpu_arg.cu.cc
…
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
Revert PR #43178: Make fast builds work with MSVC
2020-10-05 10:30:22 -07:00
cwise_op_gpu_cos.cu.cc
…
cwise_op_gpu_cosh.cu.cc
…
cwise_op_gpu_digamma.cu.cc
…
cwise_op_gpu_div.cu.cc
…
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
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
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
…
cwise_op_gpu_minimum.cu.cc
…
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
KernelGen for Pow
2021-01-14 12:05:49 -08:00
cwise_op_gpu_random_grad.cu.cc
…
cwise_op_gpu_real.cu.cc
…
cwise_op_gpu_right_shift.cu.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
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
…
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
…
cwise_op_gpu_xlogy.cu.cc
…
cwise_op_gpu_zeta.cu.cc
…
cwise_op_greater_equal.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_greater.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_igammas.cc
…
cwise_op_imag.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_invert.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_isfinite.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_isinf.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_isnan.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_left_shift.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_less_equal.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_less.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_lgamma.cc
…
cwise_op_log1p.cc
[MLIR][KernelGen] Add tf.Log1p kernel and tests
2021-01-13 05:41:07 -08:00
cwise_op_log.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_logical_and.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_logical_not.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_logical_or.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_maximum.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_minimum.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_mod.cc
…
cwise_op_mul_1.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_mul_2.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_ndtri_double.cu.cc
…
cwise_op_ndtri_float.cu.cc
…
cwise_op_ndtri.cc
…
cwise_op_neg_1.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_neg_2.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_not_equal_to_1.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_not_equal_to_2.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_pow.cc
KernelGen for Pow
2021-01-14 12:05:49 -08:00
cwise_op_random_grad.cc
…
cwise_op_real.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_reciprocal.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_right_shift.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_rint.cc
…
cwise_op_round.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_rsqrt.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_select.cc
Cleanup Selectv2 broadcasting
2020-10-09 10:13:47 +01:00
cwise_op_sigmoid.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_sign.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_sin.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_sinh.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_sqrt.cc
[KERNEL_GEN] Rename enable_unranked_kernels to enable_experimental_kernels.
2020-12-17 05:14:02 -08:00
cwise_op_square.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_squared_difference.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_sub.cc
[KERNEL_GEN] Add kernel generation for Sub.
2021-01-05 03:55:32 -08:00
cwise_op_tan.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_tanh.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_xdivy.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_xlog1py.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_xlogy.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_op_zeta.cc
…
cwise_ops_common.cc
…
cwise_ops_common.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_ops_gpu_common.cu.h
Merge pull request #42689 from ROCmSoftwarePlatform:google_upstream_switch_to_rocm37
2020-09-25 23:37:01 -07:00
cwise_ops_gpu_gradients.cu.h
…
cwise_ops_gradients.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
cwise_ops_test.cc
Fix and re-enable cwise ops benchmark.
2020-10-30 16:12:39 -07:00
cwise_ops.h
Handle negative integer powers of 1 and -1.
2021-01-15 07:06:04 -08:00
data_format_ops_gpu.cu.cc
…
data_format_ops.cc
Validate that DataFormat* attributes form a permutation.
2020-12-07 11:25:52 -08:00
data_format_ops.h
Merge pull request #43226 from kaixih:reduce_ops_layout
2020-09-25 12:56:20 -07:00
debug_ops_gpu.cu.cc
…
debug_ops_test.cc
…
debug_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
debug_ops.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
decode_compressed_op.cc
…
decode_csv_op.cc
…
decode_padded_raw_op.cc
…
decode_proto_op.cc
…
decode_raw_op.cc
…
decode_wav_op_test.cc
Migrated a subset of kernels to use tstring.
2019-08-14 09:29:06 -07:00
decode_wav_op.cc
…
deep_conv2d_test.cc
…
deep_conv2d.cc
…
deep_conv2d.h
…
dense_update_functor_gpu.cu.cc
Revert PR #43178: Make fast builds work with MSVC
2020-10-05 10:30:22 -07:00
dense_update_functor.cc
…
dense_update_functor.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
dense_update_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
depthtospace_op_gpu.cu.cc
Revert PR #43178: Make fast builds work with MSVC
2020-10-05 10:30:22 -07:00
depthtospace_op.cc
rephrase wording
2020-08-31 16:42:37 +02:00
depthtospace_op.h
…
depthwise_conv_grad_op.cc
Remove CanUseCudnn and CudnnConvComputeMode functions.
2020-08-18 12:04:59 -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
Remove CanUseCudnn and CudnnConvComputeMode functions.
2020-08-18 12:04:59 -07:00
depthwise_conv_op.h
Optimize depthwise convolution. Pscatter is somewhat slow,
2020-12-08 12:21:42 -08:00
depthwise_conv_ops_test.cc
[ROCm] Re-enabling unit-tests that are now passing on ROCm platform
2020-12-07 16:41:10 +00:00
dequantize_op_test.cc
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
dequantize_op.cc
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
deserialize_sparse_string_op.cc
…
deserialize_sparse_variant_op.cc
…
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
Internal tests cleanup.
2020-10-27 21:28:44 -07:00
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
…
dilation_ops.h
…
dynamic_partition_op_gpu.cu.cc
Support ROCm in dynamic_parition fix
2020-08-24 00:26:22 -07:00
dynamic_partition_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
dynamic_partition_op.cc
Rollforward: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-22 23:09:59 -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
Change the type of slice_size from int to auto to avoid overflow.
2020-09-25 13:19:34 -07:00
edit_distance_op.cc
…
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
…
eigen_backward_cuboid_convolutions.h
…
eigen_backward_spatial_convolutions_test.cc
…
eigen_backward_spatial_convolutions.h
…
eigen_benchmark_cpu_test.cc
…
eigen_benchmark.h
…
eigen_contraction_kernel.cc
…
eigen_contraction_kernel.h
Work around an ODR violation. This template specialization is not always seen
2020-07-13 19:01:23 -07:00
eigen_convolution_helpers.h
…
eigen_cuboid_convolution.h
…
eigen_mkldnn_contraction_kernel_test.cc
…
eigen_pooling_test.cc
…
eigen_pooling.h
Update Eigen to
2020-10-02 20:28:38 -07:00
eigen_spatial_convolutions_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
eigen_spatial_convolutions-inl.h
…
eigen_spatial_convolutions.h
Default initialize offsets to avoid UB
2020-12-21 12:48:29 -08:00
eigen_volume_patch.h
…
encode_proto_op.cc
…
encode_wav_op_test.cc
…
encode_wav_op.cc
…
example_parsing_ops_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
example_parsing_ops.cc
Fix a bug where vector::reserve was being called on config.sparse instead of config.ragged.
2020-10-14 11:59:47 -07: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
…
fft_ops.cc
Rollback PR #43237: Integrate cuBLASLt API into backend
2020-10-28 17:04:53 -07:00
fifo_queue_op.cc
…
fifo_queue.cc
…
fifo_queue.h
…
fill_functor.cc
Add tf.qint32 support for tf.zeros
2021-01-10 05:01:12 +00:00
fill_functor.cu.cc
…
fill_functor.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
fingerprint_op_test.cc
Fixing a crash in tf.fingerprint([]). We now just return []
2020-09-30 08:09:56 -07:00
fingerprint_op.cc
Fixing a crash in tf.fingerprint([]). We now just return []
2020-09-30 08:09:56 -07:00
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
Delete SYCL support
2020-09-15 11:21:47 -07:00
function_ops.h
Use TraceMeOp and TraceMeEncode to hide TraceString encoding details
2020-07-24 10:59:15 -07:00
functional_ops.cc
Prefix thread annotations with ABSL_.
2020-11-16 13:06:26 -08:00
fused_batch_norm_ex_op_test.cc
…
fused_batch_norm_op_test.cc
…
fused_batch_norm_op.cc
Roll forward of https://github.com/tensorflow/tensorflow/pull/42970 with fix for 5D tensors and unsupported axis that cannot use fused batch norm.
2020-10-02 12:08:15 -07:00
fused_batch_norm_op.cu.cc
Merged commit includes the following changes:
2020-07-02 10:46:26 -07:00
fused_batch_norm_op.h
Qualify uses of std::string
2020-07-31 13:57:36 -07:00
fused_eigen_output_kernels.cc
Enable conv + (bias/bn) + leakyrelu fusion
2020-08-21 15:47:17 +08:00
fused_eigen_output_kernels.h
Merge pull request #42551 from Intel-tensorflow:yunfeimao/leaky_relu_fusion_eigen2
2020-08-28 08:16:17 -07:00
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
…
gather_functor_batched.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_functor_batched.h
…
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
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
gather_nd_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
gather_nd_op.h
Print node name in GatherNdSlice error msg.
2020-12-10 00:32:00 -08:00
gather_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
gather_op.cc
Fix data race in GatherOp::Compute().
2020-12-15 12:17:14 -08:00
gemm_functors.h
…
generate_vocab_remapping_op.cc
…
gpu_device_array_gpu.h
…
gpu_device_array.h
…
gpu_prim.h
Merge pull request #39429 from ekuznetsov139:google-upstream-topk
2020-10-18 22:12:36 -07:00
gpu_utils.cc
Rollback PR #43237: Integrate cuBLASLt API into backend
2020-10-28 17:04:53 -07:00
gpu_utils.h
Log value of TF_AUTOTUNE_THRESHOLD
2020-11-06 12:33:42 -08:00
guarantee_const_op_test.cc
…
guarantee_const_op.cc
…
hinge-loss.h
…
histogram_op_gpu.cu.cc
…
histogram_op.cc
…
histogram_op.h
…
host_constant_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
host_constant_op.h
…
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
…
identity_n_op.cc
Add required TPU_SYSTEM device kernel registrations for TPUs
2020-07-01 11:48:41 -07:00
identity_n_op.h
fix C++ header guards.
2018-08-21 16:22:05 -07:00
identity_op_test.cc
…
identity_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
identity_op.h
…
identity_reader_op.cc
…
immutable_constant_op_test.cc
Add TF_USE_FILESYSTEM_METHODS_WITH_NO_TRANSACTION_SUPPORT macro to modified classes
2020-08-04 23:20:15 -07:00
immutable_constant_op.cc
Mark MemmappedTensorAllocator as returning opaque handle.
2020-12-04 17:12:18 -08:00
immutable_constant_op.h
…
in_topk_op_gpu.cu.cc
…
in_topk_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
in_topk_op.cc
…
in_topk_op.h
…
initializable_lookup_table.cc
[-Wsign-compare] warning fixes batch 4
2020-06-24 21:58:00 +00:00
initializable_lookup_table.h
…
inplace_ops_functor_gpu.cu.cc
Add complex64 and complex128 gpu support for tensor_scatter_nd_add
2020-06-18 16:17:52 +00:00
inplace_ops_functor.h
…
inplace_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
isotonic_regression_op_test.cc
Fix kokoro build, add guard in IsotonicreRegressionOp benchmark
2020-08-18 13:30:13 -07:00
isotonic_regression_op.cc
Differentiable isotonic regression in TensorFlow.
2020-08-18 10:48:08 -07:00
kernel_platform_strings.h
…
l2loss_op_gpu.cu.cc
…
l2loss_op.cc
…
l2loss_op.h
…
linalg_ops_common.h
Break up core/kernels/BUILD (part 1 of N):
2020-08-04 17:24:28 -07:00
list_kernels.cc
…
list_kernels.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
list_kernels.h
Make element_shape_except_first_dim as a local variable instead of instance variable.
2020-11-24 10:12:32 -08:00
listdiff_op.cc
…
lmdb_reader_op.cc
…
load_and_remap_matrix_op.cc
…
logging_ops_test.cc
…
logging_ops.cc
…
logging_ops.h
…
logistic-loss.h
…
lookup_table_init_op.cc
[tf.data] Enforcing usage of the tensorflow::data namespace for tf.data C++ classes.
2020-08-24 08:10:41 -07:00
lookup_table_init_op.h
…
lookup_table_op.cc
Merge pull request #43269 from rhdong:RFC237-patch-for-recsys-sig
2020-10-27 08:06:45 -07:00
lookup_table_op.h
…
lookup_util.cc
[tf.data] Enforcing usage of the tensorflow::data namespace for tf.data C++ classes.
2020-08-24 08:10:41 -07:00
lookup_util.h
Modify InitializeTableFromDatasetOp to be async and add the table init op to the table initializers collection
2020-07-01 02:53:55 -07:00
loss_test.cc
…
loss.h
…
lrn_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
lrn_op.cc
…
map_kernels.cc
rename ListKeys to StackKeys
2020-08-13 18:57:03 +00:00
map_kernels.h
Merge pull request #42300 from kttian:list_keys
2020-08-25 10:12:32 -07:00
map_stage_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
matching_files_op.cc
Fix asan error, when there are no matching files in matching_files_op.
2020-08-03 14:21:09 -07:00
matmul_op_complex.cc
Clean up duplicated code in matmul_op/batch_matmul_op and associated tests. Having a separate redundant implementation for non-batch matmul is a maintenance burden.
2020-10-29 11:58:12 -07:00
matmul_op_fused.cc
Fix code format
2020-12-23 16:47:36 +08:00
matmul_op_impl.h
Fix regression in (non-batch) matrix multiplication:
2020-11-17 15:58:05 -08:00
matmul_op_real.cc
Clean up duplicated code in matmul_op/batch_matmul_op and associated tests. Having a separate redundant implementation for non-batch matmul is a maintenance burden.
2020-10-29 11:58:12 -07:00
matmul_op_test.cc
Enable Eigen MatMul + Bias + LeakyRelu fusion
2020-12-21 10:47:02 +08:00
matmul_op.h
…
maxpooling_op_gpu.cu.cc
fix typos in core directory
2020-10-29 02:52:55 +03:00
maxpooling_op_gpu.h
Remove CanUseCudnn and CudnnConvComputeMode functions.
2020-08-18 12:04:59 -07:00
maxpooling_op.cc
Support int32 argmax output in MaxPoolWithArgmax
2020-12-15 19:00:07 -08:00
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
…
mfcc_mel_filterbank.h
…
mfcc_op_test.cc
…
mfcc_op.cc
…
mfcc_test.cc
…
mfcc.cc
…
mfcc.h
…
multinomial_op_gpu.cu.cc
Internal change
2020-09-17 21:37:45 -07:00
multinomial_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
multinomial_op.cc
…
multinomial_op.h
…
mutex_ops.cc
…
nccl_ops.cc
…
nextafter_op_gpu.cu.cc
…
nextafter_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
nextafter_op.h
…
nn_ops_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
no_op.cc
Add required TPU_SYSTEM device kernel registrations for TPUs
2020-07-01 11:48:41 -07:00
no_op.h
…
nth_element_op.cc
Prevent integer truncation from 64 to 32 bits.
2020-09-18 17:56:30 -07:00
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
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
one_hot_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
one_hot_op.h
…
ops_testutil_test.cc
…
ops_testutil.cc
Fixing and enabling NonMaxSuppression for ROCm
2020-06-26 19:32:26 +00:00
ops_testutil.h
Move uses of tensorflow::MakeTypeIndex() to tensorflow::TypeIndex::Make.
2020-06-23 12:50:42 -07:00
ops_util_test.cc
…
ops_util.h
…
pack_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
pad_op_gpu.cu.cc
…
pad_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
pad_op.h
…
padding_fifo_queue_op.cc
…
padding_fifo_queue.cc
…
padding_fifo_queue.h
…
parameterized_truncated_normal_op_gpu.cu.cc
…
parameterized_truncated_normal_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
parameterized_truncated_normal_op.cc
Prevent integer truncation from 64 to 32 bits.
2020-09-18 17:56:30 -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
…
partitioned_function_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -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.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
pooling_ops_3d.h
…
pooling_ops_common_gpu.h
Rollforward of 93c22a269bec68a3b4e93a2310c9282e22c6522a. Windows and RCOM builds have been fixed.
2020-08-31 14:33:25 -07:00
pooling_ops_common.cc
[ROCm] Fix for ROCm CSB Breakage - 200902 - 2
2020-09-28 13:32:21 +00:00
pooling_ops_common.h
Rollforward of 93c22a269bec68a3b4e93a2310c9282e22c6522a. Windows and RCOM builds have been fixed.
2020-08-31 14:33:25 -07:00
population_count_op_gpu.cu.cc
…
population_count_op.cc
…
population_count_op.h
…
priority_queue_op.cc
…
priority_queue.cc
…
priority_queue.h
…
quantization_utils_test.cc
…
quantization_utils.cc
…
quantization_utils.h
Update Eigen to
2020-10-02 20:28:38 -07:00
quantize_and_dequantize_op_gpu.cu.cc
Create a V2 Op to stop the gradient when the input is out of range.
2020-10-12 10:39:36 -07:00
quantize_and_dequantize_op_test.cc
Merge pull request #44408 from Molkree:typos_in_core_dir
2020-11-06 15:01:58 -08:00
quantize_and_dequantize_op.cc
Prevent segfault in quantize_and_dequantize
2020-10-19 18:03:39 -07:00
quantize_and_dequantize_op.h
Create a V2 Op to stop the gradient when the input is out of range.
2020-10-12 10:39:36 -07:00
quantize_down_and_shrink_range_op_test.cc
…
quantize_down_and_shrink_range.cc
…
quantize_op_test.cc
…
quantize_op.cc
…
quantized_activation_ops_test.cc
…
quantized_activation_ops.cc
…
quantized_add_op_test.cc
…
quantized_add_op.cc
…
quantized_batch_norm_op_test.cc
…
quantized_batch_norm_op.cc
…
quantized_bias_add_op_test.cc
…
quantized_bias_add_op.cc
…
quantized_concat_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
quantized_concat_op.cc
…
quantized_conv_ops_test.cc
…
quantized_conv_ops.cc
…
quantized_instance_norm_test.cc
…
quantized_instance_norm.cc
…
quantized_matmul_op_test.cc
…
quantized_matmul_op.cc
…
quantized_mul_op_test.cc
…
quantized_mul_op.cc
…
quantized_pooling_ops_test.cc
…
quantized_pooling_ops.cc
supporting quantized pooling op
2020-12-01 09:08:37 -08:00
quantized_reshape_op_test.cc
…
quantized_reshape_op.cc
…
quantized_resize_bilinear_op_test.cc
…
quantized_resize_bilinear_op.cc
Break up tensorflow/core/kernels/BUILD (part 3 of N):
2020-08-11 11:04:52 -07:00
queue_base.cc
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
queue_base.h
…
queue_op.cc
…
queue_op.h
…
queue_ops.cc
…
ragged_cross_op.cc
…
ragged_gather_op_test.cc
…
ragged_gather_op.cc
Rollforward: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
2020-06-22 23:09:59 -07:00
ragged_range_op_test.cc
…
ragged_range_op.cc
…
ragged_tensor_from_variant_op_test.cc
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
ragged_tensor_from_variant_op.cc
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
ragged_tensor_to_sparse_kernel_test.cc
…
ragged_tensor_to_sparse_kernel.cc
…
ragged_tensor_to_tensor_op_test.cc
…
ragged_tensor_to_tensor_op.cc
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
ragged_tensor_to_variant_op_test.cc
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
ragged_tensor_to_variant_op.cc
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
ragged_tensor_variant.cc
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
ragged_tensor_variant.h
Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)
2020-10-14 10:09:39 -07:00
random_binomial_op_test.cc
Internal tests cleanup.
2020-10-29 10:15:29 -07:00
random_binomial_op.cc
Remove duplicate definition of dcount.
2020-09-30 14:59:53 -07:00
random_binomial_op.h
…
random_op_cpu.h
Internal change
2020-09-17 21:37:45 -07:00
random_op_gpu.cu.cc
…
random_op_gpu.h
Merge pull request #42288 from ROCmSoftwarePlatform:google-upstream-rocm35-0812
2020-09-25 08:48:20 -07:00
random_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
random_op.cc
Prevent int64 to int truncation in Shard API usage.
2020-09-18 17:34:10 -07:00
random_op.h
Internal change
2020-09-17 21:37:45 -07:00
random_ops_util.h
Internal change
2020-09-17 21:37:45 -07:00
random_poisson_op_test.cc
…
random_poisson_op.cc
random_poisson(rate=+inf) should return +inf, not nan.
2020-10-07 15:45:04 -07:00
random_poisson_op.h
…
random_shuffle_op.cc
…
random_shuffle_queue_op.cc
…
range_sampler_test.cc
…
range_sampler.cc
Merge pull request #41755 from tg-at-google:wsign-compare-semi-final-core2
2020-07-29 16:10:27 -07:00
range_sampler.h
…
reader_ops.cc
…
record_input_op.cc
…
record_yielder.cc
…
record_yielder.h
…
reduce_join_op.cc
…
reduction_gpu_kernels.cu.h
PR #44471: [ROCm] Update to use ROCm 3.9 (when building TF with --config=rocm)
2020-11-10 00:57:26 -08:00
reduction_ops_all.cc
…
reduction_ops_any.cc
…
reduction_ops_common_gpu.h
…
reduction_ops_common.cc
Require that axis arguments to reduction ops are unique.
2020-07-01 15:45:41 -07:00
reduction_ops_common.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
reduction_ops_euclidean.cc
Delete SYCL support
2020-09-15 11:21:47 -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
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_gpu_float.cu.cc
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_gpu_int.cu.cc
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_half_mean_sum.cu.cc
…
reduction_ops_half_prod_max_min.cu.cc
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_max.cc
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_mean.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
reduction_ops_min.cc
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
reduction_ops_prod.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
reduction_ops_sum.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
reduction_ops_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
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
Make RegexFullMatchOp and RegexReplaceOp cache RE2 objects.
2020-08-11 09:44:05 -07:00
regex_replace_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
regex_replace_op.cc
Make RegexFullMatchOp and RegexReplaceOp cache RE2 objects.
2020-08-11 09:44:05 -07:00
relu_op_functor.h
…
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 #43553 from Intel-tensorflow:fix-3898
2020-09-27 15:59:12 -07:00
relu_op.h
…
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
…
remote_fused_graph_execute_utils.cc
tensorflow/core resolutions, set 2 of 3
2020-07-26 23:56:14 +00:00
remote_fused_graph_execute_utils.h
…
remote_fused_graph_rewriter_transform_test.cc
…
remote_fused_graph_rewriter_transform.cc
…
requantization_range_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
requantization_range_op.cc
…
requantize_op_test.cc
…
requantize.cc
…
reshape_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
reshape_op.h
…
reshape_util.cc
…
reshape_util.h
…
resource_variable_ops.cc
Enable half for resource scatter
2020-07-10 20:36:44 -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
…
restore_op.cc
…
restore_v2_op_test.cc
…
reverse_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
reverse_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
reverse_op.cc
Delete SYCL support
2020-09-15 11:21:47 -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
…
roll_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
roll_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
roll_op.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
roll_op.h
…
rpc_op.cc
…
save_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
save_op.cc
…
save_restore_tensor.cc
Use double for average computation.
2020-07-06 16:24:55 -07:00
save_restore_tensor.h
…
save_restore_v2_ops.cc
Use double for average computation.
2020-07-06 16:24:55 -07:00
save_v2_op_test.cc
Add a number of missing headers being transitively pulled in.
2020-01-23 18:18:03 -08:00
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
PR #44471: [ROCm] Update to use ROCm 3.9 (when building TF with --config=rocm)
2020-11-10 00:57:26 -08:00
scan_ops_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
scan_ops.cc
Implement a cumulative log-sum-exp operation.
2019-07-22 06:37:37 -07:00
scan_ops.h
Make tf.min, tf.max always propagate NaNs. Previously these ops followed the semantics of std::min and std:max and forwarded the first argument if either was NaN.
2020-10-20 21:29:20 -07:00
scatter_functor_gpu.cu.cc
Enable half for resource scatter
2020-07-10 20:36:44 -07:00
scatter_functor_gpu.cu.h
…
scatter_functor.cc
Enable half for resource scatter
2020-07-10 20:36:44 -07:00
scatter_functor.h
Delete SYCL support
2020-09-15 11:21:47 -07: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
Delete SYCL support
2020-09-15 11:21:47 -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
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
scatter_nd_op.cc
Delete SYCL support
2020-09-15 11:21:47 -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
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
scatter_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
scoped_allocator_ops_test.cc
…
scoped_allocator_ops.cc
…
sdca_internal.cc
…
sdca_internal.h
…
sdca_ops_test.cc
…
sdca_ops.cc
…
searchsorted_op_gpu.cu.cc
…
searchsorted_op.cc
…
searchsorted_op.h
…
segment_reduction_ops_gpu.cu.cc
Merge pull request #44436 from firejq:patch-5
2020-11-21 07:27:48 -08:00
segment_reduction_ops_impl_1.cc
Merge pull request #44436 from firejq:patch-5
2020-11-21 07:27:48 -08:00
segment_reduction_ops_impl_2.cc
Merge pull request #44436 from firejq:patch-5
2020-11-21 07:27:48 -08:00
segment_reduction_ops_impl_3.cc
Rename Segment*ReductionF->*ReductionF
2020-11-15 23:56:00 +08:00
segment_reduction_ops_impl_4.cc
Rename Segment*ReductionF->*ReductionF
2020-11-15 23:56:00 +08: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
Merge pull request #44436 from firejq:patch-5
2020-11-21 07:27:48 -08:00
segment_reduction_ops_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
segment_reduction_ops.h
Rename Segment*ReductionF->*ReductionF
2020-11-15 23:56:00 +08:00
sendrecv_ops_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
sendrecv_ops.cc
Use TraceMeOp and TraceMeEncode to hide TraceString encoding details
2020-07-24 10:59:15 -07:00
sendrecv_ops.h
Use TraceMeOp and TraceMeEncode to hide TraceString encoding details
2020-07-24 10:59:15 -07:00
sequence_ops_test.cc
…
sequence_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
serialize_sparse_op.cc
[tf.data] Further optimizations for SerializeManySparseOp.
2020-01-23 09:00:57 -08:00
session_ops.cc
Prevent segfault in GetSessionHandle{,V2}.
2020-09-18 16:30:02 -07:00
set_kernels.cc
…
shape_ops_test.cc
Update benchmarks to use newer API
2020-11-24 16:59:20 -08:00
shape_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
shape_ops.h
…
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
Delete SYCL support
2020-09-15 11:21:47 -07:00
slice_op_gpu.cu.cc
Merge various kernel registrations with macros
2020-06-08 00:42:40 -07:00
slice_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
slice_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
slice_op.h
…
smooth-hinge-loss.h
…
snapshot_op_gpu.cu.cc
…
snapshot_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
snapshot_op.h
…
sobol_op.cc
…
softmax_op_functor.h
…
softmax_op_gpu.cu.cc
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
softmax_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
softplus_op_gpu.cu.cc
…
softplus_op.cc
…
softplus_op.h
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
softsign_op_gpu.cu.cc
…
softsign_op.cc
…
softsign_op.h
…
spacetobatch_benchmark_test.cc
…
spacetobatch_functor_gpu.cu.cc
…
spacetobatch_functor.cc
…
spacetobatch_functor.h
…
spacetobatch_op.cc
…
spacetodepth_op_gpu.cu.cc
Revert PR #43178: Make fast builds work with MSVC
2020-10-05 10:30:22 -07:00
spacetodepth_op.cc
rephrase wording
2020-08-31 16:42:37 +02:00
spacetodepth_op.h
…
sparse_add_grad_op.cc
…
sparse_add_op_test.cc
…
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
…
sparse_conditional_accumulator.h
…
sparse_cross_op.cc
Modify type-specific mappings to be explicit about long/long long rather than
2020-07-15 11:47:51 -07:00
sparse_dense_binary_op_shared_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
sparse_dense_binary_op_shared.cc
…
sparse_fill_empty_rows_op.cc
Merge pull request #44719 from benbarsdell:gpu-SparseFillEmptyRows-cpu-refactor
2020-12-18 18:06:07 -08:00
sparse_fill_empty_rows_op.h
Rollback of: PR #44719: Refactor SparseFillEmptyRows[Grad] CPU kernels into classes
2020-12-03 15:15:13 -08:00
sparse_matmul_op_test.cc
…
sparse_matmul_op.cc
Merge pull request #40962 from Intel-tensorflow:yang/eigen-bf16
2020-07-14 17:21:38 -07:00
sparse_matmul_op.h
…
sparse_reduce_op.cc
…
sparse_reduce_sum_op_test.cc
…
sparse_reorder_op.cc
Merge pull request #45675 from yongtang:45392-tf.sparse.reorder
2020-12-30 12:42:27 -08:00
sparse_reshape_op.cc
…
sparse_slice_grad_op.cc
…
sparse_slice_op.cc
…
sparse_softmax_op.cc
…
sparse_sparse_binary_op_shared.cc
…
sparse_split_op.cc
Add support for negative indices to sparse_split.
2020-07-07 16:37:22 -07:00
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
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
sparse_tensor_dense_matmul_op.h
…
sparse_tensors_map_ops.cc
…
sparse_to_dense_op_test.cc
Internal tests cleanup.
2020-10-29 14:08:15 -07:00
sparse_to_dense_op.cc
Make sparse_to_dense_op_py_test v2 friendly
2020-07-10 16:05:14 -07:00
sparse_utils_test.cc
…
sparse_utils.cc
…
sparse_utils.h
…
sparse_xent_op_gpu.cu.cc
…
sparse_xent_op_test.cc
Rolling back PR #44894 because it caused type check errors (got bfloat16 when expecting float).
2020-12-07 17:30:30 -08:00
sparse_xent_op.cc
Rolling back PR #44894 because it caused type check errors (got bfloat16 when expecting float).
2020-12-07 17:30:30 -08:00
sparse_xent_op.h
…
spectrogram_convert_test_data.cc
…
spectrogram_op_test.cc
…
spectrogram_op.cc
…
spectrogram_test_utils.cc
…
spectrogram_test_utils.h
…
spectrogram_test.cc
…
spectrogram.cc
…
spectrogram.h
…
split_lib_cpu.cc
Delete SYCL support
2020-09-15 11:21:47 -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
Delete SYCL support
2020-09-15 11:21:47 -07:00
split_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
split_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
split_v_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
split_v_op.cc
Merge pull request #40116 from kitstar:master
2020-07-22 10:54:12 -07:00
squared-loss.h
…
stack_ops.cc
…
stack.cc
Merge pull request #40697 from tg-at-google:sign-compare-warning-fixes-batch-3
2020-06-24 14:38:10 -07:00
stack.h
…
stage_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
stateful_random_ops_cpu_gpu.h
Internal change
2020-09-17 21:37:45 -07:00
stateful_random_ops_gpu.cu.cc
Internal change
2020-09-17 21:37:45 -07:00
stateful_random_ops.cc
Internal change
2020-09-17 21:37:45 -07:00
stateful_random_ops.h
Internal change
2020-09-17 21:37:45 -07:00
stateless_random_gamma_op_gpu.cu.cc
Add GPU kernel for tf.random.stateless_gamma.
2021-01-15 03:01:47 -08:00
stateless_random_gamma_op.cc
Add GPU kernel for tf.random.stateless_gamma.
2021-01-15 03:01:47 -08:00
stateless_random_gamma_op.h
Add GPU kernel for tf.random.stateless_gamma.
2021-01-15 03:01:47 -08:00
stateless_random_ops_v2.cc
Splits StatelessRandomGetKeyCounter into two ops because XLA propagates a compile-time-constant requirement (originating from V2 stateless RNG ops) on output alg to input seed, which prevents seed from depending on a variable. With the split ops, alg no longer depends on seed.
2020-11-16 12:17:48 -08:00
stateless_random_ops_v2.h
Internal change
2020-09-17 21:37:45 -07:00
stateless_random_ops.cc
Move tf.random.stateless_gamma CPU kernel to a separate file.
2020-12-23 10:22:47 -08: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
Delete SYCL support
2020-09-15 11:21:47 -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
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
strided_slice_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
strided_slice_op.h
…
string_format_op_test.cc
…
string_format_op.cc
…
string_join_op.cc
…
string_length_op.cc
…
string_lower_op.cc
…
string_ngrams_op_test.cc
…
string_ngrams_op.cc
Validate data_splits for tf.StringNGrams.
2020-09-18 16:00:29 -07:00
string_split_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
string_split_op.cc
…
string_strip_op.cc
…
string_to_hash_bucket_fast_op.cc
Adds StringToHashBucketFast Op to TF android selective registration builds, by split out the fast op into its own .h/.cc
2020-11-17 17:36:31 -08:00
string_to_hash_bucket_fast_op.h
Adds StringToHashBucketFast Op to TF android selective registration builds, by split out the fast op into its own .h/.cc
2020-11-17 17:36:31 -08:00
string_to_hash_bucket_op.cc
Adds StringToHashBucketFast Op to TF android selective registration builds, by split out the fast op into its own .h/.cc
2020-11-17 17:36:31 -08:00
string_to_hash_bucket_op.h
Adds StringToHashBucketFast Op to TF android selective registration builds, by split out the fast op into its own .h/.cc
2020-11-17 17:36:31 -08:00
string_to_number_op.cc
…
string_upper_op.cc
…
string_util.cc
…
string_util.h
…
substr_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
substr_op.cc
…
summary_audio_op_test.cc
…
summary_audio_op.cc
…
summary_image_op_test.cc
…
summary_image_op.cc
…
summary_interface.h
…
summary_kernels.cc
…
summary_op_test.cc
removed scalar summary tests from tensorflow/core
2020-07-27 23:31:57 +00:00
summary_tensor_op_test.cc
…
summary_tensor_op.cc
…
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
…
tensor_cord.cc
…
tensor_cord.h
…
tensor_flag_utils_test.cc
…
tensor_flag_utils.cc
…
tensor_flag_utils.h
…
tensor_list.cc
…
tensor_list.h
Fix several narrowing issues on 32-bit devices
2020-08-11 15:48:13 -07:00
tensor_map_test.cc
Merge pull request #42300 from kttian:list_keys
2020-08-25 10:12:32 -07:00
tensor_map.cc
Merge pull request #42141 from kttian:erase_grad
2020-08-12 17:42:08 -07:00
tensor_map.h
Merge pull request #42300 from kttian:list_keys
2020-08-25 10:12:32 -07:00
text_line_reader_op.cc
…
tf_record_reader_op.cc
…
tile_functor_cpu_bfloat16.cc
…
tile_functor_cpu_bool.cc
…
tile_functor_cpu_complex64.cc
…
tile_functor_cpu_complex128.cc
…
tile_functor_cpu_double.cc
…
tile_functor_cpu_float.cc
…
tile_functor_cpu_half.cc
…
tile_functor_cpu_int8.cc
…
tile_functor_cpu_int16.cc
…
tile_functor_cpu_int32.cc
…
tile_functor_cpu_int64.cc
…
tile_functor_cpu_tstring.cc
…
tile_functor_cpu_uint8.cc
…
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
Delete SYCL support
2020-09-15 11:21:47 -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
Fix stack-buffer-overflow in tile_functor_gpu.h. host_buf has size = size(int32)*num elements but we are trying to copy size(int64)*num elements out of it.
2020-10-22 14:40:40 -07:00
tile_functor_sycl.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
tile_functor.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
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
Delete SYCL support
2020-09-15 11:21:47 -07:00
tile_ops_gpu_impl_1.cu.cc
…
tile_ops_gpu_impl_2.cu.cc
…
tile_ops_gpu_impl_3.cu.cc
…
tile_ops_gpu_impl_4.cu.cc
…
tile_ops_gpu_impl_5.cu.cc
…
tile_ops_gpu_impl_6.cu.cc
…
tile_ops_gpu_impl_7.cu.cc
…
tile_ops_gpu_impl_8.cu.cc
…
tile_ops_gpu_impl.h
…
tile_ops_impl.h
…
tile_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
topk_op_gpu_double.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_float.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_half.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_int8.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_int16.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_int32.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_int64.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_uint8.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_uint16.cu.cc
Fixing and enabling TopK on ROCm
2020-09-16 16:05:37 +00:00
topk_op_gpu_uint32.cu.cc
Define ROCM TopKFunctor for uint{32,64}.
2020-10-20 16:27:58 -07:00
topk_op_gpu_uint64.cu.cc
Define ROCM TopKFunctor for uint{32,64}.
2020-10-20 16:27:58 -07:00
topk_op_gpu.h
Fix ROCM build after #39429 and ce2a22c1dd
2020-10-19 17:06:19 -07:00
topk_op.cc
Merge pull request #39429 from ekuznetsov139:google-upstream-topk
2020-10-18 22:12:36 -07:00
topk_op.h
…
training_op_helpers.cc
…
training_op_helpers.h
…
training_ops_gpu.cu.cc
Add GPU kernels for SparseApply[Proximal]Adagrad
2020-12-18 10:52:39 +11:00
training_ops_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
training_ops.cc
Add GPU kernels for SparseApply[Proximal]Adagrad
2020-12-18 10:52:39 +11:00
training_ops.h
Address minor review comments from PR 44919
2020-11-23 18:14:03 +11:00
transpose_functor_cpu.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
transpose_functor_gpu.cu.cc
Fix size computation logic in TransposeSimple
2020-07-14 11:36:38 -07:00
transpose_functor.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
transpose_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
transpose_op.h
Delete SYCL support
2020-09-15 11:21:47 -07:00
transpose_util_test.cc
…
typed_conditional_accumulator_base.h
…
typed_queue.h
…
unary_ops_composition_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
unary_ops_composition.cc
…
unicode_ops.cc
…
unicode_script_op.cc
…
unique_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
unique_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
unpack_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
unravel_index_op.cc
Fix several narrowing issues on 32-bit devices
2020-08-11 15:48:13 -07:00
unsorted_segment_join_op.cc
…
variable_ops_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
variable_ops.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
variable_ops.h
…
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
…
where_op.cc
fix typos in core directory
2020-10-29 02:52:55 +03:00
where_op.h
…
whole_file_read_ops.cc
…
winograd_transform.h
…
word2vec_kernels.cc
…
xent_op_gpu.cu.cc
…
xent_op_test.cc
Internal tests cleanup.
2020-10-29 15:56:05 -07:00
xent_op.cc
Delete SYCL support
2020-09-15 11:21:47 -07:00
xent_op.h
…
xsmm_conv2d_test.cc
…
xsmm_conv2d.cc
…
xsmm_conv2d.h
…