kse-01/tensorflow/tensorflow/core/kernels
github-classroom[bot] 1122cdd8b0
Initial commit
2023-10-09 11:37:31 +00:00
..
batching_util Initial commit 2023-10-09 11:37:31 +00:00
boosted_trees Initial commit 2023-10-09 11:37:31 +00:00
data Initial commit 2023-10-09 11:37:31 +00:00
fuzzing Initial commit 2023-10-09 11:37:31 +00:00
hexagon Initial commit 2023-10-09 11:37:31 +00:00
mlir_generated Initial commit 2023-10-09 11:37:31 +00:00
neon Initial commit 2023-10-09 11:37:31 +00:00
rnn Initial commit 2023-10-09 11:37:31 +00:00
sparse Initial commit 2023-10-09 11:37:31 +00:00
special_math Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_test_data Initial commit 2023-10-09 11:37:31 +00:00
tensor_forest Initial commit 2023-10-09 11:37:31 +00:00
adjust_contrast_op.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_contrast_op.h Initial commit 2023-10-09 11:37:31 +00:00
adjust_contrast_op_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_contrast_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_contrast_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_hsv_gpu.cu.h Initial commit 2023-10-09 11:37:31 +00:00
adjust_hue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_hue_op.h Initial commit 2023-10-09 11:37:31 +00:00
adjust_hue_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_saturation_op.cc Initial commit 2023-10-09 11:37:31 +00:00
adjust_saturation_op.h Initial commit 2023-10-09 11:37:31 +00:00
adjust_saturation_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
aggregate_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
aggregate_ops.h Initial commit 2023-10-09 11:37:31 +00:00
aggregate_ops_cpu.h Initial commit 2023-10-09 11:37:31 +00:00
aggregate_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
argmax_op.cc Initial commit 2023-10-09 11:37:31 +00:00
argmax_op.h Initial commit 2023-10-09 11:37:31 +00:00
argmax_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
as_string_op.cc Initial commit 2023-10-09 11:37:31 +00:00
assign_op.h Initial commit 2023-10-09 11:37:31 +00:00
attention_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
avgpooling_op.cc Initial commit 2023-10-09 11:37:31 +00:00
avgpooling_op.h Initial commit 2023-10-09 11:37:31 +00:00
avgpooling_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
banded_triangular_solve_op.cc Initial commit 2023-10-09 11:37:31 +00:00
banded_triangular_solve_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
barrier_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
base64_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
basic_ops_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_matmul_op_complex.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_matmul_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
batch_matmul_op_real.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_matmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_norm_op.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_norm_op.h Initial commit 2023-10-09 11:37:31 +00:00
batch_norm_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_norm_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
batch_util.h Initial commit 2023-10-09 11:37:31 +00:00
batchtospace_op.cc Initial commit 2023-10-09 11:37:31 +00:00
bcast_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
betainc_op.cc Initial commit 2023-10-09 11:37:31 +00:00
betainc_op.h Initial commit 2023-10-09 11:37:31 +00:00
betainc_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
bias_op.cc Initial commit 2023-10-09 11:37:31 +00:00
bias_op.h Initial commit 2023-10-09 11:37:31 +00:00
bias_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
bias_op_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
bias_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
bincount_op.cc Initial commit 2023-10-09 11:37:31 +00:00
bincount_op.h Initial commit 2023-10-09 11:37:31 +00:00
bincount_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
bincount_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
broadcast_to_op.cc Initial commit 2023-10-09 11:37:31 +00:00
broadcast_to_op.h Initial commit 2023-10-09 11:37:31 +00:00
broadcast_to_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
broadcast_to_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
bucketize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
bucketize_op.h Initial commit 2023-10-09 11:37:31 +00:00
bucketize_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
BUILD Initial commit 2023-10-09 11:37:31 +00:00
candidate_sampler_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op.h Initial commit 2023-10-09 11:37:31 +00:00
cast_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_bfloat.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_bool.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_double.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_float.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_half.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_int8.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_int16.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_int32.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_int64.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_uint8.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_uint16.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_uint32.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_impl_uint64.cc Initial commit 2023-10-09 11:37:31 +00:00
cast_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
check_numerics_op.cc Initial commit 2023-10-09 11:37:31 +00:00
check_numerics_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cholesky_grad.cc Initial commit 2023-10-09 11:37:31 +00:00
cholesky_op.cc Initial commit 2023-10-09 11:37:31 +00:00
clustering_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
clustering_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl.h Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_broadcaster.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_broadcaster.h Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_gatherer.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_gatherer.h Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_reducer.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_reducer.h Initial commit 2023-10-09 11:37:31 +00:00
collective_nccl_test.cc Initial commit 2023-10-09 11:37:31 +00:00
collective_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
colorspace_op.cc Initial commit 2023-10-09 11:37:31 +00:00
colorspace_op.h Initial commit 2023-10-09 11:37:31 +00:00
colorspace_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
colorspace_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
compare_and_bitpack_op.cc Initial commit 2023-10-09 11:37:31 +00:00
compare_and_bitpack_op.h Initial commit 2023-10-09 11:37:31 +00:00
compare_and_bitpack_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
concat_lib.h Initial commit 2023-10-09 11:37:31 +00:00
concat_lib_cpu.cc Initial commit 2023-10-09 11:37:31 +00:00
concat_lib_cpu.h Initial commit 2023-10-09 11:37:31 +00:00
concat_lib_gpu.cc Initial commit 2023-10-09 11:37:31 +00:00
concat_lib_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
concat_lib_gpu_impl.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
concat_op.cc Initial commit 2023-10-09 11:37:31 +00:00
concat_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator.h Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator_base.cc Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator_base.h Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator_base_op.cc Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator_base_op.h Initial commit 2023-10-09 11:37:31 +00:00
conditional_accumulator_op.cc Initial commit 2023-10-09 11:37:31 +00:00
constant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
constant_op.h Initial commit 2023-10-09 11:37:31 +00:00
constant_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
constant_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
control_flow_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
control_flow_ops.h Initial commit 2023-10-09 11:37:31 +00:00
control_flow_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d.h Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_half.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_int.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_int_spatial_convolution.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_int_spatial_convolution_backward.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_uint8.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_uint16.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_uint32.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_2d_gpu_uint64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_3d.h Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_filter_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_filter_ops_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_input_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_input_ops_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_ops.h Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_ops_3d.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_shape_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_grad_shape_utils.h Initial commit 2023-10-09 11:37:31 +00:00
conv_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops.h Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_3d.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_3d.h Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_fused_double.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_fused_float.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_fused_half.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_fused_image_transform.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_fused_impl.h Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
conv_ops_using_gemm.cc Initial commit 2023-10-09 11:37:31 +00:00
count_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
count_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
count_up_to_op.cc Initial commit 2023-10-09 11:37:31 +00:00
crop_and_resize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
crop_and_resize_op.h Initial commit 2023-10-09 11:37:31 +00:00
crop_and_resize_op_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
crop_and_resize_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
crop_and_resize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
cross_op.cc Initial commit 2023-10-09 11:37:31 +00:00
cross_op.h Initial commit 2023-10-09 11:37:31 +00:00
cross_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cross_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ctc_decoder_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
ctc_loss_op.cc Initial commit 2023-10-09 11:37:31 +00:00
cuda_solvers.cc Initial commit 2023-10-09 11:37:31 +00:00
cuda_solvers.h Initial commit 2023-10-09 11:37:31 +00:00
cuda_sparse.cc Initial commit 2023-10-09 11:37:31 +00:00
cuda_sparse.h Initial commit 2023-10-09 11:37:31 +00:00
cudnn_pooling_gpu.cc Initial commit 2023-10-09 11:37:31 +00:00
cudnn_pooling_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
cudnn_rnn_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_abs.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_acos.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_acosh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_add_1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_add_2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_arg.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_asin.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_asinh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_atan.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_atan2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_atanh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_bitwise_and.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_bitwise_or.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_bitwise_xor.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_ceil.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_clip.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_clip.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_clip_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_complex.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_conj.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_cos.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_cosh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_digamma.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_div.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_equal_to_1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_equal_to_2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_erf.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_erfc.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_exp.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_expm1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_floor.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_floor_div.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_floor_mod.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_abs.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_acos.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_acosh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_add.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_arg.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_asin.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_asinh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_atan.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_atan2.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_atanh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_bitwise_and.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_bitwise_or.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_bitwise_xor.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_ceil.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_complex.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_conj.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_cos.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_cosh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_digamma.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_div.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_equal_to.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_erf.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_erfc.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_exp.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_expm1.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_floor.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_floor_div.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_floor_mod.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_greater.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_greater_equal.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_igammas_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_igammas_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_imag.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_inverse.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_invert.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_isfinite.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_isinf.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_isnan.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_left_shift.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_less.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_less_equal.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_lgamma.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_log.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_log1p.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_logical_and.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_logical_not.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_logical_or.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_maximum.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_minimum.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_mod.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_mul.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_neg.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_not_equal_to.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_pow.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_random_grad.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_real.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_right_shift.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_rint.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_round.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_rsqrt.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_select.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sigmoid.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sign.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sin.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sinh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sqrt.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_square.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_squared_difference.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_sub.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_tan.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_tanh.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_xdivy.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_xlog1py.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_xlogy.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_gpu_zeta.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_greater.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_greater_equal.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_igammas.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_imag.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_invert.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_isfinite.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_isinf.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_isnan.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_left_shift.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_less.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_less_equal.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_lgamma.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_log.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_log1p.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_logical_and.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_logical_not.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_logical_or.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_maximum.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_minimum.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_mod.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_mul_1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_mul_2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_ndtri.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_ndtri_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_ndtri_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_neg_1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_neg_2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_not_equal_to_1.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_not_equal_to_2.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_pow.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_random_grad.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_real.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_reciprocal.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_right_shift.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_rint.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_round.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_rsqrt.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_select.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sigmoid.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sign.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sin.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sinh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sqrt.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_square.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_squared_difference.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_sub.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_tan.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_tanh.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_xdivy.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_xlog1py.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_xlogy.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_op_zeta.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_gpu_common.cu.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_gpu_gradients.cu.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_gradients.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_sycl_common.h Initial commit 2023-10-09 11:37:31 +00:00
cwise_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
data_format_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
data_format_ops.h Initial commit 2023-10-09 11:37:31 +00:00
data_format_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
debug_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
debug_ops.h Initial commit 2023-10-09 11:37:31 +00:00
debug_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
debug_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_compressed_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_csv_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_image_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_padded_raw_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_proto_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_raw_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_wav_op.cc Initial commit 2023-10-09 11:37:31 +00:00
decode_wav_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
deep_conv2d.cc Initial commit 2023-10-09 11:37:31 +00:00
deep_conv2d.h Initial commit 2023-10-09 11:37:31 +00:00
deep_conv2d_test.cc Initial commit 2023-10-09 11:37:31 +00:00
dense_update_functor.cc Initial commit 2023-10-09 11:37:31 +00:00
dense_update_functor.h Initial commit 2023-10-09 11:37:31 +00:00
dense_update_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
dense_update_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
depthtospace_op.cc Initial commit 2023-10-09 11:37:31 +00:00
depthtospace_op.h Initial commit 2023-10-09 11:37:31 +00:00
depthtospace_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_grad_op.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op.h Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_op_gpu_half.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
depthwise_conv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
dequantize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
dequantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
deserialize_sparse_string_op.cc Initial commit 2023-10-09 11:37:31 +00:00
deserialize_sparse_variant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
determinant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
determinant_op.h Initial commit 2023-10-09 11:37:31 +00:00
determinant_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
diag_op.cc Initial commit 2023-10-09 11:37:31 +00:00
diag_op.h Initial commit 2023-10-09 11:37:31 +00:00
diag_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
diag_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
dilation_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
dilation_ops.h Initial commit 2023-10-09 11:37:31 +00:00
dilation_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
draw_bounding_box_op.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_partition_op.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_partition_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_partition_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_stitch_op.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_stitch_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
dynamic_stitch_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
edit_distance_op.cc Initial commit 2023-10-09 11:37:31 +00:00
eig_op_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
eig_op_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
eig_op_double.cc Initial commit 2023-10-09 11:37:31 +00:00
eig_op_float.cc Initial commit 2023-10-09 11:37:31 +00:00
eig_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_activations.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_activations_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_attention.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_attention_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_backward_cuboid_convolutions.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_backward_cuboid_convolutions_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_backward_spatial_convolutions.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_backward_spatial_convolutions_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_benchmark.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_benchmark_cpu_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_contraction_kernel.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_contraction_kernel.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_convolution_helpers.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_cuboid_convolution.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_mkldnn_contraction_kernel_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_pooling.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_pooling_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_spatial_convolutions-inl.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_spatial_convolutions.h Initial commit 2023-10-09 11:37:31 +00:00
eigen_spatial_convolutions_test.cc Initial commit 2023-10-09 11:37:31 +00:00
eigen_volume_patch.h Initial commit 2023-10-09 11:37:31 +00:00
einsum_op.h Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_bfloat16.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_double.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_float.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_half.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_int32.cc Initial commit 2023-10-09 11:37:31 +00:00
einsum_op_impl_int64.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_jpeg_op.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_jpeg_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_png_op.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_proto_op.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_wav_op.cc Initial commit 2023-10-09 11:37:31 +00:00
encode_wav_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
example_parsing_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
example_parsing_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
extract_image_patches_op.cc Initial commit 2023-10-09 11:37:31 +00:00
extract_image_patches_op.h Initial commit 2023-10-09 11:37:31 +00:00
extract_image_patches_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
extract_jpeg_shape_op.cc Initial commit 2023-10-09 11:37:31 +00:00
extract_volume_patches_op.cc Initial commit 2023-10-09 11:37:31 +00:00
extract_volume_patches_op.h Initial commit 2023-10-09 11:37:31 +00:00
extract_volume_patches_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
eye_functor.h Initial commit 2023-10-09 11:37:31 +00:00
eye_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
fact_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fake_quant_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
fake_quant_ops_functor.h Initial commit 2023-10-09 11:37:31 +00:00
fake_quant_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
fake_quant_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
fft_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
fifo_queue.cc Initial commit 2023-10-09 11:37:31 +00:00
fifo_queue.h Initial commit 2023-10-09 11:37:31 +00:00
fifo_queue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fill_functor.cc Initial commit 2023-10-09 11:37:31 +00:00
fill_functor.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
fill_functor.h Initial commit 2023-10-09 11:37:31 +00:00
fingerprint_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fingerprint_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
fixed_length_record_reader_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fractional_avg_pool_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fractional_max_pool_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fractional_pool_common.cc Initial commit 2023-10-09 11:37:31 +00:00
fractional_pool_common.h Initial commit 2023-10-09 11:37:31 +00:00
function_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
function_ops.h Initial commit 2023-10-09 11:37:31 +00:00
functional_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_batch_norm_ex_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_batch_norm_op.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_batch_norm_op.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_batch_norm_op.h Initial commit 2023-10-09 11:37:31 +00:00
fused_batch_norm_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_eigen_output_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
fused_eigen_output_kernels.h Initial commit 2023-10-09 11:37:31 +00:00
gather_functor.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_functor.h Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_batched.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_batched.h Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_batched_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_batched_gpu.cu.h Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_functor_gpu.cu.h Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op.h Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_0.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_6.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_cpu_impl_7.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_nd_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_op.cc Initial commit 2023-10-09 11:37:31 +00:00
gather_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
gemm_functors.h Initial commit 2023-10-09 11:37:31 +00:00
generate_box_proposals_op.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
generate_vocab_remapping_op.cc Initial commit 2023-10-09 11:37:31 +00:00
gpu_device_array.h Initial commit 2023-10-09 11:37:31 +00:00
gpu_device_array_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
gpu_prim.h Initial commit 2023-10-09 11:37:31 +00:00
gpu_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
gpu_utils.h Initial commit 2023-10-09 11:37:31 +00:00
guarantee_const_op.cc Initial commit 2023-10-09 11:37:31 +00:00
guarantee_const_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
hinge-loss.h Initial commit 2023-10-09 11:37:31 +00:00
histogram_op.cc Initial commit 2023-10-09 11:37:31 +00:00
histogram_op.h Initial commit 2023-10-09 11:37:31 +00:00
histogram_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
host_constant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
host_constant_op.h Initial commit 2023-10-09 11:37:31 +00:00
i_remote_fused_graph_executor.h Initial commit 2023-10-09 11:37:31 +00:00
i_remote_fused_graph_ops_definitions.cc Initial commit 2023-10-09 11:37:31 +00:00
i_remote_fused_graph_ops_definitions.h Initial commit 2023-10-09 11:37:31 +00:00
identity_n_op.cc Initial commit 2023-10-09 11:37:31 +00:00
identity_n_op.h Initial commit 2023-10-09 11:37:31 +00:00
identity_n_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
identity_op.cc Initial commit 2023-10-09 11:37:31 +00:00
identity_op.h Initial commit 2023-10-09 11:37:31 +00:00
identity_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
identity_reader_op.cc Initial commit 2023-10-09 11:37:31 +00:00
image_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
image_ops.h Initial commit 2023-10-09 11:37:31 +00:00
image_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
image_resizer_state.h Initial commit 2023-10-09 11:37:31 +00:00
immutable_constant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
immutable_constant_op.h Initial commit 2023-10-09 11:37:31 +00:00
immutable_constant_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
in_topk_op.cc Initial commit 2023-10-09 11:37:31 +00:00
in_topk_op.h Initial commit 2023-10-09 11:37:31 +00:00
in_topk_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
in_topk_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
initializable_lookup_table.cc Initial commit 2023-10-09 11:37:31 +00:00
initializable_lookup_table.h Initial commit 2023-10-09 11:37:31 +00:00
inplace_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
inplace_ops_functor.h Initial commit 2023-10-09 11:37:31 +00:00
inplace_ops_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
kernel_platform_strings.h Initial commit 2023-10-09 11:37:31 +00:00
l2loss_op.cc Initial commit 2023-10-09 11:37:31 +00:00
l2loss_op.h Initial commit 2023-10-09 11:37:31 +00:00
l2loss_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
linalg_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
linalg_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
list_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
list_kernels.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
list_kernels.h Initial commit 2023-10-09 11:37:31 +00:00
listdiff_op.cc Initial commit 2023-10-09 11:37:31 +00:00
lmdb_reader_op.cc Initial commit 2023-10-09 11:37:31 +00:00
load_and_remap_matrix_op.cc Initial commit 2023-10-09 11:37:31 +00:00
logging_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
logging_ops.h Initial commit 2023-10-09 11:37:31 +00:00
logging_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
logistic-loss.h Initial commit 2023-10-09 11:37:31 +00:00
lookup_table_init_op.cc Initial commit 2023-10-09 11:37:31 +00:00
lookup_table_init_op.h Initial commit 2023-10-09 11:37:31 +00:00
lookup_table_op.cc Initial commit 2023-10-09 11:37:31 +00:00
lookup_table_op.h Initial commit 2023-10-09 11:37:31 +00:00
lookup_util.cc Initial commit 2023-10-09 11:37:31 +00:00
lookup_util.h Initial commit 2023-10-09 11:37:31 +00:00
loss.h Initial commit 2023-10-09 11:37:31 +00:00
loss_test.cc Initial commit 2023-10-09 11:37:31 +00:00
lrn_op.cc Initial commit 2023-10-09 11:37:31 +00:00
lrn_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
lu_op.cc Initial commit 2023-10-09 11:37:31 +00:00
lu_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
map_stage_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matching_files_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matmul_op.h Initial commit 2023-10-09 11:37:31 +00:00
matmul_op_fused.cc Initial commit 2023-10-09 11:37:31 +00:00
matmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_band_part_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_band_part_op.h Initial commit 2023-10-09 11:37:31 +00:00
matrix_band_part_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_diag_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_diag_op.h Initial commit 2023-10-09 11:37:31 +00:00
matrix_diag_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_exponential_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_inverse_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_logarithm_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_set_diag_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_set_diag_op.h Initial commit 2023-10-09 11:37:31 +00:00
matrix_set_diag_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_ls_op_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_ls_op_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_ls_op_double.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_ls_op_float.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_ls_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
matrix_solve_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_square_root_op.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_triangular_solve_op_complex.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_triangular_solve_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
matrix_triangular_solve_op_real.cc Initial commit 2023-10-09 11:37:31 +00:00
matrix_triangular_solve_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
maxpooling_op.cc Initial commit 2023-10-09 11:37:31 +00:00
maxpooling_op.h Initial commit 2023-10-09 11:37:31 +00:00
maxpooling_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
maxpooling_op_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
merge_v2_checkpoints_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
meta_support.cc Initial commit 2023-10-09 11:37:31 +00:00
meta_support.h Initial commit 2023-10-09 11:37:31 +00:00
mfcc.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc.h Initial commit 2023-10-09 11:37:31 +00:00
mfcc_dct.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_dct.h Initial commit 2023-10-09 11:37:31 +00:00
mfcc_dct_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_mel_filterbank.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_mel_filterbank.h Initial commit 2023-10-09 11:37:31 +00:00
mfcc_mel_filterbank_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mfcc_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op.h Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_cpu_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
mirror_pad_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_aggregate_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_avgpooling_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_batch_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_concat_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_conv_grad_filter_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_conv_grad_input_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_conv_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_conv_ops.h Initial commit 2023-10-09 11:37:31 +00:00
mkl_conv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_cwise_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_dequantize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_dequantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_fused_batch_norm_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_fused_batch_norm_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_fused_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_identity_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_input_conversion_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_lrn_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_matmul_op_fused.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_matmul_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
mkl_maxpooling_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_pooling_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_pooling_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
mkl_qmatmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_qmatmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantized_concat_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantized_conv_ops.h Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantized_conv_ops_perchannel_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantized_conv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_quantized_pooling_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_relu_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_relu_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_requantization_range_per_channel_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_requantize_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_requantize_per_channel_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_reshape_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_slice_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_softmax_op.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_tfconv_op.h Initial commit 2023-10-09 11:37:31 +00:00
mkl_tmp_bf16_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
mkl_transpose_op.cc Initial commit 2023-10-09 11:37:31 +00:00
multinomial_op.cc Initial commit 2023-10-09 11:37:31 +00:00
multinomial_op.h Initial commit 2023-10-09 11:37:31 +00:00
multinomial_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
multinomial_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
mutex_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
nccl_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
nextafter_op.cc Initial commit 2023-10-09 11:37:31 +00:00
nextafter_op.h Initial commit 2023-10-09 11:37:31 +00:00
nextafter_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
nn_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
no_op.cc Initial commit 2023-10-09 11:37:31 +00:00
no_op.h Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op.cc Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op.h Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op_gpu_test.cc Initial commit 2023-10-09 11:37:31 +00:00
non_max_suppression_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
nth_element_op.cc Initial commit 2023-10-09 11:37:31 +00:00
nth_element_op.h Initial commit 2023-10-09 11:37:31 +00:00
one_hot_op.cc Initial commit 2023-10-09 11:37:31 +00:00
one_hot_op.h Initial commit 2023-10-09 11:37:31 +00:00
one_hot_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
one_hot_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ops_testutil.cc Initial commit 2023-10-09 11:37:31 +00:00
ops_testutil.h Initial commit 2023-10-09 11:37:31 +00:00
ops_testutil_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ops_util.h Initial commit 2023-10-09 11:37:31 +00:00
ops_util_test.cc Initial commit 2023-10-09 11:37:31 +00:00
pack_op.cc Initial commit 2023-10-09 11:37:31 +00:00
pad_op.cc Initial commit 2023-10-09 11:37:31 +00:00
pad_op.h Initial commit 2023-10-09 11:37:31 +00:00
pad_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
padding_fifo_queue.cc Initial commit 2023-10-09 11:37:31 +00:00
padding_fifo_queue.h Initial commit 2023-10-09 11:37:31 +00:00
padding_fifo_queue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
parameterized_truncated_normal_op.cc Initial commit 2023-10-09 11:37:31 +00:00
parameterized_truncated_normal_op.h Initial commit 2023-10-09 11:37:31 +00:00
parameterized_truncated_normal_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
parameterized_truncated_normal_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
parse_tensor_op.cc Initial commit 2023-10-09 11:37:31 +00:00
parse_tensor_test.cc Initial commit 2023-10-09 11:37:31 +00:00
partitioned_function_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
partitioned_function_ops.h Initial commit 2023-10-09 11:37:31 +00:00
poisson-loss.h Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_3d.cc Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_3d.h Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_3d_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_3d_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_3d_sycl.h Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
pooling_ops_common_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
population_count_op.cc Initial commit 2023-10-09 11:37:31 +00:00
population_count_op.h Initial commit 2023-10-09 11:37:31 +00:00
population_count_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
priority_queue.cc Initial commit 2023-10-09 11:37:31 +00:00
priority_queue.h Initial commit 2023-10-09 11:37:31 +00:00
priority_queue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
qr_op_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
qr_op_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
qr_op_double.cc Initial commit 2023-10-09 11:37:31 +00:00
qr_op_float.cc Initial commit 2023-10-09 11:37:31 +00:00
qr_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
quantization_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
quantization_utils.h Initial commit 2023-10-09 11:37:31 +00:00
quantization_utils_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_and_dequantize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_and_dequantize_op.h Initial commit 2023-10-09 11:37:31 +00:00
quantize_and_dequantize_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_and_dequantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_down_and_shrink_range.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_down_and_shrink_range_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_activation_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_activation_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_add_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_add_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_batch_norm_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_batch_norm_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_bias_add_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_bias_add_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_concat_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_concat_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_conv_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_conv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_instance_norm.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_instance_norm_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_matmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_mul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_mul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_pooling_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_pooling_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_reshape_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_reshape_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_resize_bilinear_op.cc Initial commit 2023-10-09 11:37:31 +00:00
quantized_resize_bilinear_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
queue_base.cc Initial commit 2023-10-09 11:37:31 +00:00
queue_base.h Initial commit 2023-10-09 11:37:31 +00:00
queue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
queue_op.h Initial commit 2023-10-09 11:37:31 +00:00
queue_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_cross_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_gather_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_gather_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_range_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_range_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_from_variant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_from_variant_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_sparse_kernel.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_sparse_kernel_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_tensor_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_tensor_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_variant_op.cc Initial commit 2023-10-09 11:37:31 +00:00
ragged_tensor_to_variant_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
random_binomial_op.cc Initial commit 2023-10-09 11:37:31 +00:00
random_binomial_op.h Initial commit 2023-10-09 11:37:31 +00:00
random_binomial_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
random_crop_op.cc Initial commit 2023-10-09 11:37:31 +00:00
random_op.cc Initial commit 2023-10-09 11:37:31 +00:00
random_op.h Initial commit 2023-10-09 11:37:31 +00:00
random_op_cpu.h Initial commit 2023-10-09 11:37:31 +00:00
random_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
random_op_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
random_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
random_poisson_op.cc Initial commit 2023-10-09 11:37:31 +00:00
random_poisson_op.h Initial commit 2023-10-09 11:37:31 +00:00
random_poisson_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
random_shuffle_op.cc Initial commit 2023-10-09 11:37:31 +00:00
random_shuffle_queue_op.cc Initial commit 2023-10-09 11:37:31 +00:00
range_sampler.cc Initial commit 2023-10-09 11:37:31 +00:00
range_sampler.h Initial commit 2023-10-09 11:37:31 +00:00
range_sampler_test.cc Initial commit 2023-10-09 11:37:31 +00:00
reader_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
record_input_op.cc Initial commit 2023-10-09 11:37:31 +00:00
record_yielder.cc Initial commit 2023-10-09 11:37:31 +00:00
record_yielder.h Initial commit 2023-10-09 11:37:31 +00:00
reduce_join_op.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_gpu_kernels.cu.h Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops.h Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_all.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_any.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_common.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_common.h Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_common_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_euclidean.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_bool.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_complex64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_complex128.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_gpu_int.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_half_mean_sum.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_half_prod_max_min.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_max.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_mean.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_min.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_prod.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_sum.cc Initial commit 2023-10-09 11:37:31 +00:00
reduction_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
redux_functor.h Initial commit 2023-10-09 11:37:31 +00:00
reference_gemm.h Initial commit 2023-10-09 11:37:31 +00:00
regex_full_match_op.cc Initial commit 2023-10-09 11:37:31 +00:00
regex_replace_op.cc Initial commit 2023-10-09 11:37:31 +00:00
regex_replace_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
relu_op.cc Initial commit 2023-10-09 11:37:31 +00:00
relu_op.h Initial commit 2023-10-09 11:37:31 +00:00
relu_op_functor.h Initial commit 2023-10-09 11:37:31 +00:00
relu_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_op.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_op_test_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_op_test_utils.h Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_utils.h Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_execute_utils_test.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_rewriter_transform.cc Initial commit 2023-10-09 11:37:31 +00:00
remote_fused_graph_rewriter_transform_test.cc Initial commit 2023-10-09 11:37:31 +00:00
requantization_range_op.cc Initial commit 2023-10-09 11:37:31 +00:00
requantization_range_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
requantize.cc Initial commit 2023-10-09 11:37:31 +00:00
requantize_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
reshape_op.cc Initial commit 2023-10-09 11:37:31 +00:00
reshape_op.h Initial commit 2023-10-09 11:37:31 +00:00
reshape_util.cc Initial commit 2023-10-09 11:37:31 +00:00
reshape_util.h Initial commit 2023-10-09 11:37:31 +00:00
resize_area_op.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_area_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_bicubic_op.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_bicubic_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_bilinear_op.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_bilinear_op.h Initial commit 2023-10-09 11:37:31 +00:00
resize_bilinear_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_bilinear_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_nearest_neighbor_op.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_nearest_neighbor_op.h Initial commit 2023-10-09 11:37:31 +00:00
resize_nearest_neighbor_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_nearest_neighbor_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
resize_op_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
resource_variable_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
resource_variable_ops.h Initial commit 2023-10-09 11:37:31 +00:00
restore_op.cc Initial commit 2023-10-09 11:37:31 +00:00
restore_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
restore_v2_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
reverse_op.cc Initial commit 2023-10-09 11:37:31 +00:00
reverse_op.h Initial commit 2023-10-09 11:37:31 +00:00
reverse_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
reverse_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
reverse_sequence_op.cc Initial commit 2023-10-09 11:37:31 +00:00
reverse_sequence_op.h Initial commit 2023-10-09 11:37:31 +00:00
reverse_sequence_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
rocm_solvers.cc Initial commit 2023-10-09 11:37:31 +00:00
rocm_solvers.h Initial commit 2023-10-09 11:37:31 +00:00
rocm_sparse.cc Initial commit 2023-10-09 11:37:31 +00:00
roll_op.cc Initial commit 2023-10-09 11:37:31 +00:00
roll_op.h Initial commit 2023-10-09 11:37:31 +00:00
roll_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
roll_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
rpc_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sample_distorted_bounding_box_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sampling_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
sampling_kernels.h Initial commit 2023-10-09 11:37:31 +00:00
sampling_kernels_test.cc Initial commit 2023-10-09 11:37:31 +00:00
save_op.cc Initial commit 2023-10-09 11:37:31 +00:00
save_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
save_restore_tensor.cc Initial commit 2023-10-09 11:37:31 +00:00
save_restore_tensor.h Initial commit 2023-10-09 11:37:31 +00:00
save_restore_v2_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
save_v2_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
scale_and_translate_op.cc Initial commit 2023-10-09 11:37:31 +00:00
scale_and_translate_op.h Initial commit 2023-10-09 11:37:31 +00:00
scale_and_translate_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops.h Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_gpu_half.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_gpu_int.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scan_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_functor.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_functor.h Initial commit 2023-10-09 11:37:31 +00:00
scatter_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_functor_gpu.cu.h Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op.h Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_0.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_6.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_cpu_impl_7.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_nd_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_op.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
scatter_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
scoped_allocator_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
scoped_allocator_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sdca_internal.cc Initial commit 2023-10-09 11:37:31 +00:00
sdca_internal.h Initial commit 2023-10-09 11:37:31 +00:00
sdca_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
sdca_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
searchsorted_op.cc Initial commit 2023-10-09 11:37:31 +00:00
searchsorted_op.h Initial commit 2023-10-09 11:37:31 +00:00
searchsorted_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops.h Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl.h Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
segment_reduction_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_op.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_double.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_float.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_gpu.cc Initial commit 2023-10-09 11:37:31 +00:00
self_adjoint_eig_v2_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
sendrecv_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
sendrecv_ops.h Initial commit 2023-10-09 11:37:31 +00:00
sendrecv_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sequence_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
sequence_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
serialize_sparse_op.cc Initial commit 2023-10-09 11:37:31 +00:00
session_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
set_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
shape_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
shape_ops.h Initial commit 2023-10-09 11:37:31 +00:00
shape_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op.h Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_6.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_7.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_cpu_impl_8.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
slice_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
smooth-hinge-loss.h Initial commit 2023-10-09 11:37:31 +00:00
snapshot_op.cc Initial commit 2023-10-09 11:37:31 +00:00
snapshot_op.h Initial commit 2023-10-09 11:37:31 +00:00
snapshot_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
sobol_op.cc Initial commit 2023-10-09 11:37:31 +00:00
softmax_op.cc Initial commit 2023-10-09 11:37:31 +00:00
softmax_op_functor.h Initial commit 2023-10-09 11:37:31 +00:00
softmax_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
softplus_op.cc Initial commit 2023-10-09 11:37:31 +00:00
softplus_op.h Initial commit 2023-10-09 11:37:31 +00:00
softplus_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
softsign_op.cc Initial commit 2023-10-09 11:37:31 +00:00
softsign_op.h Initial commit 2023-10-09 11:37:31 +00:00
softsign_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetobatch_benchmark_test.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetobatch_functor.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetobatch_functor.h Initial commit 2023-10-09 11:37:31 +00:00
spacetobatch_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetobatch_op.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetodepth_op.cc Initial commit 2023-10-09 11:37:31 +00:00
spacetodepth_op.h Initial commit 2023-10-09 11:37:31 +00:00
spacetodepth_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_add_grad_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_add_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_add_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_concat_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_conditional_accumulator.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_conditional_accumulator_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_cross_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_dense_binary_op_shared.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_dense_binary_op_shared_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_fill_empty_rows_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_matmul_op.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_matmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_reduce_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_reduce_sum_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_reorder_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_reshape_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_slice_grad_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_slice_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_softmax_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_sparse_binary_op_shared.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_split_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_add_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_add_op.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_matmul_op.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_matmul_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensor_dense_matmul_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_tensors_map_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_to_dense_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_to_dense_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_utils.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_utils_test.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_xent_op.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_xent_op.h Initial commit 2023-10-09 11:37:31 +00:00
sparse_xent_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
sparse_xent_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram.h Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_convert_test_data.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_op.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_test.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_test_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
spectrogram_test_utils.h Initial commit 2023-10-09 11:37:31 +00:00
split_lib.h Initial commit 2023-10-09 11:37:31 +00:00
split_lib_cpu.cc Initial commit 2023-10-09 11:37:31 +00:00
split_lib_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
split_lib_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
split_op.cc Initial commit 2023-10-09 11:37:31 +00:00
split_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
split_v_op.cc Initial commit 2023-10-09 11:37:31 +00:00
split_v_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
squared-loss.h Initial commit 2023-10-09 11:37:31 +00:00
stack.cc Initial commit 2023-10-09 11:37:31 +00:00
stack.h Initial commit 2023-10-09 11:37:31 +00:00
stack_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
stage_op.cc Initial commit 2023-10-09 11:37:31 +00:00
stateful_random_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
stateful_random_ops.h Initial commit 2023-10-09 11:37:31 +00:00
stateful_random_ops_cpu_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
stateful_random_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
stateless_random_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
stateless_random_ops.h Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op.h Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_define_grad.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_gpu_bool.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_gpu_complex.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_gpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_gpu_int.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_gpu_number_types.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_0.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_1.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_2.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_3.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_4.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_5.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_6.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_7.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_inst_8.cc Initial commit 2023-10-09 11:37:31 +00:00
strided_slice_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
string_format_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_format_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
string_join_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_length_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_lower_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_ngrams_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_ngrams_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
string_split_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_split_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
string_strip_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_to_hash_bucket_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_to_hash_bucket_op.h Initial commit 2023-10-09 11:37:31 +00:00
string_to_number_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_upper_op.cc Initial commit 2023-10-09 11:37:31 +00:00
string_util.cc Initial commit 2023-10-09 11:37:31 +00:00
string_util.h Initial commit 2023-10-09 11:37:31 +00:00
substr_op.cc Initial commit 2023-10-09 11:37:31 +00:00
substr_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_audio_op.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_audio_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_image_op.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_image_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_interface.h Initial commit 2023-10-09 11:37:31 +00:00
summary_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_op.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_tensor_op.cc Initial commit 2023-10-09 11:37:31 +00:00
summary_tensor_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_double.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_float.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
svd_op_impl.h Initial commit 2023-10-09 11:37:31 +00:00
tensor_array.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_array.h Initial commit 2023-10-09 11:37:31 +00:00
tensor_array_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_cord.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_cord.h Initial commit 2023-10-09 11:37:31 +00:00
tensor_cord_test.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_flag_utils.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_flag_utils.h Initial commit 2023-10-09 11:37:31 +00:00
tensor_flag_utils_test.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_list.cc Initial commit 2023-10-09 11:37:31 +00:00
tensor_list.h Initial commit 2023-10-09 11:37:31 +00:00
text_line_reader_op.cc Initial commit 2023-10-09 11:37:31 +00:00
tf_record_reader_op.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor.h Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu.h Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_bfloat16.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_bool.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_complex64.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_complex128.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_double.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_float.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_half.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_int8.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_int16.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_int32.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_int64.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_tstring.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_uint8.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_uint32.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_uint64.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_cpu_variant.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_bool.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_complex64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_complex128.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_half.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_int16.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_int32.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_gpu_int64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_functor_sycl.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_1.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_2.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_3.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_4.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_5.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_6.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_cpu_impl_7.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl.h Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_1.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_2.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_3.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_4.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_5.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_6.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_7.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_gpu_impl_8.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tile_ops_impl.h Initial commit 2023-10-09 11:37:31 +00:00
topk_op.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op.h Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu.h Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_double.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_float.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_half.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_int8.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_int16.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_int32.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_int64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_uint8.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_uint16.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_uint32.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
topk_op_gpu_uint64.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
training_op_helpers.cc Initial commit 2023-10-09 11:37:31 +00:00
training_op_helpers.h Initial commit 2023-10-09 11:37:31 +00:00
training_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
training_ops.h Initial commit 2023-10-09 11:37:31 +00:00
training_ops_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
training_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
transpose_functor.h Initial commit 2023-10-09 11:37:31 +00:00
transpose_functor_cpu.cc Initial commit 2023-10-09 11:37:31 +00:00
transpose_functor_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
transpose_op.cc Initial commit 2023-10-09 11:37:31 +00:00
transpose_op.h Initial commit 2023-10-09 11:37:31 +00:00
transpose_util_test.cc Initial commit 2023-10-09 11:37:31 +00:00
tridiagonal_matmul_op.cc Initial commit 2023-10-09 11:37:31 +00:00
tridiagonal_matmul_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
tridiagonal_solve_op.cc Initial commit 2023-10-09 11:37:31 +00:00
tridiagonal_solve_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
typed_conditional_accumulator_base.h Initial commit 2023-10-09 11:37:31 +00:00
typed_queue.h Initial commit 2023-10-09 11:37:31 +00:00
unary_ops_composition.cc Initial commit 2023-10-09 11:37:31 +00:00
unary_ops_composition_test.cc Initial commit 2023-10-09 11:37:31 +00:00
unicode_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
unicode_script_op.cc Initial commit 2023-10-09 11:37:31 +00:00
unique_op.cc Initial commit 2023-10-09 11:37:31 +00:00
unique_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
unpack_op.cc Initial commit 2023-10-09 11:37:31 +00:00
unravel_index_op.cc Initial commit 2023-10-09 11:37:31 +00:00
unsorted_segment_join_op.cc Initial commit 2023-10-09 11:37:31 +00:00
variable_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
variable_ops.h Initial commit 2023-10-09 11:37:31 +00:00
variable_ops_test.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op.h Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu.cu.h Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_1.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_2.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_3.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_4.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_5.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_6.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_7.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
where_op_gpu_impl_8.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
whole_file_read_ops.cc Initial commit 2023-10-09 11:37:31 +00:00
winograd_transform.h Initial commit 2023-10-09 11:37:31 +00:00
word2vec_kernels.cc Initial commit 2023-10-09 11:37:31 +00:00
xent_op.cc Initial commit 2023-10-09 11:37:31 +00:00
xent_op.h Initial commit 2023-10-09 11:37:31 +00:00
xent_op_gpu.cu.cc Initial commit 2023-10-09 11:37:31 +00:00
xent_op_test.cc Initial commit 2023-10-09 11:37:31 +00:00
xsmm_conv2d.cc Initial commit 2023-10-09 11:37:31 +00:00
xsmm_conv2d.h Initial commit 2023-10-09 11:37:31 +00:00
xsmm_conv2d_test.cc Initial commit 2023-10-09 11:37:31 +00:00