Skip to content
Branch: master
Find file History
jsimsa and tensorflower-gardener [tf.data] Follow up to cl/270460372 which extends the multi-device fu…
…nction check with an op kernel registration check.

PiperOrigin-RevId: 270469526
Latest commit 6e17966 Sep 21, 2019
Permalink
Type Name Latest commit message Commit time
..
Sorry, we had to truncate this directory to 1,000 files. 203 entries were omitted from the list.
Failed to load latest commit information.
batching_util Instrument Batch Scheduler Queue with TraceMe Sep 6, 2019
boosted_trees Support empty ensemble with vector leafs. Sep 11, 2019
data [tf.data] Follow up to cl/270460372 which extends the multi-device fu… Sep 21, 2019
fuzzing Fix one_hot fuzzer timeout when input is way too long. Sep 13, 2019
hexagon Remove SplitAndParseAsInts. Aug 12, 2019
neon Adjust structure of all BUILD files to recommended style (https://doc… Jun 7, 2019
rnn Added BlockLSTMV2 and BlockLSTMGradV2 with IFCO layout Aug 15, 2019
sparse De-duplicate the CSRSparseMatrixToCOOSparseMatrix functor in the CSRS… Sep 4, 2019
BUILD Merge pull request #32567 from houtoms:support_cpu_nchw_batch_norm Sep 18, 2019
adjust_contrast_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
adjust_contrast_op.h Add fp16 support to color distortion ops for better performance on GP… Jan 15, 2019
adjust_contrast_op_benchmark_test.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
adjust_contrast_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
adjust_contrast_op_test.cc Branch 183429339 (#16469) Jan 26, 2018
adjust_hsv_gpu.cu.h Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
adjust_hue_op.cc Add ROCm support for adjust_hue and adjust_saturation op May 30, 2019
adjust_hue_op.h Add ROCm support for adjust_hue and adjust_saturation op May 30, 2019
adjust_hue_op_gpu.cu.cc Merge pull request #29169 from ROCmSoftwarePlatform:google_upstream_a… Jun 4, 2019
adjust_saturation_op.cc Add ROCm support for adjust_hue and adjust_saturation op May 30, 2019
adjust_saturation_op.h Add ROCm support for adjust_hue and adjust_saturation op May 30, 2019
adjust_saturation_op_gpu.cu.cc Merge pull request #29169 from ROCmSoftwarePlatform:google_upstream_a… Jun 4, 2019
aggregate_ops.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
aggregate_ops.h fix C++ header guards. Aug 21, 2018
aggregate_ops_cpu.h fix C++ header guards. Aug 21, 2018
aggregate_ops_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
argmax_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
argmax_op.h fix C++ header guards. Aug 21, 2018
argmax_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
as_string_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
assign_op.h fix C++ header guards. Aug 21, 2018
attention_ops.cc Merge pull request #17690 from yongtang:2134-zero-noise Jan 29, 2019
avgpooling_op.cc Merge pull request #31207 from houtoms:remote_nhwc_avg_pooling Aug 5, 2019
avgpooling_op.h fix C++ header guards. Aug 21, 2018
avgpooling_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
barrier_ops.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
base64_ops.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
basic_ops_benchmark_test.cc Merge changes from github. May 19, 2017
batch_kernels.cc Make all function library runtime invocations use a new randomly gene… May 7, 2019
batch_matmul_op_complex.cc Adding ROCm support for the batch_matmul op Jul 2, 2019
batch_matmul_op_impl.h [SE] Remove Stream* argument from ScratchAllocator methods Aug 5, 2019
batch_matmul_op_real.cc Adding ROCm support for the batch_matmul op Jul 2, 2019
batch_matmul_op_test.cc Add broadcasting support to tf.matmul. Mar 29, 2019
batch_norm_op.cc Adding ROCm support for the batch_norm op May 29, 2019
batch_norm_op.h fix C++ header guards. Aug 21, 2018
batch_norm_op_gpu.cu.cc Adding ROCm support for the batch_norm op May 29, 2019
batch_norm_op_test.cc Branch 183429339 (#16469) Jan 26, 2018
batch_util.h batch_util.h is generally useful so moved to util/ from kernels/ wher… May 22, 2018
batchtospace_op.cc Adding ROCm support for batch_to_space op Jul 3, 2019
bcast_ops.cc Use absl instead of deprecated str_util May 21, 2019
betainc_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
betainc_op.h fix C++ header guards. Aug 21, 2018
betainc_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
bias_op.cc Adding ROCm support for the bias op Jul 10, 2019
bias_op.h fix C++ header guards. Aug 21, 2018
bias_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
bias_op_gpu.h ENH: The gradient op of bias_add supports 3/4/5D NCHW format Dec 28, 2018
bias_op_test.cc Use real time in BiasAddGrad benchmarks Mar 8, 2019
bincount_op.cc Adding ROCm support for the bincount op Jun 25, 2019
bincount_op.h fix C++ header guards. Aug 21, 2018
bincount_op_gpu.cu.cc Adding ROCm support for the bincount op Jun 25, 2019
bincount_op_test.cc Merge changes from github. Nov 22, 2017
broadcast_to_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
broadcast_to_op.h Implement tf.broadcast_to in a way the can handle higher ranks. This … Feb 8, 2019
broadcast_to_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
bucketize_op.cc Enable GPU tests for bucketize op and fix a hard crash on empty input… Sep 10, 2019
bucketize_op.h fix C++ header guards. Aug 21, 2018
bucketize_op_gpu.cu.cc added __restrict__ to all possible __global__ kernels Aug 1, 2019
candidate_sampler_ops.cc Remove (Mutable)ArraySlice implementation and alias them to absl::Span. Aug 30, 2018
cast_op.cc Implement default kernels and move a few ops to use the default kernels. Jun 8, 2019
cast_op.h fix C++ header guards. Aug 21, 2018
cast_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl.h Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_bfloat.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_bool.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_complex128.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_complex64.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_double.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_float.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_half.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_int16.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_int32.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_int64.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_int8.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_uint16.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_uint32.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_uint64.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_impl_uint8.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
cast_op_test.cc enabling the cast op unit-test on the ROCm platform Jul 11, 2019
check_numerics_op.cc Adding ROCm support for check_numerics op May 29, 2019
check_numerics_op_gpu.cu.cc added __restrict__ to all possible __global__ kernels Aug 1, 2019
cholesky_grad.cc Internal change May 30, 2018
cholesky_op.cc Automated rollback of commit b52fac8 Aug 13, 2019
clustering_ops.cc Move clustering ops to core. Jan 11, 2019
clustering_ops_test.cc Move clustering ops to core. Jan 11, 2019
collective_nccl.cc PR #31485: [ROCm] add ROCm RCCL support Aug 22, 2019
collective_nccl.h PR #31485: [ROCm] add ROCm RCCL support Aug 22, 2019
collective_nccl_broadcaster.cc Merge pull request #31483 from ROCmSoftwarePlatform:google_upstream_n… Sep 5, 2019
collective_nccl_broadcaster.h PR #31485: [ROCm] add ROCm RCCL support Aug 22, 2019
collective_nccl_gatherer.cc Merge pull request #31483 from ROCmSoftwarePlatform:google_upstream_n… Sep 5, 2019
collective_nccl_gatherer.h PR #31485: [ROCm] add ROCm RCCL support Aug 22, 2019
collective_nccl_reducer.cc Do not block inside NCCL OpKernel when computing all-sum. Sep 13, 2019
collective_nccl_reducer.h PR #31485: [ROCm] add ROCm RCCL support Aug 22, 2019
collective_nccl_test.cc Make DeviceMgr a pure virtual interface with StaticDeviceMgr as its o… Aug 31, 2019
collective_ops.cc Add a `communication_hint` argument to collective ops. Aug 13, 2019
colorspace_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
colorspace_op.h Fix include guards Aug 20, 2018
colorspace_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
colorspace_op_test.cc Branch 183429339 (#16469) Jan 26, 2018
compare_and_bitpack_op.cc Adding ROCm support for comapre_and_bitpack op May 29, 2019
compare_and_bitpack_op.h Remove THIRD_PARTY_ from #include guards Jan 24, 2018
compare_and_bitpack_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
concat_lib.h Adding ROCm support for the "concat" op May 21, 2019
concat_lib_cpu.cc Automated rollback of commit 4485ff2 Aug 12, 2019
concat_lib_cpu.h Fix include guards Aug 17, 2018
concat_lib_gpu.cc Adding ROCm support for the "concat" op May 21, 2019
concat_lib_gpu.h Changing from Cuda* to Gpu*, the contents for tensforflow/core/kernel… Mar 14, 2019
concat_lib_gpu_impl.cu.cc Remove unsupported __restrict__ annotation Aug 22, 2019
concat_op.cc Register uint32 and uint64 types for Concat and StridedSlice on CPU. Aug 1, 2019
concat_op_test.cc Register tf.concat with uint8 data type. Apr 9, 2018
conditional_accumulator.h Extend ConditionalAccumulator with SUM functionality. Sep 6, 2018
conditional_accumulator_base.cc Extend ConditionalAccumulator with SUM functionality. Sep 6, 2018
conditional_accumulator_base.h Make DebugString() in ResourceBase constant Jan 8, 2019
conditional_accumulator_base_op.cc Boosted Trees: switching to version 2 cond (even when running in v1).… Jun 17, 2019
conditional_accumulator_base_op.h Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
conditional_accumulator_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
constant_op.cc New GPU BFC memory allocation analysis tool. Sep 19, 2019
constant_op.h Automated rollback of commit a00fba3 Aug 15, 2018
constant_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
constant_op_test.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
control_flow_ops.cc Register XLA device kernel for _SwitchN op. Sep 4, 2019
control_flow_ops.h Adds a lowering from Case to _SwitchN+Merge. May 31, 2019
control_flow_ops_test.cc Automated rollback of commit 4485ff2 Aug 12, 2019
conv_2d.h Keep TransformDepth Sep 7, 2019
conv_2d_gpu.h Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
conv_2d_gpu_double.cu.cc Add ROCm support for launching 2D GPU convolutions May 30, 2019
conv_2d_gpu_float.cu.cc Keep TransformDepth Sep 7, 2019
conv_2d_gpu_half.cu.cc clean-up unused functors Sep 7, 2019
conv_2d_gpu_int.cu.cc Split conv GPU code into multiple files Nov 26, 2018
conv_2d_gpu_uint16.cu.cc Add ROCm support for launching 2D GPU convolutions May 30, 2019
conv_2d_gpu_uint32.cu.cc Add ROCm support for launching 2D GPU convolutions May 30, 2019
conv_2d_gpu_uint64.cu.cc Add ROCm support for launching 2D GPU convolutions May 30, 2019
conv_2d_gpu_uint8.cu.cc Add ROCm support for launching 2D GPU convolutions May 30, 2019
conv_3d.h Use Eigen::CuboidConvolutionBackwardKernel in Conv3DBackpropFilter. Sep 12, 2018
conv_grad_filter_ops.cc Merge pull request #27524 from ppwwyyxx:master Sep 10, 2019
conv_grad_filter_ops_benchmark_test.cc Compute Conv2DBackpropFilter in NHWC format for fp16 on Volta. May 29, 2019
conv_grad_input_ops.cc Merge pull request #27524 from ppwwyyxx:master Sep 10, 2019
conv_grad_input_ops_benchmark_test.cc Compute Conv2DBackpropInput in NHWC format for fp16 on Volta May 29, 2019
conv_grad_ops.cc [TF:XLA] Add explicit padding support for Conv2D in tf2xla. Feb 6, 2019
conv_grad_ops.h Add ROCm support for 2D and 3D convolution ops. Jun 26, 2019
conv_grad_ops_3d.cc Merge pull request #31908 from ROCmSoftwarePlatform:google-upstream-p… Sep 7, 2019
conv_ops.cc Merge pull request #31908 from ROCmSoftwarePlatform:google-upstream-p… Sep 7, 2019
conv_ops.h Add ROCm support for 2D and 3D convolution ops. Jun 26, 2019
conv_ops_3d.cc Merge pull request #31908 from ROCmSoftwarePlatform:google-upstream-p… Sep 7, 2019
conv_ops_benchmark_test.cc Compute Conv2DBackpropFilter in NHWC format for fp16 on Volta. May 29, 2019
conv_ops_fused_double.cc Remove EIGEN_USE_LIBXSMM support from Tensorflow. Aug 19, 2019
conv_ops_fused_float.cc Remove EIGEN_USE_LIBXSMM support from Tensorflow. Aug 19, 2019
conv_ops_fused_half.cc Roll-forward: Mar 7, 2019
conv_ops_fused_image_transform.cc Adding 'half_pixel_centers' bool attribute (default False) to resize … Feb 26, 2019
conv_ops_fused_impl.h Merge pull request #31908 from ROCmSoftwarePlatform:google-upstream-p… Sep 7, 2019
conv_ops_gpu.h Merge pull request #31492 from danganea:3DGroupConvs Aug 12, 2019
conv_ops_test.cc Merge pull request #31460 from ROCmSoftwarePlatform:google_upstream_r… Aug 12, 2019
conv_ops_using_gemm.cc Update references to bounds_checks.h Jan 22, 2019
count_up_to_op.cc Use RefCountPtr in LookupResource to avoid leaks May 29, 2019
crop_and_resize_op.cc Add ROCm support for crop and resize op May 29, 2019
crop_and_resize_op.h Merge pull request #24011 from Intel-tensorflow:feature/daoxin/cropAn… Mar 21, 2019
crop_and_resize_op_benchmark_test.cc Automated rollback of commit 1bf16be. Revert #21681. Nov 5, 2018
crop_and_resize_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
crop_and_resize_op_test.cc Use absl instead of deprecated str_util May 21, 2019
cross_op.cc Adding ROCm support to cross_op May 30, 2019
cross_op.h Fix include guards Aug 20, 2018
cross_op_gpu.cu.cc Adding ROCm support to cross_op May 30, 2019
cross_op_test.cc Remove unused protobuf header inclusions Jun 6, 2017
ctc_decoder_ops.cc Merge pull request #31164 from aprimostka:ctc-float64 Aug 20, 2019
ctc_loss_op.cc Merge pull request #31164 from aprimostka:ctc-float64 Aug 20, 2019
cuda_solvers.cc Wrap cuSolver implementation of batched Cholesky decomposition in pot… Jul 12, 2019
cuda_solvers.h Automated rollback of commit b52fac8 Aug 13, 2019
cuda_sparse.cc Add CSRSparseMatrix ops. Sep 3, 2019
cuda_sparse.h Add CSRSparseMatrix ops. Sep 3, 2019
cudnn_pooling_gpu.cc Add ROCm support for pooling operators Jun 28, 2019
cudnn_pooling_gpu.h Add ROCm support for pooling operators Jun 28, 2019
cudnn_rnn_ops.cc Merge pull request #30889 from houtoms:pr_padded_io_cudnn_rnn Aug 7, 2019
cwise_op_abs.cc Support tf.abs on CPU for int8, int16, and bfloat16. Sep 12, 2019
cwise_op_acos.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_acosh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_add_1.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_add_2.cc Automated rollback of commit 4485ff2 Aug 12, 2019
cwise_op_arg.cc Enable complex for tf.angle on GPU. Dec 5, 2018
cwise_op_asin.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_asinh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_atan.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_atan2.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_atanh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_bessel.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_bessel.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_bitwise_and.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_bitwise_or.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_bitwise_xor.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_ceil.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_clip.cc Add ROCm support for remaining cwise ops and tests Jun 3, 2019
cwise_op_clip.h Merge changes from github. Apr 14, 2018
cwise_op_clip_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
cwise_op_complex.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_conj.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_cos.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_cosh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_digamma.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_div.cc Add ROCm support for remaining cwise ops and tests Jun 3, 2019
cwise_op_equal_to_1.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_equal_to_2.cc Automated rollback of commit 4485ff2 Aug 12, 2019
cwise_op_erf.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_erfc.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_exp.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_expm1.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_floor.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_floor_div.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_floor_mod.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_abs.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_acos.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_acosh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_add.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_arg.cu.cc Enable complex for tf.angle on GPU. Dec 5, 2018
cwise_op_gpu_asin.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_asinh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_atan.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_atan2.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_atanh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_bitwise_and.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_bitwise_or.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_bitwise_xor.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_ceil.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_complex.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_conj.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_cos.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_cosh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_digamma.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_div.cu.cc Add ROCm support for remaining cwise ops and tests Jun 3, 2019
cwise_op_gpu_equal_to.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_erf.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_erfc.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_exp.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_expm1.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_floor.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_floor_div.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_floor_mod.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_greater.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_greater_equal.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_igammas_double.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_igammas_float.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_imag.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_inverse.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_invert.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_isfinite.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_isinf.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_isnan.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_left_shift.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_less.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_less_equal.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_lgamma.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_log.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_log1p.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_logical_and.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_logical_not.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_logical_or.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_maximum.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_minimum.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_mod.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_mul.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_neg.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_not_equal_to.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_pow.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_random_grad.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_real.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_right_shift.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_rint.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_round.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_rsqrt.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_select.cu.cc Let SelectV2 handle dimensions up to 8. This matches recent changes t… Jun 26, 2019
cwise_op_gpu_sigmoid.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_sign.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_sin.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_sinh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_sqrt.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_square.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_squared_difference.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_sub.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_tan.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_tanh.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_xdivy.cu.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_gpu_xlogy.cu.cc Update cwise_op_gpu_xlogy.cu.cc Aug 15, 2019
cwise_op_gpu_zeta.cu.cc Add ROCm support for remaining cwise ops and tests Jun 3, 2019
cwise_op_greater.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_greater_equal.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_igammas.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_imag.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_invert.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_isfinite.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_isinf.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_isnan.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_left_shift.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_less.cc Removed duplicated registration of Less with bfloat16 Jul 8, 2019
cwise_op_less_equal.cc Removed duplicated registration of LessEqual with bfloat16 Aug 8, 2019
cwise_op_lgamma.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_log.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_log1p.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_logical_and.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_logical_not.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_logical_or.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_maximum.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_minimum.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_mod.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_mul_1.cc Add ROCm support for remaining cwise ops and tests Jun 3, 2019
cwise_op_mul_2.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_neg.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_not_equal_to_1.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_not_equal_to_2.cc Automated rollback of commit 4485ff2 Aug 12, 2019
cwise_op_pow.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_random_grad.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_real.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_reciprocal.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_right_shift.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_rint.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_round.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_rsqrt.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_select.cc Let SelectV2 handle dimensions up to 8. This matches recent changes t… Jun 26, 2019
cwise_op_sigmoid.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_sign.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_sin.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_sinh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_sqrt.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_square.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_squared_difference.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_sub.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_tan.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_tanh.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_xdivy.cc Add ROCm support for most cwise ops May 31, 2019
cwise_op_xlogy.cc Add xlogy and xdivy op. Sep 27, 2018
cwise_op_zeta.cc enable cwise op zeta for ROCm Jun 3, 2019
cwise_ops.h Pre compute scalar to packet if expression is using host memory Jun 28, 2019
cwise_ops_common.cc Add incompatible_shape_error attribute to equal op Sep 5, 2019
cwise_ops_common.h BinaryOp::Compute stops when BinaryOpState's constructor OOMs. Sep 16, 2019
cwise_ops_gpu_common.cu.h Add ROCm support for most cwise ops May 31, 2019
cwise_ops_gpu_gradients.cu.h Add ROCm support for most cwise ops May 31, 2019
cwise_ops_gradients.h set PacketAccess to false for TENSORFLOW_USE_ROCM in cwise_ops_gradie… Jun 4, 2019
cwise_ops_sycl_common.h Branch 183429339 (#16469) Jan 26, 2018
cwise_ops_test.cc Add a grappler optimization (CPU) for `pow(x, 3)` -> `mul(x, square(x… Jul 9, 2019
data_format_ops.cc [Grappler] Register DataFormatDimMap on host for CPU. Jul 3, 2019
data_format_ops.h fix C++ header guards. Aug 21, 2018
data_format_ops_gpu.cu.cc Adding ROCm support for data format ops May 28, 2019
debug_ops.cc Adding ROCm support for debug ops May 28, 2019
debug_ops.h Adding ROCm support for debug ops May 28, 2019
debug_ops_test.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
decode_bmp_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
decode_compressed_op.cc Updated ReaderInterface and subclasses to use tstring. Aug 28, 2019
decode_csv_op.cc Fix some unnecessary string copies when reading from or writing to a … Aug 8, 2019
decode_image_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
decode_padded_raw_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
decode_proto_op.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
decode_raw_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
decode_wav_op.cc Fix some unnecessary string copies when reading from or writing to a … Aug 8, 2019
decode_wav_op_test.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
deep_conv2d.cc Merge pull request #23382 from pmgdeb:spell Nov 27, 2018
deep_conv2d.h Remove THIRD_PARTY_ from #include guards Jan 24, 2018
deep_conv2d_test.cc Experimental Conv2D implementation specialized for deep convolutions … Aug 31, 2016
dense_update_functor.cc Automated rollback of commit 4485ff2 Aug 12, 2019
dense_update_functor.h fix C++ header guards. Aug 21, 2018
dense_update_functor_gpu.cu.cc Adding ROCm support for the dense update ops May 6, 2019
dense_update_ops.cc Adding ROCm support for the dense update ops May 6, 2019
depthtospace_op.cc Adding ROCm support for the "depth_to_space" op May 21, 2019
depthtospace_op.h Implement NCHW_VECT_C support for tf.depth_to_space on GPU. Oct 12, 2017
depthtospace_op_gpu.cu.cc Adding ROCm support for the "depth_to_space" op May 21, 2019
depthwise_conv_grad_op.cc Adding ROCm support to depthwise_conv_grad_op Jul 11, 2019
depthwise_conv_op.cc [ROCm] Adding support to depthwise_conv_op Jul 9, 2019
depthwise_conv_op.h [ROCm] Adding support to depthwise_conv_op Jul 9, 2019
depthwise_conv_op_gpu.h Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
depthwise_conv_op_gpu_double.cu.cc [ROCm] Adding support to depthwise_conv_op Jul 9, 2019
depthwise_conv_op_gpu_float.cu.cc [ROCm] Adding support to depthwise_conv_op Jul 9, 2019
depthwise_conv_op_gpu_half.cu.cc [ROCm] Adding support to depthwise_conv_op Jul 9, 2019
depthwise_conv_ops_test.cc Support for pseudo half precision in depthwise convolution kernels. Jul 17, 2018
dequantize_op.cc Add 'narrow_range' and 'int axis' options to tf.quantize and_tf.dequa… Sep 7, 2019
dequantize_op_test.cc Add 'narrow_range' and 'int axis' options to tf.quantize and_tf.dequa… Sep 7, 2019
deserialize_sparse_string_op.cc Automated rollback of commit 4485ff2 Aug 12, 2019
deserialize_sparse_variant_op.cc [tf.data] Optimize the implementation of DeserializeSparseOp<Variant>. Jun 29, 2018
determinant_op.cc * Add GPU implementation of LogDeterminant op. Oct 24, 2017
determinant_op.h Remove THIRD_PARTY_ from #include guards Jan 24, 2018
determinant_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
diag_op.cc Adding ROCm support to diag_op May 31, 2019
diag_op.h Branch 183429339 (#16469) Jan 26, 2018
diag_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
diag_op_test.cc Branch 183429339 (#16469) Jan 26, 2018
dilation_ops.cc Adding ROCm support for the dilation ops May 30, 2019
dilation_ops.h Morphological filtering operations: Dilation and erosion. Jun 3, 2016
dilation_ops_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
draw_bounding_box_op.cc Merge pull request #24282 from yongtang:15692-draw_bounding_boxes-colors Apr 5, 2019
dynamic_partition_op.cc Update references to bounds_checks.h Jan 22, 2019
dynamic_partition_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
dynamic_partition_op_test.cc Use absl instead of deprecated str_util May 21, 2019
dynamic_stitch_op.cc Adding ROCm support for the dynamic_stitch op Jun 10, 2019
dynamic_stitch_op_gpu.cu.cc Adding ROCm support for the dynamic_stitch op Jun 10, 2019
dynamic_stitch_op_test.cc Use absl instead of deprecated str_util May 21, 2019
edit_distance_op.cc Use the safe sparse tensor API that returns errors rather than crashing Jul 16, 2018
eigen_activations.h Branch 183429339 (#16469) Jan 26, 2018
eigen_activations_test.cc Add benchmark for the SpatialConvolution specialized gemm_pack_rhs. Nov 2, 2018
eigen_attention.h Convert string to enum for performance/GPU Jan 28, 2019
eigen_attention_test.cc Add benchmark for the SpatialConvolution specialized gemm_pack_rhs. Nov 2, 2018
eigen_backward_cuboid_convolutions.h Use Eigen::CuboidConvolutionBackwardKernel in Conv3DBackpropFilter. Sep 12, 2018
eigen_backward_spatial_convolutions.h Register Eigen kernel for Conv2DBackpropFilter on CPU Jul 3, 2019
eigen_backward_spatial_convolutions_test.cc Register Eigen kernel for Conv2DBackpropFilter on CPU Jul 3, 2019
eigen_benchmark.h Fix bugs in backward convolution benchmarks + add labels. Sep 4, 2018
eigen_benchmark_cpu_test.cc PR #26235: typos in tensorflow/core fixed Mar 8, 2019
eigen_contraction_kernel.cc Merge pull request #29395 from ROCmSoftwarePlatform:google-upstream-p… Jun 11, 2019
eigen_contraction_kernel.h Update subroutine to make it device compatible Jun 4, 2019
eigen_convolution_helpers.h Optimized packet load in gemm_pack_rhs (Packet16f/8f) for CuboidalCon… Jul 3, 2019
eigen_cuboid_convolution.h Merge pull request #30138 from anuj-rawat:eigen_cuboid_convolution_ge… Jul 4, 2019
eigen_mkldnn_contraction_kernel_test.cc Ensure that clang-format won't reorder includes Jul 30, 2019
eigen_pooling.h Merge pull request #30611 from ROCmSoftwarePlatform:google_upstream_r… Jul 18, 2019
eigen_pooling_test.cc Add benchmark for the SpatialConvolution specialized gemm_pack_rhs. Nov 2, 2018
eigen_spatial_convolutions-inl.h Merge pull request #30138 from anuj-rawat:eigen_cuboid_convolution_ge… Jul 4, 2019
eigen_spatial_convolutions.h Fast gemm_pack_colmajor for non standard patches. Jun 26, 2019
eigen_spatial_convolutions_test.cc Merge pull request #26252 from markdryan:fix-spatial-convolutions-ben… Aug 20, 2019
eigen_volume_patch.h Define PreferBlockAccess enum to prepare for Eigen upgrade. Sep 14, 2018
einsum_op.h Roll CL 258349270 forward with fix. Aug 27, 2019
einsum_op_gpu.cu.cc Add more dtypes to Einsum Op. Second try, as the first one broke glob… Sep 3, 2019
einsum_op_impl.h Add support for zero-sized inputs to EinsumOp. Sep 10, 2019
einsum_op_impl_bfloat16.cc Add more dtypes to Einsum Op. Second try, as the first one broke glob… Sep 3, 2019
einsum_op_impl_complex128.cc Split einsum_op implementatin in multiple files to reduce compilation… Aug 30, 2019
einsum_op_impl_complex64.cc Split einsum_op implementatin in multiple files to reduce compilation… Aug 30, 2019
einsum_op_impl_double.cc Split einsum_op implementatin in multiple files to reduce compilation… Aug 30, 2019
einsum_op_impl_float.cc Split einsum_op implementatin in multiple files to reduce compilation… Aug 30, 2019
einsum_op_impl_half.cc Add more dtypes to Einsum Op. Second try, as the first one broke glob… Sep 3, 2019
einsum_op_impl_int32.cc Add more dtypes to Einsum Op. Second try, as the first one broke glob… Sep 3, 2019
einsum_op_impl_int64.cc Add more dtypes to Einsum Op. Second try, as the first one broke glob… Sep 3, 2019
encode_jpeg_op.cc Update kernels and related libs to use tstring. Aug 28, 2019
encode_png_op.cc Update kernels and related libs to use tstring. Aug 28, 2019
encode_proto_op.cc Update kernels and related libs to use tstring. Aug 28, 2019
encode_wav_op.cc Update kernels and related libs to use tstring. Aug 28, 2019
encode_wav_op_test.cc Remove using directives. Test appertaining to ops have been moved int… Jan 2, 2018
example_parsing_ops.cc Automated rollback of commit c839f14 Sep 9, 2019
example_parsing_ops_test.cc Update FastParseSingleExample to handle ragged tensors. Sep 16, 2019
extract_image_patches_op.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
extract_image_patches_op.h
extract_image_patches_op_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
extract_jpeg_shape_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
extract_volume_patches_op.cc Added ROCm support for the extract_volume_patches op Mar 19, 2019
extract_volume_patches_op.h clang-format Sep 15, 2018
extract_volume_patches_op_gpu.cu.cc Added ROCm support for the extract_volume_patches op Mar 19, 2019
eye_functor.h Remove THIRD_PARTY_ from #include guards Jan 24, 2018
eye_functor_gpu.cu.cc added __restrict__ to all possible __global__ kernels Aug 1, 2019
fact_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
fake_quant_ops.cc Add metrics to track the usage of quantization ops Sep 11, 2019
fake_quant_ops_functor.h fix C++ header guards. Aug 21, 2018
fake_quant_ops_gpu.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
fake_quant_ops_test.cc Branch 183429339 (#16469) Jan 26, 2018
fft_ops.cc Add complex128 support to RFFT, RFFT2D, RFFT3D, IRFFT, IRFFT2D, and I… Sep 12, 2019
fifo_queue.cc [TF:XLA] Add partial implementation of tf.FIFOQueue for XLA devices (… Jun 29, 2018
fifo_queue.h Make DebugString() in ResourceBase constant Jan 8, 2019
fifo_queue_op.cc [TF:XLA] Add partial implementation of tf.FIFOQueue for XLA devices (… Jun 29, 2018
fill_functor.cc Automated rollback of commit 4485ff2 Aug 12, 2019
fill_functor.cu.cc Check if GOOGLE_CUDA and TENSORFLOW_USE_ROCM are defined before using… Apr 25, 2019
fill_functor.h Automated rollback of commit 4485ff2 Aug 12, 2019
fingerprint_op.cc Update kernels and related libs to use tstring. Aug 28, 2019
fingerprint_op_test.cc Update kernels and related libs to use tstring. Aug 28, 2019
fixed_length_record_reader_op.cc Updated ReaderInterface and subclasses to use tstring. Aug 28, 2019
fractional_avg_pool_op.cc Qualify calls to some functions from <cmath>. Mar 25, 2019
fractional_max_pool_op.cc Qualify calls to some functions from <cmath>. Mar 25, 2019
fractional_pool_common.cc Adds fractional_max_pool and fractional_avg_pool ops. Fixes #2953. Aug 30, 2016
fractional_pool_common.h fix C++ header guards. Aug 21, 2018
function_ops.cc Automated rollback of commit 4485ff2 Aug 12, 2019
function_ops.h [tf.data] Add experimental transformation for applying `map()` on GPU… Nov 13, 2018
functional_ops.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
fused_batch_norm_ex_op_test.cc Add custom FusedBatchNorm+SideInput+Activation CUDA kernel for is_tra… Jun 14, 2019
fused_batch_norm_op.cc Merge pull request #32567 from houtoms:support_cpu_nchw_batch_norm Sep 18, 2019
fused_batch_norm_op.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
fused_batch_norm_op.h Optimize FusedBatchNormGrad on CPU device. Jul 9, 2019
fused_batch_norm_op_test.cc Optimize FusedBatchNormGrad freezing on GPU Jul 15, 2019
fused_eigen_output_kernels.cc Initial FusedMatMul kernel implementation: MatMul+BiasAdd. Apr 9, 2019
fused_eigen_output_kernels.h Cleanup eigen_contraction_kernel includes Apr 12, 2019
gather_functor.cc Adding ROCm support for the gather_nd op Jun 10, 2019
gather_functor.h : Aug 16, 2019
gather_functor_batched.cc Created a Gather functor that works with batch dimensions. Aug 6, 2019
gather_functor_batched.h : Aug 16, 2019
gather_functor_batched_gpu.cu.cc Created a Gather functor that works with batch dimensions. Aug 6, 2019
gather_functor_batched_gpu.cu.h Created a Gather functor that works with batch dimensions. Aug 6, 2019
gather_functor_gpu.cu.cc Adding ROCm support for gather_op May 30, 2019
gather_functor_gpu.cu.h Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
gather_nd_op.cc Adding ROCm support for the gather_nd op Jun 10, 2019
gather_nd_op.h Move "third_party/tensorflow/core/util/util.h" import from gather_nd_… Apr 10, 2019
gather_nd_op_cpu_impl.h Use parallelFor in GatherNdSlice. Saw speedups from 3-12% on Intel Sk… Jul 12, 2019
gather_nd_op_cpu_impl_0.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_1.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_2.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_3.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_4.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_5.cc Split gather_nd CPU functors into multiple files for faster compile t… Aug 2, 2016
gather_nd_op_cpu_impl_6.cc Make gather_nd & scatter_nd handle rank 6 and 7 indices. Dec 4, 2017
gather_nd_op_cpu_impl_7.cc Make gather_nd & scatter_nd handle rank 6 and 7 indices. Dec 4, 2017
gather_nd_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
gather_nd_op_test.cc [INTEL MKL] Adding support for quantized type gather nd op registration Apr 30, 2019
gather_op.cc Created a Gather functor that works with batch dimensions. Aug 6, 2019
gather_op_test.cc Use absl instead of deprecated str_util May 21, 2019
gemm_functors.h Use mkldnn sgemm in Eigen contractions. Nov 9, 2018
generate_vocab_remapping_op.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
gpu_device_array.h [ROCm] Misc updates for the ROCm platform Jul 11, 2019
gpu_device_array_gpu.h [ROCm] Misc updates for the ROCm platform Jul 11, 2019
gpu_utils.cc Merge pull request #31908 from ROCmSoftwarePlatform:google-upstream-p… Sep 7, 2019
gpu_utils.h Refactor redzone_allocator and ptxas_utils Aug 29, 2019
guarantee_const_op.cc Adds a GuaranteeConstOp. Dec 2, 2017
guarantee_const_op_test.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
hinge-loss.h fix C++ header guards. Aug 21, 2018
histogram_op.cc Segmentation Fault 11 thrown by tensorflow::HistogramFixedWidthFuncto… Jul 11, 2019
histogram_op.h fix C++ header guards. Aug 21, 2018
histogram_op_gpu.cu.cc Adding ROCm support for the histogram op Jun 25, 2019
host_constant_op.cc Register defaults for more kernels. Jun 14, 2019
host_constant_op.h Automated rollback of commit a00fba3 Aug 15, 2018
i_remote_fused_graph_executor.h fix C++ header guards. Aug 21, 2018
i_remote_fused_graph_ops_definitions.cc Add a way to fuse a graph by remote graph executor so that users don'… Jul 18, 2017
i_remote_fused_graph_ops_definitions.h Remove THIRD_PARTY_ from #include guards Jan 24, 2018
identity_n_op.cc Do not perform colocation checks for IdentityN since it just forwards… Jul 9, 2019
identity_n_op.h fix C++ header guards. Aug 21, 2018
identity_n_op_test.cc Automated rollback of commit 4485ff2 Aug 12, 2019
identity_op.cc Automated rollback of commit 4485ff2 Aug 12, 2019
identity_op.h fix C++ header guards. Aug 21, 2018
identity_op_test.cc Automated rollback of commit 4485ff2 Aug 12, 2019
identity_reader_op.cc Updated ReaderInterface and subclasses to use tstring. Aug 28, 2019
image_resizer_state.h Merge pull request #29346 from guoyinchen:aar-build-error-clang3.8 Jun 4, 2019
immutable_constant_op.cc Merge changes from github. Nov 4, 2016
immutable_constant_op.h fix C++ header guards. Aug 21, 2018
immutable_constant_op_test.cc Fix uses of private, mangled names for proto enumerators. Jul 4, 2019
in_topk_op.cc [ROCm] enable InTopK op on ROCm. Aug 13, 2019
in_topk_op.h [ROCm] enable InTopK op on ROCm. Aug 13, 2019
in_topk_op_gpu.cu.cc Merge branch 'master' into restrict Aug 28, 2019
in_topk_op_test.cc Add InTopKV2 kernel for GPU device May 14, 2019
initializable_lookup_table.cc Rolling forward change to add a shared_name by default for HashTableV… May 29, 2019
initializable_lookup_table.h Rolling forward change to add a shared_name by default for HashTableV… May 29, 2019
inplace_ops.cc Automated rollback of commit 4485ff2 Aug 12, 2019
inplace_ops_functor.h fix C++ header guards. Aug 21, 2018
inplace_ops_functor_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
kernel_platform_strings.h Create a shared object that houses all tf kernels. Feb 21, 2019
l2loss_op.cc Branch 183429339 (#16469) Jan 26, 2018
l2loss_op.h fix C++ header guards. Aug 21, 2018
l2loss_op_gpu.cu.cc Adding ROCm support for the l2loss op Jun 25, 2019
linalg_ops_common.cc Branch 183429339 (#16469) Jan 26, 2018
linalg_ops_common.h Add tf.linalg.tridiagonal_mul op for multiplying a tridiagonal matrix… Apr 30, 2019
list_kernels.cc Make TensorList objects Refcounted. Jul 23, 2019
list_kernels.cu.cc Adding ROCm support for tensor list kernels/methods May 30, 2019
list_kernels.h Make TensorList objects Refcounted. Jul 23, 2019
listdiff_op.cc Automated rollback of commit 4485ff2 Aug 12, 2019
lmdb_reader_op.cc Updated ReaderInterface and subclasses to use tstring. Aug 28, 2019
load_and_remap_matrix_op.cc Fix some unnecessary string copies when reading from or writing to a … Aug 8, 2019
logging_ops.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
logging_ops.h Add support for AssertOp in XLA May 20, 2019
logging_ops_test.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
logistic-loss.h Replace log(1 + x) with numerically more stable log1p(x) Feb 27, 2019
lookup_table_init_op.cc Fix some unnecessary string copies when reading from or writing to a … Aug 8, 2019
lookup_table_init_op.h fix C++ header guards. Aug 21, 2018
lookup_table_op.cc Migrated a subset of kernels to use tstring. Aug 14, 2019
lookup_table_op.h Update kernels and related libs to use tstring. Aug 28, 2019
lookup_util.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
lookup_util.h Changing KeyValueTensorInitializer to be more general. Aug 25, 2018
loss.h fix C++ header guards. Aug 21, 2018
loss_test.cc Add the poisson log loss to the SDCA optimizer. Aug 31, 2018
lrn_op.cc Adding ROCm support for the LRN op Jul 26, 2019
lrn_op_test.cc Merge pull request #24169 from markdryan:optimize-lrngrad-cpu May 15, 2019
lu_op.cc Add TensorFlow op to compute the LU decomposition of a matrix. Dec 6, 2018
lu_op_gpu.cu.cc Merge pull request #31223 from ThisIsIsaac:restrict Aug 28, 2019
map_stage_op.cc Adding ROCm support for the map_stage_op Jun 25, 2019
matching_files_op.cc Updated the majority of string tensor accessors to use tstring type. Aug 7, 2019
matmul_op.cc [INTEL MKL] Modification to graph rewrite pass to enable rewrite for … May 25, 2019
matmul_op.h Adding ROCm support for the "matmul" op May 13, 2019
matmul_op_fused.cc Remove EIGEN_USE_LIBXSMM support from Tensorflow. Aug 19, 2019
matmul_op_test.cc Test transpose_a/transpose_b combinations in FusedMatMul test. Apr 9, 2019
matrix_band_part_op.cc Merge pull request #29605 from ROCmSoftwarePlatform:google_upstream_m… Jun 12, 2019
matrix_band_part_op.h Fix include guards Aug 20, 2018
matrix_band_part_op_gpu.cu.cc added __restrict__ to all possible __global__ kernels Aug 1, 2019
matrix_diag_op.cc Add MatrixDiagV2, MatrixDiagPartV2, and MatrixSetDiagV2 for CPU. Jun 12, 2019
matrix_diag_op.h Add MatrixDiagV2, MatrixDiagPartV2, and MatrixSetDiagV2 for CPU. Jun 12, 2019
matrix_diag_op_gpu.cu.cc [ROCm] Fix for the broken `--config=rocm` build Jun 12, 2019
matrix_exponential_op.cc Deprecate C++ kernel for matrix exponential, which is now implemented… Aug 25, 2018
matrix_inverse_op.cc Merge changes from github. Jan 4, 2018
matrix_logarithm_op.cc Branch 183429339 (