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