| .. |
|
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 93c22a269b. 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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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 93c22a269b. 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 93c22a269b. 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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
tile_functor_cpu_uint64.cc
|
|
|
|
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
|
|
|
