history | zip | gz
lin
2025-08-14 dae8bad597b6607a449b32bf76c523423f7720ed
.. drwxr-xr-x
batching_util drwxr-xr-x tree | history | zip | gz
boosted_trees drwxr-xr-x tree | history | zip | gz
data drwxr-xr-x tree | history | zip | gz
fuzzing drwxr-xr-x tree | history | zip | gz
hexagon drwxr-xr-x tree | history | zip | gz
lookup_tables drwxr-xr-x tree | history | zip | gz
neon drwxr-xr-x tree | history | zip | gz
spectrogram_test_data drwxr-xr-x tree | history | zip | gz
tensor_forest drwxr-xr-x tree | history | zip | gz
BUILD 176 KB -rwxr-xr-x view | raw | blame | history
adjust_contrast_op.cc 19 KB -rwxr-xr-x view | raw | blame | history
adjust_contrast_op.h 6 KB -rwxr-xr-x view | raw | blame | history
adjust_contrast_op_benchmark_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
adjust_contrast_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
adjust_contrast_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
adjust_hsv_gpu.cu.h 5 KB -rwxr-xr-x view | raw | blame | history
adjust_hue_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
adjust_hue_op.h 1 KB -rwxr-xr-x view | raw | blame | history
adjust_hue_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
adjust_saturation_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
adjust_saturation_op.h 1 KB -rwxr-xr-x view | raw | blame | history
adjust_saturation_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
aggregate_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
aggregate_ops.h 9 KB -rwxr-xr-x view | raw | blame | history
aggregate_ops_cpu.h 10 KB -rwxr-xr-x view | raw | blame | history
aggregate_ops_gpu.cu.cc 6 KB -rwxr-xr-x view | raw | blame | history
argmax_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
argmax_op.h 2 KB -rwxr-xr-x view | raw | blame | history
argmax_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
as_string_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
assign_op.h 6 KB -rwxr-xr-x view | raw | blame | history
attention_ops.cc 6 KB -rwxr-xr-x view | raw | blame | history
avgpooling_op.cc 26 KB -rwxr-xr-x view | raw | blame | history
avgpooling_op.h 3 KB -rwxr-xr-x view | raw | blame | history
avgpooling_op_gpu.cu.cc 5 KB -rwxr-xr-x view | raw | blame | history
barrier_ops.cc 26 KB -rwxr-xr-x view | raw | blame | history
base64_ops.cc 2 KB -rwxr-xr-x view | raw | blame | history
basic_ops_benchmark_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
batch_kernels.cc 48 KB -rwxr-xr-x view | raw | blame | history
batch_matmul_op_complex.cc 1 KB -rwxr-xr-x view | raw | blame | history
batch_matmul_op_impl.h 25 KB -rwxr-xr-x view | raw | blame | history
batch_matmul_op_real.cc 2 KB -rwxr-xr-x view | raw | blame | history
batch_matmul_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
batch_norm_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
batch_norm_op.h 6 KB -rwxr-xr-x view | raw | blame | history
batch_norm_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
batch_norm_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
batch_util.h 955 b -rwxr-xr-x view | raw | blame | history
batchtospace_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
bcast_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
betainc_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
betainc_op.h 2 KB -rwxr-xr-x view | raw | blame | history
betainc_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
bias_op.cc 24 KB -rwxr-xr-x view | raw | blame | history
bias_op.h 2 KB -rwxr-xr-x view | raw | blame | history
bias_op_gpu.cu.cc 11 KB -rwxr-xr-x view | raw | blame | history
bias_op_gpu.h 3 KB -rwxr-xr-x view | raw | blame | history
bias_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
bincount_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
bincount_op.h 1 KB -rwxr-xr-x view | raw | blame | history
bincount_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
bincount_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
broadcast_to_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
broadcast_to_op.h 4 KB -rwxr-xr-x view | raw | blame | history
broadcast_to_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
bucketize_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
bucketize_op.h 1 KB -rwxr-xr-x view | raw | blame | history
bucketize_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
candidate_sampler_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
cast_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
cast_op.h 12 KB -rwxr-xr-x view | raw | blame | history
cast_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl.h 5 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_bfloat.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_bool.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_complex128.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_complex64.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_double.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_float.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_half.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_int16.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_int32.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_int64.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_int8.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_uint16.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_uint32.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_uint64.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_impl_uint8.cc 1 KB -rwxr-xr-x view | raw | blame | history
cast_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
check_numerics_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
check_numerics_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
cholesky_grad.cc 6 KB -rwxr-xr-x view | raw | blame | history
cholesky_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
clustering_ops.cc 26 KB -rwxr-xr-x view | raw | blame | history
clustering_ops_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
collective_nccl_reducer.cc 9 KB -rwxr-xr-x view | raw | blame | history
collective_nccl_reducer.h 2 KB -rwxr-xr-x view | raw | blame | history
collective_nccl_reducer_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
collective_ops.cc 15 KB -rwxr-xr-x view | raw | blame | history
colorspace_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
colorspace_op.h 3 KB -rwxr-xr-x view | raw | blame | history
colorspace_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
colorspace_op_test.cc 9 KB -rwxr-xr-x view | raw | blame | history
compare_and_bitpack_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
compare_and_bitpack_op.h 1 KB -rwxr-xr-x view | raw | blame | history
compare_and_bitpack_op_gpu.cu.cc 6 KB -rwxr-xr-x view | raw | blame | history
concat_lib.h 3 KB -rwxr-xr-x view | raw | blame | history
concat_lib_cpu.cc 4 KB -rwxr-xr-x view | raw | blame | history
concat_lib_cpu.h 5 KB -rwxr-xr-x view | raw | blame | history
concat_lib_gpu.cc 4 KB -rwxr-xr-x view | raw | blame | history
concat_lib_gpu.h 4 KB -rwxr-xr-x view | raw | blame | history
concat_lib_gpu_impl.cu.cc 10 KB -rwxr-xr-x view | raw | blame | history
concat_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
concat_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator.h 6 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator_base.cc 7 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator_base.h 8 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator_base_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator_base_op.h 8 KB -rwxr-xr-x view | raw | blame | history
conditional_accumulator_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
constant_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
constant_op.h 2 KB -rwxr-xr-x view | raw | blame | history
constant_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
constant_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
control_flow_ops.cc 27 KB -rwxr-xr-x view | raw | blame | history
control_flow_ops.h 4 KB -rwxr-xr-x view | raw | blame | history
control_flow_ops_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
conv_2d.h 13 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu.h 41 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_double.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_float.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_half.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_int.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_uint16.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_uint32.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_uint64.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_2d_gpu_uint8.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_3d.h 4 KB -rwxr-xr-x view | raw | blame | history
conv_grad_filter_ops.cc 44 KB -rwxr-xr-x view | raw | blame | history
conv_grad_input_ops.cc 49 KB -rwxr-xr-x view | raw | blame | history
conv_grad_ops.cc 8 KB -rwxr-xr-x view | raw | blame | history
conv_grad_ops.h 9 KB -rwxr-xr-x view | raw | blame | history
conv_grad_ops_3d.cc 84 KB -rwxr-xr-x view | raw | blame | history
conv_ops.cc 42 KB -rwxr-xr-x view | raw | blame | history
conv_ops.h 4 KB -rwxr-xr-x view | raw | blame | history
conv_ops_3d.cc 24 KB -rwxr-xr-x view | raw | blame | history
conv_ops_fused_double.cc 2 KB -rwxr-xr-x view | raw | blame | history
conv_ops_fused_float.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_ops_fused_half.cc 1 KB -rwxr-xr-x view | raw | blame | history
conv_ops_fused_image_transform.cc 41 KB -rwxr-xr-x view | raw | blame | history
conv_ops_fused_impl.h 40 KB -rwxr-xr-x view | raw | blame | history
conv_ops_gpu.h 7 KB -rwxr-xr-x view | raw | blame | history
conv_ops_gpu_2.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
conv_ops_test.cc 65 KB -rwxr-xr-x view | raw | blame | history
conv_ops_using_gemm.cc 28 KB -rwxr-xr-x view | raw | blame | history
count_up_to_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
crop_and_resize_op.cc 34 KB -rwxr-xr-x view | raw | blame | history
crop_and_resize_op.h 3 KB -rwxr-xr-x view | raw | blame | history
crop_and_resize_op_benchmark_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
crop_and_resize_op_gpu.cu.cc 18 KB -rwxr-xr-x view | raw | blame | history
crop_and_resize_op_test.cc 16 KB -rwxr-xr-x view | raw | blame | history
cross_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
cross_op.h 2 KB -rwxr-xr-x view | raw | blame | history
cross_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cross_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
ctc_decoder_ops.cc 12 KB -rwxr-xr-x view | raw | blame | history
ctc_loss_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
cuda_solvers.cc 43 KB -rwxr-xr-x view | raw | blame | history
cuda_solvers.h 19 KB -rwxr-xr-x view | raw | blame | history
cudnn_pooling_gpu.cc 10 KB -rwxr-xr-x view | raw | blame | history
cudnn_pooling_gpu.h 2 KB -rwxr-xr-x view | raw | blame | history
cudnn_rnn_ops.cc 75 KB -rwxr-xr-x view | raw | blame | history
cwise_op_abs.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_acos.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_acosh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_add_1.cc 3 KB -rwxr-xr-x view | raw | blame | history
cwise_op_add_2.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_arg.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_asin.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_asinh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_atan.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_atan2.cc 930 b -rwxr-xr-x view | raw | blame | history
cwise_op_atanh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_bessel.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_bessel.cu.cc 977 b -rwxr-xr-x view | raw | blame | history
cwise_op_bitwise_and.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_bitwise_or.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_bitwise_xor.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_ceil.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_clip.cc 8 KB -rwxr-xr-x view | raw | blame | history
cwise_op_clip.h 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_clip_gpu.cu.cc 5 KB -rwxr-xr-x view | raw | blame | history
cwise_op_complex.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_conj.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_cos.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_cosh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_digamma.cc 998 b -rwxr-xr-x view | raw | blame | history
cwise_op_div.cc 3 KB -rwxr-xr-x view | raw | blame | history
cwise_op_equal_to_1.cc 3 KB -rwxr-xr-x view | raw | blame | history
cwise_op_equal_to_2.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_erf.cc 962 b -rwxr-xr-x view | raw | blame | history
cwise_op_erfc.cc 966 b -rwxr-xr-x view | raw | blame | history
cwise_op_exp.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_expm1.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_floor.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_floor_div.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_floor_mod.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_abs.cu.cc 945 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_acos.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_acosh.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_add.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_arg.cu.cc 916 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_asin.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_asinh.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_atan.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_atan2.cu.cc 905 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_atanh.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_bitwise_and.cu.cc 969 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_bitwise_or.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_bitwise_xor.cu.cc 969 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_ceil.cu.cc 916 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_complex.cu.cc 912 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_conj.cu.cc 932 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_cos.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_cosh.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_digamma.cu.cc 919 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_div.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_equal_to.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_erf.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_erfc.cu.cc 916 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_exp.cu.cc 938 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_expm1.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_floor.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_floor_div.cu.cc 990 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_floor_mod.cu.cc 911 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_greater.cu.cc 947 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_greater_equal.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_igammas_double.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_igammas_float.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_imag.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_inverse.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_invert.cu.cc 948 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_isfinite.cu.cc 920 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_isinf.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_isnan.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_left_shift.cu.cc 968 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_less.cu.cc 944 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_less_equal.cu.cc 965 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_lgamma.cu.cc 918 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_log.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_log1p.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_logical_and.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_logical_not.cu.cc 921 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_logical_or.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_maximum.cu.cc 927 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_minimum.cu.cc 927 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_mod.cu.cc 911 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_mul.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_neg.cu.cc 966 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_not_equal_to.cu.cc 998 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_pow.cu.cc 923 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_random_grad.cu.cc 917 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_real.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_right_shift.cu.cc 969 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_rint.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_round.cu.cc 931 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_rsqrt.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_select.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sigmoid.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sign.cu.cc 946 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sin.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sinh.cu.cc 903 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sqrt.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_square.cu.cc 925 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_squared_difference.cu.cc 938 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_sub.cu.cc 946 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_tan.cu.cc 915 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_tanh.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_xdivy.cu.cc 941 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_xlogy.cu.cc 941 b -rwxr-xr-x view | raw | blame | history
cwise_op_gpu_zeta.cu.cc 946 b -rwxr-xr-x view | raw | blame | history
cwise_op_greater.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_greater_equal.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_igammas.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_imag.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_invert.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_isfinite.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_isinf.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_isnan.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_left_shift.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_less.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_less_equal.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_lgamma.cc 959 b -rwxr-xr-x view | raw | blame | history
cwise_op_log.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_log1p.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_logical_and.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_logical_not.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_logical_or.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_maximum.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_minimum.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_mod.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_mul_1.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_mul_2.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_neg.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_not_equal_to_1.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_not_equal_to_2.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_pow.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_random_grad.cc 994 b -rwxr-xr-x view | raw | blame | history
cwise_op_real.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_reciprocal.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_right_shift.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_rint.cc 912 b -rwxr-xr-x view | raw | blame | history
cwise_op_round.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_rsqrt.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_select.cc 12 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sigmoid.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sign.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sin.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sinh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sqrt.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_square.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_squared_difference.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_sub.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_tan.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_tanh.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_xdivy.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_op_xlogy.cc 2 KB -rwxr-xr-x view | raw | blame | history
cwise_op_zeta.cc 1 KB -rwxr-xr-x view | raw | blame | history
cwise_ops.h 38 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_common.cc 3 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_common.h 24 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_gpu_common.cu.h 8 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_gpu_gradients.cu.h 3 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_gradients.h 8 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_sycl_common.h 7 KB -rwxr-xr-x view | raw | blame | history
cwise_ops_test.cc 15 KB -rwxr-xr-x view | raw | blame | history
data_format_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
data_format_ops.h 3 KB -rwxr-xr-x view | raw | blame | history
data_format_ops_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
debug_ops.cc 6 KB -rwxr-xr-x view | raw | blame | history
debug_ops.h 14 KB -rwxr-xr-x view | raw | blame | history
debug_ops_test.cc 30 KB -rwxr-xr-x view | raw | blame | history
decode_bmp_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
decode_compressed_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
decode_csv_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
decode_image_op.cc 14 KB -rwxr-xr-x view | raw | blame | history
decode_proto_op.cc 44 KB -rwxr-xr-x view | raw | blame | history
decode_raw_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
decode_wav_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
decode_wav_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
deep_conv2d.cc 48 KB -rwxr-xr-x view | raw | blame | history
deep_conv2d.h 3 KB -rwxr-xr-x view | raw | blame | history
deep_conv2d_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
dense_update_functor.cc 6 KB -rwxr-xr-x view | raw | blame | history
dense_update_functor.h 3 KB -rwxr-xr-x view | raw | blame | history
dense_update_functor_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
dense_update_ops.cc 6 KB -rwxr-xr-x view | raw | blame | history
depthtospace_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
depthtospace_op.h 3 KB -rwxr-xr-x view | raw | blame | history
depthtospace_op_gpu.cu.cc 10 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_grad_op.cc 56 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op.cc 21 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op.h 11 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op_gpu.h 78 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op_gpu_double.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op_gpu_float.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_op_gpu_half.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
depthwise_conv_ops_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
dequantize_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
dequantize_op_test.cc 7 KB -rwxr-xr-x view | raw | blame | history
deserialize_sparse_string_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
deserialize_sparse_variant_op.cc 17 KB -rwxr-xr-x view | raw | blame | history
determinant_op.cc 18 KB -rwxr-xr-x view | raw | blame | history
determinant_op.h 2 KB -rwxr-xr-x view | raw | blame | history
determinant_op_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
diag_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
diag_op.h 1 KB -rwxr-xr-x view | raw | blame | history
diag_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
diag_op_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
dilation_ops.cc 20 KB -rwxr-xr-x view | raw | blame | history
dilation_ops.h 3 KB -rwxr-xr-x view | raw | blame | history
dilation_ops_gpu.cu.cc 12 KB -rwxr-xr-x view | raw | blame | history
draw_bounding_box_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
dynamic_partition_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
dynamic_partition_op_gpu.cu.cc 19 KB -rwxr-xr-x view | raw | blame | history
dynamic_partition_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
dynamic_stitch_op.cc 16 KB -rwxr-xr-x view | raw | blame | history
dynamic_stitch_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
dynamic_stitch_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
edit_distance_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
eigen_activations.h 4 KB -rwxr-xr-x view | raw | blame | history
eigen_activations_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
eigen_attention.h 11 KB -rwxr-xr-x view | raw | blame | history
eigen_attention_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
eigen_backward_cuboid_convolutions.h 27 KB -rwxr-xr-x view | raw | blame | history
eigen_backward_spatial_convolutions.h 21 KB -rwxr-xr-x view | raw | blame | history
eigen_backward_spatial_convolutions_test.cc 82 KB -rwxr-xr-x view | raw | blame | history
eigen_benchmark.h 12 KB -rwxr-xr-x view | raw | blame | history
eigen_benchmark_cpu_test.cc 21 KB -rwxr-xr-x view | raw | blame | history
eigen_contraction_kernel.cc 2 KB -rwxr-xr-x view | raw | blame | history
eigen_contraction_kernel.h 12 KB -rwxr-xr-x view | raw | blame | history
eigen_cuboid_convolution.h 73 KB -rwxr-xr-x view | raw | blame | history
eigen_mkldnn_contraction_kernel_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
eigen_pooling.h 24 KB -rwxr-xr-x view | raw | blame | history
eigen_pooling_test.cc 26 KB -rwxr-xr-x view | raw | blame | history
eigen_softmax.h 3 KB -rwxr-xr-x view | raw | blame | history
eigen_softmax_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
eigen_spatial_convolutions-inl.h 61 KB -rwxr-xr-x view | raw | blame | history
eigen_spatial_convolutions.h 9 KB -rwxr-xr-x view | raw | blame | history
eigen_spatial_convolutions_test.cc 65 KB -rwxr-xr-x view | raw | blame | history
eigen_volume_patch.h 26 KB -rwxr-xr-x view | raw | blame | history
encode_jpeg_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
encode_png_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
encode_proto_op.cc 26 KB -rwxr-xr-x view | raw | blame | history
encode_wav_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
encode_wav_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
example_parsing_ops.cc 39 KB -rwxr-xr-x view | raw | blame | history
example_parsing_ops_test.cc 12 KB -rwxr-xr-x view | raw | blame | history
extract_image_patches_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
extract_image_patches_op.h 2 KB -rwxr-xr-x view | raw | blame | history
extract_image_patches_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
extract_jpeg_shape_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
extract_volume_patches_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
extract_volume_patches_op.h 2 KB -rwxr-xr-x view | raw | blame | history
extract_volume_patches_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
eye_functor.h 1 KB -rwxr-xr-x view | raw | blame | history
eye_functor_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
fact_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
fake_quant_ops.cc 18 KB -rwxr-xr-x view | raw | blame | history
fake_quant_ops_functor.h 11 KB -rwxr-xr-x view | raw | blame | history
fake_quant_ops_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
fake_quant_ops_test.cc 111 KB -rwxr-xr-x view | raw | blame | history
fft_ops.cc 22 KB -rwxr-xr-x view | raw | blame | history
fifo_queue.cc 15 KB -rwxr-xr-x view | raw | blame | history
fifo_queue.h 3 KB -rwxr-xr-x view | raw | blame | history
fifo_queue_op.cc 974 b -rwxr-xr-x view | raw | blame | history
fill_functor.cc 5 KB -rwxr-xr-x view | raw | blame | history
fill_functor.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
fill_functor.h 3 KB -rwxr-xr-x view | raw | blame | history
fixed_length_record_reader_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
fractional_avg_pool_op.cc 14 KB -rwxr-xr-x view | raw | blame | history
fractional_max_pool_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
fractional_pool_common.cc 5 KB -rwxr-xr-x view | raw | blame | history
fractional_pool_common.h 3 KB -rwxr-xr-x view | raw | blame | history
function_ops.cc 16 KB -rwxr-xr-x view | raw | blame | history
function_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
functional_ops.cc 22 KB -rwxr-xr-x view | raw | blame | history
fused_batch_norm_op.cc 34 KB -rwxr-xr-x view | raw | blame | history
fused_batch_norm_op.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
fused_batch_norm_op.h 6 KB -rwxr-xr-x view | raw | blame | history
fused_batch_norm_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
gather_functor.cc 2 KB -rwxr-xr-x view | raw | blame | history
gather_functor.h 7 KB -rwxr-xr-x view | raw | blame | history
gather_functor_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
gather_functor_gpu.cu.h 4 KB -rwxr-xr-x view | raw | blame | history
gather_nd_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
gather_nd_op.h 2 KB -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl.h 6 KB -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_0.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_1.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_2.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_3.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_4.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_5.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_6.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_cpu_impl_7.cc 784 b -rwxr-xr-x view | raw | blame | history
gather_nd_op_gpu.cu.cc 5 KB -rwxr-xr-x view | raw | blame | history
gather_nd_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
gather_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
gather_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
gemm_functors.h 5 KB -rwxr-xr-x view | raw | blame | history
generate_vocab_remapping_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
gpu_device_array.h 4 KB -rwxr-xr-x view | raw | blame | history
gpu_device_array_gpu.h 2 KB -rwxr-xr-x view | raw | blame | history
gpu_utils.cc 6 KB -rwxr-xr-x view | raw | blame | history
gpu_utils.h 7 KB -rwxr-xr-x view | raw | blame | history
guarantee_const_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
guarantee_const_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
hinge-loss.h 5 KB -rwxr-xr-x view | raw | blame | history
histogram_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
histogram_op.h 1 KB -rwxr-xr-x view | raw | blame | history
histogram_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
host_constant_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
host_constant_op.h 1 KB -rwxr-xr-x view | raw | blame | history
i_remote_fused_graph_executor.h 3 KB -rwxr-xr-x view | raw | blame | history
i_remote_fused_graph_ops_definitions.cc 847 b -rwxr-xr-x view | raw | blame | history
i_remote_fused_graph_ops_definitions.h 2 KB -rwxr-xr-x view | raw | blame | history
identity_n_op.cc 1 KB -rwxr-xr-x view | raw | blame | history
identity_n_op.h 2 KB -rwxr-xr-x view | raw | blame | history
identity_n_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
identity_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
identity_op.h 1 KB -rwxr-xr-x view | raw | blame | history
identity_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
identity_reader_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
image_resizer_state.h 9 KB -rwxr-xr-x view | raw | blame | history
immutable_constant_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
immutable_constant_op.h 2 KB -rwxr-xr-x view | raw | blame | history
immutable_constant_op_test.cc 7 KB -rwxr-xr-x view | raw | blame | history
in_topk_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
initializable_lookup_table.cc 2 KB -rwxr-xr-x view | raw | blame | history
initializable_lookup_table.h 8 KB -rwxr-xr-x view | raw | blame | history
inplace_ops.cc 20 KB -rwxr-xr-x view | raw | blame | history
inplace_ops_functor.h 2 KB -rwxr-xr-x view | raw | blame | history
inplace_ops_functor_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
kernel_platform_strings.h 978 b -rwxr-xr-x view | raw | blame | history
l2loss_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
l2loss_op.h 1 KB -rwxr-xr-x view | raw | blame | history
l2loss_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
linalg_ops_common.cc 11 KB -rwxr-xr-x view | raw | blame | history
linalg_ops_common.h 9 KB -rwxr-xr-x view | raw | blame | history
list_kernels.cc 28 KB -rwxr-xr-x view | raw | blame | history
list_kernels.cu.cc 8 KB -rwxr-xr-x view | raw | blame | history
list_kernels.h 43 KB -rwxr-xr-x view | raw | blame | history
listdiff_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
lmdb_reader_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
load_and_remap_matrix_op.cc 14 KB -rwxr-xr-x view | raw | blame | history
logging_ops.cc 8 KB -rwxr-xr-x view | raw | blame | history
logging_ops.h 1 KB -rwxr-xr-x view | raw | blame | history
logging_ops_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
logistic-loss.h 5 KB -rwxr-xr-x view | raw | blame | history
lookup_table_init_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
lookup_table_init_op.h 1 KB -rwxr-xr-x view | raw | blame | history
lookup_table_op.cc 42 KB -rwxr-xr-x view | raw | blame | history
lookup_table_op.h 9 KB -rwxr-xr-x view | raw | blame | history
lookup_util.cc 14 KB -rwxr-xr-x view | raw | blame | history
lookup_util.h 2 KB -rwxr-xr-x view | raw | blame | history
loss.h 2 KB -rwxr-xr-x view | raw | blame | history
loss_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
lrn_op.cc 19 KB -rwxr-xr-x view | raw | blame | history
lrn_op_test.cc 7 KB -rwxr-xr-x view | raw | blame | history
lu_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
lu_op_gpu.cu.cc 11 KB -rwxr-xr-x view | raw | blame | history
map_stage_op.cc 28 KB -rwxr-xr-x view | raw | blame | history
matching_files_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
matmul_op.cc 25 KB -rwxr-xr-x view | raw | blame | history
matmul_op.h 4 KB -rwxr-xr-x view | raw | blame | history
matmul_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
matrix_band_part_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
matrix_band_part_op.h 1 KB -rwxr-xr-x view | raw | blame | history
matrix_band_part_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
matrix_diag_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
matrix_diag_op.h 3 KB -rwxr-xr-x view | raw | blame | history
matrix_diag_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
matrix_exponential_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
matrix_inverse_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
matrix_logarithm_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
matrix_set_diag_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
matrix_set_diag_op.h 1 KB -rwxr-xr-x view | raw | blame | history
matrix_set_diag_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
matrix_solve_ls_op_complex128.cc 893 b -rwxr-xr-x view | raw | blame | history
matrix_solve_ls_op_complex64.cc 891 b -rwxr-xr-x view | raw | blame | history
matrix_solve_ls_op_double.cc 933 b -rwxr-xr-x view | raw | blame | history
matrix_solve_ls_op_float.cc 929 b -rwxr-xr-x view | raw | blame | history
matrix_solve_ls_op_impl.h 7 KB -rwxr-xr-x view | raw | blame | history
matrix_solve_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
matrix_square_root_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
matrix_triangular_solve_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
maxpooling_op.cc 67 KB -rwxr-xr-x view | raw | blame | history
maxpooling_op.h 2 KB -rwxr-xr-x view | raw | blame | history
maxpooling_op_gpu.cu.cc 23 KB -rwxr-xr-x view | raw | blame | history
maxpooling_op_gpu.h 4 KB -rwxr-xr-x view | raw | blame | history
merge_v2_checkpoints_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
meta_support.cc 14 KB -rwxr-xr-x view | raw | blame | history
meta_support.h 5 KB -rwxr-xr-x view | raw | blame | history
mfcc.cc 2 KB -rwxr-xr-x view | raw | blame | history
mfcc.h 3 KB -rwxr-xr-x view | raw | blame | history
mfcc_dct.cc 2 KB -rwxr-xr-x view | raw | blame | history
mfcc_dct.h 1 KB -rwxr-xr-x view | raw | blame | history
mfcc_dct_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
mfcc_mel_filterbank.cc 8 KB -rwxr-xr-x view | raw | blame | history
mfcc_mel_filterbank.h 2 KB -rwxr-xr-x view | raw | blame | history
mfcc_mel_filterbank_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
mfcc_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
mfcc_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
mfcc_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
mirror_pad_op.cc 18 KB -rwxr-xr-x view | raw | blame | history
mirror_pad_op.h 16 KB -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl.h 2 KB -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl_1.cc 785 b -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl_2.cc 785 b -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl_3.cc 785 b -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl_4.cc 785 b -rwxr-xr-x view | raw | blame | history
mirror_pad_op_cpu_impl_5.cc 785 b -rwxr-xr-x view | raw | blame | history
mirror_pad_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
mkl_aggregate_ops.cc 12 KB -rwxr-xr-x view | raw | blame | history
mkl_avgpooling_op.cc 31 KB -rwxr-xr-x view | raw | blame | history
mkl_batch_matmul_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
mkl_concat_op.cc 24 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_grad_bias_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_grad_filter_ops.cc 31 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_grad_input_ops.cc 24 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_ops.cc 92 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_ops.h 27 KB -rwxr-xr-x view | raw | blame | history
mkl_conv_ops_test.cc 16 KB -rwxr-xr-x view | raw | blame | history
mkl_cwise_ops_common.cc 3 KB -rwxr-xr-x view | raw | blame | history
mkl_fused_batch_norm_op.cc 45 KB -rwxr-xr-x view | raw | blame | history
mkl_fused_ops_test.cc 28 KB -rwxr-xr-x view | raw | blame | history
mkl_identity_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
mkl_input_conversion_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
mkl_lrn_op.cc 29 KB -rwxr-xr-x view | raw | blame | history
mkl_matmul_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
mkl_maxpooling_op.cc 38 KB -rwxr-xr-x view | raw | blame | history
mkl_pooling_ops_common.cc 18 KB -rwxr-xr-x view | raw | blame | history
mkl_pooling_ops_common.h 28 KB -rwxr-xr-x view | raw | blame | history
mkl_quantized_concat_op_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
mkl_quantized_conv_ops.h 4 KB -rwxr-xr-x view | raw | blame | history
mkl_quantized_conv_ops_test.cc 18 KB -rwxr-xr-x view | raw | blame | history
mkl_quantized_pooling_ops_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
mkl_relu_op.cc 44 KB -rwxr-xr-x view | raw | blame | history
mkl_requantization_range_per_channel_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
mkl_requantize_ops_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
mkl_requantize_per_channel_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
mkl_reshape_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
mkl_slice_op.cc 19 KB -rwxr-xr-x view | raw | blame | history
mkl_softmax_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
mkl_tfconv_op.h 6 KB -rwxr-xr-x view | raw | blame | history
mkl_transpose_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
multinomial_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
multinomial_op.h 1 KB -rwxr-xr-x view | raw | blame | history
multinomial_op_gpu.cu.cc 6 KB -rwxr-xr-x view | raw | blame | history
multinomial_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
mutex_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
nccl_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
nextafter_op.cc 1 KB -rwxr-xr-x view | raw | blame | history
nextafter_op.h 1 KB -rwxr-xr-x view | raw | blame | history
nextafter_op_gpu.cu.cc 961 b -rwxr-xr-x view | raw | blame | history
nn_ops_test.cc 69 KB -rwxr-xr-x view | raw | blame | history
no_op.cc 991 b -rwxr-xr-x view | raw | blame | history
no_op.h 1 KB -rwxr-xr-x view | raw | blame | history
non_max_suppression_op.cc 32 KB -rwxr-xr-x view | raw | blame | history
non_max_suppression_op.h 1 KB -rwxr-xr-x view | raw | blame | history
non_max_suppression_op_test.cc 53 KB -rwxr-xr-x view | raw | blame | history
nth_element_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
nth_element_op.h 1 KB -rwxr-xr-x view | raw | blame | history
one_hot_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
one_hot_op.h 2 KB -rwxr-xr-x view | raw | blame | history
one_hot_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
ops_testutil.cc 2 KB -rwxr-xr-x view | raw | blame | history
ops_testutil.h 9 KB -rwxr-xr-x view | raw | blame | history
ops_testutil_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
ops_util.h 931 b -rwxr-xr-x view | raw | blame | history
ops_util_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
pack_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
pad_op.cc 22 KB -rwxr-xr-x view | raw | blame | history
pad_op.h 2 KB -rwxr-xr-x view | raw | blame | history
pad_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
padding_fifo_queue.cc 16 KB -rwxr-xr-x view | raw | blame | history
padding_fifo_queue.h 3 KB -rwxr-xr-x view | raw | blame | history
padding_fifo_queue_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
parameterized_truncated_normal_op.cc 20 KB -rwxr-xr-x view | raw | blame | history
parameterized_truncated_normal_op.h 2 KB -rwxr-xr-x view | raw | blame | history
parameterized_truncated_normal_op_gpu.cu.cc 10 KB -rwxr-xr-x view | raw | blame | history
parameterized_truncated_normal_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
parse_tensor_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
parse_tensor_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
partitioned_function_ops.cc 11 KB -rwxr-xr-x view | raw | blame | history
partitioned_function_ops.h 3 KB -rwxr-xr-x view | raw | blame | history
poisson-loss.h 4 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_3d.cc 38 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_3d.h 2 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_3d_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_3d_gpu.h 2 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_3d_sycl.h 32 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_common.cc 18 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_common.h 26 KB -rwxr-xr-x view | raw | blame | history
pooling_ops_common_gpu.h 3 KB -rwxr-xr-x view | raw | blame | history
population_count_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
population_count_op.h 1 KB -rwxr-xr-x view | raw | blame | history
population_count_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
priority_queue.cc 18 KB -rwxr-xr-x view | raw | blame | history
priority_queue.h 4 KB -rwxr-xr-x view | raw | blame | history
priority_queue_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
qr_op_complex128.cc 1 KB -rwxr-xr-x view | raw | blame | history
qr_op_complex64.cc 1 KB -rwxr-xr-x view | raw | blame | history
qr_op_double.cc 1 KB -rwxr-xr-x view | raw | blame | history
qr_op_float.cc 1 KB -rwxr-xr-x view | raw | blame | history
qr_op_impl.h 11 KB -rwxr-xr-x view | raw | blame | history
quantization_utils.cc 2 KB -rwxr-xr-x view | raw | blame | history
quantization_utils.h 42 KB -rwxr-xr-x view | raw | blame | history
quantization_utils_test.cc 41 KB -rwxr-xr-x view | raw | blame | history
quantize_and_dequantize_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
quantize_and_dequantize_op.h 7 KB -rwxr-xr-x view | raw | blame | history
quantize_and_dequantize_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
quantize_and_dequantize_op_test.cc 22 KB -rwxr-xr-x view | raw | blame | history
quantize_down_and_shrink_range.cc 4 KB -rwxr-xr-x view | raw | blame | history
quantize_down_and_shrink_range_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
quantize_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
quantize_op_test.cc 17 KB -rwxr-xr-x view | raw | blame | history
quantized_activation_ops.cc 5 KB -rwxr-xr-x view | raw | blame | history
quantized_activation_ops_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
quantized_add_op.cc 25 KB -rwxr-xr-x view | raw | blame | history
quantized_add_op_test.cc 11 KB -rwxr-xr-x view | raw | blame | history
quantized_batch_norm_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
quantized_batch_norm_op_test.cc 11 KB -rwxr-xr-x view | raw | blame | history
quantized_bias_add_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
quantized_bias_add_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
quantized_concat_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
quantized_concat_op_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
quantized_conv_ops.cc 30 KB -rwxr-xr-x view | raw | blame | history
quantized_conv_ops_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
quantized_instance_norm.cc 17 KB -rwxr-xr-x view | raw | blame | history
quantized_instance_norm_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
quantized_matmul_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
quantized_matmul_op_test.cc 16 KB -rwxr-xr-x view | raw | blame | history
quantized_mul_op.cc 17 KB -rwxr-xr-x view | raw | blame | history
quantized_mul_op_test.cc 11 KB -rwxr-xr-x view | raw | blame | history
quantized_pooling_ops.cc 5 KB -rwxr-xr-x view | raw | blame | history
quantized_pooling_ops_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
quantized_reshape_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
quantized_reshape_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
quantized_resize_bilinear_op.cc 32 KB -rwxr-xr-x view | raw | blame | history
quantized_resize_bilinear_op_test.cc 18 KB -rwxr-xr-x view | raw | blame | history
queue_base.cc 12 KB -rwxr-xr-x view | raw | blame | history
queue_base.h 7 KB -rwxr-xr-x view | raw | blame | history
queue_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
queue_op.h 9 KB -rwxr-xr-x view | raw | blame | history
queue_ops.cc 3 KB -rwxr-xr-x view | raw | blame | history
ragged_gather_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
ragged_gather_op_test.cc 12 KB -rwxr-xr-x view | raw | blame | history
ragged_range_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
ragged_range_op_test.cc 9 KB -rwxr-xr-x view | raw | blame | history
ragged_tensor_to_sparse_kernel.cc 9 KB -rwxr-xr-x view | raw | blame | history
ragged_tensor_to_sparse_kernel_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
random_crop_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
random_op.cc 33 KB -rwxr-xr-x view | raw | blame | history
random_op.h 3 KB -rwxr-xr-x view | raw | blame | history
random_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
random_op_gpu.h 7 KB -rwxr-xr-x view | raw | blame | history
random_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
random_poisson_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
random_poisson_op.h 1 KB -rwxr-xr-x view | raw | blame | history
random_poisson_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
random_shuffle_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
random_shuffle_queue_op.cc 20 KB -rwxr-xr-x view | raw | blame | history
range_sampler.cc 11 KB -rwxr-xr-x view | raw | blame | history
range_sampler.h 9 KB -rwxr-xr-x view | raw | blame | history
range_sampler_test.cc 11 KB -rwxr-xr-x view | raw | blame | history
reader_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
record_input_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
record_yielder.cc 7 KB -rwxr-xr-x view | raw | blame | history
record_yielder.h 5 KB -rwxr-xr-x view | raw | blame | history
reduce_join_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
reduction_gpu_kernels.cu.h 40 KB -rwxr-xr-x view | raw | blame | history
reduction_ops.h 6 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_all.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_any.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_common.cc 5 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_common.h 11 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_euclidean.cc 4 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_bool.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_complex128.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_complex64.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_double.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_float.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_gpu_int.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_half_mean_sum.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_half_prod_max_min.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_max.cc 6 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_mean.cc 4 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_min.cc 6 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_prod.cc 4 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_sum.cc 6 KB -rwxr-xr-x view | raw | blame | history
reduction_ops_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
redux_functor.h 5 KB -rwxr-xr-x view | raw | blame | history
reference_gemm.h 3 KB -rwxr-xr-x view | raw | blame | history
regex_full_match_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
regex_replace_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
regex_replace_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
relu_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
relu_op.h 10 KB -rwxr-xr-x view | raw | blame | history
relu_op_functor.h 8 KB -rwxr-xr-x view | raw | blame | history
relu_op_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_op_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_op_test_utils.cc 5 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_op_test_utils.h 3 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_utils.cc 53 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_utils.h 15 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_execute_utils_test.cc 33 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_rewriter_transform.cc 11 KB -rwxr-xr-x view | raw | blame | history
remote_fused_graph_rewriter_transform_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
requantization_range_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
requantization_range_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
requantize.cc 4 KB -rwxr-xr-x view | raw | blame | history
requantize_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
reshape_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
reshape_op.h 5 KB -rwxr-xr-x view | raw | blame | history
reshape_util.cc 6 KB -rwxr-xr-x view | raw | blame | history
reshape_util.h 1 KB -rwxr-xr-x view | raw | blame | history
resize_area_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
resize_area_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
resize_bicubic_op.cc 25 KB -rwxr-xr-x view | raw | blame | history
resize_bicubic_op_test.cc 12 KB -rwxr-xr-x view | raw | blame | history
resize_bilinear_op.cc 17 KB -rwxr-xr-x view | raw | blame | history
resize_bilinear_op.h 2 KB -rwxr-xr-x view | raw | blame | history
resize_bilinear_op_gpu.cu.cc 14 KB -rwxr-xr-x view | raw | blame | history
resize_bilinear_op_test.cc 18 KB -rwxr-xr-x view | raw | blame | history
resize_nearest_neighbor_op.cc 14 KB -rwxr-xr-x view | raw | blame | history
resize_nearest_neighbor_op.h 2 KB -rwxr-xr-x view | raw | blame | history
resize_nearest_neighbor_op_gpu.cu.cc 10 KB -rwxr-xr-x view | raw | blame | history
resize_nearest_neighbor_op_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
resize_op_benchmark_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
resource_variable_ops.cc 37 KB -rwxr-xr-x view | raw | blame | history
resource_variable_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
restore_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
restore_op_test.cc 16 KB -rwxr-xr-x view | raw | blame | history
restore_v2_op_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
reverse_op.cc 19 KB -rwxr-xr-x view | raw | blame | history
reverse_op.h 2 KB -rwxr-xr-x view | raw | blame | history
reverse_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
reverse_op_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
reverse_sequence_op.cc 9 KB -rwxr-xr-x view | raw | blame | history
reverse_sequence_op.h 3 KB -rwxr-xr-x view | raw | blame | history
reverse_sequence_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
roll_op.cc 15 KB -rwxr-xr-x view | raw | blame | history
roll_op_test.cc 18 KB -rwxr-xr-x view | raw | blame | history
rpc_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
sample_distorted_bounding_box_op.cc 17 KB -rwxr-xr-x view | raw | blame | history
sampling_kernels.cc 2 KB -rwxr-xr-x view | raw | blame | history
sampling_kernels.h 6 KB -rwxr-xr-x view | raw | blame | history
sampling_kernels_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
save_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
save_op_test.cc 23 KB -rwxr-xr-x view | raw | blame | history
save_restore_tensor.cc 15 KB -rwxr-xr-x view | raw | blame | history
save_restore_tensor.h 3 KB -rwxr-xr-x view | raw | blame | history
save_restore_v2_ops.cc 11 KB -rwxr-xr-x view | raw | blame | history
save_v2_op_test.cc 11 KB -rwxr-xr-x view | raw | blame | history
scale_and_translate_op.cc 27 KB -rwxr-xr-x view | raw | blame | history
scale_and_translate_op.h 3 KB -rwxr-xr-x view | raw | blame | history
scale_and_translate_op_test.cc 18 KB -rwxr-xr-x view | raw | blame | history
scan_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
scan_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
scan_ops_gpu.h 11 KB -rwxr-xr-x view | raw | blame | history
scan_ops_gpu_double.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
scan_ops_gpu_float.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
scan_ops_gpu_half.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
scan_ops_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
scatter_functor.cc 3 KB -rwxr-xr-x view | raw | blame | history
scatter_functor.h 19 KB -rwxr-xr-x view | raw | blame | history
scatter_functor_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
scatter_functor_gpu.cu.h 6 KB -rwxr-xr-x view | raw | blame | history
scatter_nd_op.cc 32 KB -rwxr-xr-x view | raw | blame | history
scatter_nd_op.h 3 KB -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl.h 10 KB -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_0.cc 858 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_1.cc 785 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_2.cc 785 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_3.cc 785 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_4.cc 785 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_5.cc 786 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_6.cc 785 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_cpu_impl_7.cc 786 b -rwxr-xr-x view | raw | blame | history
scatter_nd_op_gpu.cu.cc 6 KB -rwxr-xr-x view | raw | blame | history
scatter_nd_op_test.cc 12 KB -rwxr-xr-x view | raw | blame | history
scatter_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
scatter_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
scatter_op_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
scoped_allocator_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
scoped_allocator_ops_test.cc 14 KB -rwxr-xr-x view | raw | blame | history
sdca_internal.cc 23 KB -rwxr-xr-x view | raw | blame | history
sdca_internal.h 14 KB -rwxr-xr-x view | raw | blame | history
sdca_ops.cc 13 KB -rwxr-xr-x view | raw | blame | history
sdca_ops_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
searchsorted_op.cc 10 KB -rwxr-xr-x view | raw | blame | history
searchsorted_op.h 2 KB -rwxr-xr-x view | raw | blame | history
searchsorted_op_gpu.cu.cc 5 KB -rwxr-xr-x view | raw | blame | history
segment_reduction_ops.cc 47 KB -rwxr-xr-x view | raw | blame | history
segment_reduction_ops.h 5 KB -rwxr-xr-x view | raw | blame | history
segment_reduction_ops_gpu.cu.cc 11 KB -rwxr-xr-x view | raw | blame | history
segment_reduction_ops_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_complex128.cc 1 KB -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_complex64.cc 1,022 b -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_double.cc 991 b -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_float.cc 968 b -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_gpu.cc 7 KB -rwxr-xr-x view | raw | blame | history
self_adjoint_eig_v2_op_impl.h 3 KB -rwxr-xr-x view | raw | blame | history
sendrecv_ops.cc 8 KB -rwxr-xr-x view | raw | blame | history
sendrecv_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
sendrecv_ops_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
sequence_ops.cc 8 KB -rwxr-xr-x view | raw | blame | history
sequence_ops_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
serialize_sparse_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
session_ops.cc 7 KB -rwxr-xr-x view | raw | blame | history
set_kernels.cc 30 KB -rwxr-xr-x view | raw | blame | history
shape_ops.cc 24 KB -rwxr-xr-x view | raw | blame | history
shape_ops.h 9 KB -rwxr-xr-x view | raw | blame | history
slice_op.cc 14 KB -rwxr-xr-x view | raw | blame | history
slice_op.h 2 KB -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl.h 2 KB -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_1.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_2.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_3.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_4.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_5.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_6.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_cpu_impl_7.cc 780 b -rwxr-xr-x view | raw | blame | history
slice_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
slice_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
smooth-hinge-loss.h 4 KB -rwxr-xr-x view | raw | blame | history
snapshot_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
snapshot_op.h 1 KB -rwxr-xr-x view | raw | blame | history
snapshot_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
softmax_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
softmax_op_functor.h 4 KB -rwxr-xr-x view | raw | blame | history
softmax_op_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
softplus_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
softplus_op.h 3 KB -rwxr-xr-x view | raw | blame | history
softplus_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
softsign_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
softsign_op.h 2 KB -rwxr-xr-x view | raw | blame | history
softsign_op_gpu.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
spacetobatch_benchmark_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
spacetobatch_functor.cc 7 KB -rwxr-xr-x view | raw | blame | history
spacetobatch_functor.h 4 KB -rwxr-xr-x view | raw | blame | history
spacetobatch_functor_gpu.cu.cc 7 KB -rwxr-xr-x view | raw | blame | history
spacetobatch_op.cc 12 KB -rwxr-xr-x view | raw | blame | history
spacetodepth_op.cc 8 KB -rwxr-xr-x view | raw | blame | history
spacetodepth_op.h 3 KB -rwxr-xr-x view | raw | blame | history
spacetodepth_op_gpu.cu.cc 11 KB -rwxr-xr-x view | raw | blame | history
sparse_add_grad_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_add_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
sparse_add_op_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
sparse_concat_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
sparse_conditional_accumulator.h 18 KB -rwxr-xr-x view | raw | blame | history
sparse_conditional_accumulator_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_cross_op.cc 22 KB -rwxr-xr-x view | raw | blame | history
sparse_dense_binary_op_shared.cc 9 KB -rwxr-xr-x view | raw | blame | history
sparse_dense_binary_op_shared_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
sparse_fill_empty_rows_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
sparse_matmul_op.cc 61 KB -rwxr-xr-x view | raw | blame | history
sparse_matmul_op.h 16 KB -rwxr-xr-x view | raw | blame | history
sparse_matmul_op_test.cc 17 KB -rwxr-xr-x view | raw | blame | history
sparse_reduce_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
sparse_reduce_sum_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_reorder_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_reshape_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
sparse_slice_grad_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_slice_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_softmax_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
sparse_sparse_binary_op_shared.cc 10 KB -rwxr-xr-x view | raw | blame | history
sparse_split_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_add_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_add_op.h 2 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_matmul_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_matmul_op.h 2 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_matmul_op_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
sparse_tensor_dense_matmul_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_tensors_map_ops.cc 19 KB -rwxr-xr-x view | raw | blame | history
sparse_to_dense_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_to_dense_op_test.cc 9 KB -rwxr-xr-x view | raw | blame | history
sparse_utils.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_utils.h 3 KB -rwxr-xr-x view | raw | blame | history
sparse_utils_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
sparse_xent_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
sparse_xent_op.h 9 KB -rwxr-xr-x view | raw | blame | history
sparse_xent_op_gpu.cu.cc 3 KB -rwxr-xr-x view | raw | blame | history
sparse_xent_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
spectrogram.cc 8 KB -rwxr-xr-x view | raw | blame | history
spectrogram.h 4 KB -rwxr-xr-x view | raw | blame | history
spectrogram_convert_test_data.cc 2 KB -rwxr-xr-x view | raw | blame | history
spectrogram_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
spectrogram_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
spectrogram_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
spectrogram_test_utils.cc 10 KB -rwxr-xr-x view | raw | blame | history
spectrogram_test_utils.h 3 KB -rwxr-xr-x view | raw | blame | history
split_lib.h 3 KB -rwxr-xr-x view | raw | blame | history
split_lib_cpu.cc 2 KB -rwxr-xr-x view | raw | blame | history
split_lib_gpu.cu.cc 10 KB -rwxr-xr-x view | raw | blame | history
split_lib_gpu.h 2 KB -rwxr-xr-x view | raw | blame | history
split_op.cc 17 KB -rwxr-xr-x view | raw | blame | history
split_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
split_v_op.cc 19 KB -rwxr-xr-x view | raw | blame | history
split_v_op_test.cc 7 KB -rwxr-xr-x view | raw | blame | history
squared-loss.h 3 KB -rwxr-xr-x view | raw | blame | history
stack.cc 12 KB -rwxr-xr-x view | raw | blame | history
stack.h 2 KB -rwxr-xr-x view | raw | blame | history
stack_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
stage_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
stateful_random_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
stateful_random_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
stateful_random_ops_cpu_gpu.h 4 KB -rwxr-xr-x view | raw | blame | history
stateful_random_ops_gpu.cu.cc 4 KB -rwxr-xr-x view | raw | blame | history
stateless_random_ops.cc 9 KB -rwxr-xr-x view | raw | blame | history
stateless_random_ops.h 1 KB -rwxr-xr-x view | raw | blame | history
strided_slice_op.cc 25 KB -rwxr-xr-x view | raw | blame | history
strided_slice_op.h 5 KB -rwxr-xr-x view | raw | blame | history
strided_slice_op_define_grad.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_gpu_bool.cu.cc 940 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_gpu_complex.cu.cc 985 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_gpu_impl.h 3 KB -rwxr-xr-x view | raw | blame | history
strided_slice_op_gpu_int.cu.cc 1,010 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_gpu_number_types.cu.cc 952 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_impl.h 14 KB -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_0.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_1.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_2.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_3.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_4.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_5.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_6.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_inst_7.cc 878 b -rwxr-xr-x view | raw | blame | history
strided_slice_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
string_format_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
string_format_op_test.cc 3 KB -rwxr-xr-x view | raw | blame | history
string_join_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
string_length_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
string_split_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
string_split_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
string_strip_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
string_to_hash_bucket_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
string_to_hash_bucket_op.h 4 KB -rwxr-xr-x view | raw | blame | history
string_to_number_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
string_util.cc 2 KB -rwxr-xr-x view | raw | blame | history
string_util.h 4 KB -rwxr-xr-x view | raw | blame | history
string_view_variant_wrapper.cc 848 b -rwxr-xr-x view | raw | blame | history
string_view_variant_wrapper.h 2 KB -rwxr-xr-x view | raw | blame | history
substr_op.cc 16 KB -rwxr-xr-x view | raw | blame | history
substr_op_test.cc 8 KB -rwxr-xr-x view | raw | blame | history
summary_audio_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
summary_audio_op_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
summary_image_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
summary_image_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
summary_interface.h 2 KB -rwxr-xr-x view | raw | blame | history
summary_kernels.cc 12 KB -rwxr-xr-x view | raw | blame | history
summary_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
summary_op_test.cc 13 KB -rwxr-xr-x view | raw | blame | history
summary_tensor_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
summary_tensor_op_test.cc 4 KB -rwxr-xr-x view | raw | blame | history
svd_op_complex128.cc 897 b -rwxr-xr-x view | raw | blame | history
svd_op_complex64.cc 893 b -rwxr-xr-x view | raw | blame | history
svd_op_double.cc 881 b -rwxr-xr-x view | raw | blame | history
svd_op_float.cc 877 b -rwxr-xr-x view | raw | blame | history
svd_op_gpu.cu.cc 14 KB -rwxr-xr-x view | raw | blame | history
svd_op_impl.h 4 KB -rwxr-xr-x view | raw | blame | history
tensor_array.cc 4 KB -rwxr-xr-x view | raw | blame | history
tensor_array.h 24 KB -rwxr-xr-x view | raw | blame | history
tensor_array_ops.cc 64 KB -rwxr-xr-x view | raw | blame | history
tensor_flag_utils.cc 7 KB -rwxr-xr-x view | raw | blame | history
tensor_flag_utils.h 3 KB -rwxr-xr-x view | raw | blame | history
tensor_flag_utils_test.cc 10 KB -rwxr-xr-x view | raw | blame | history
text_line_reader_op.cc 4 KB -rwxr-xr-x view | raw | blame | history
tf_record_reader_op.cc 3 KB -rwxr-xr-x view | raw | blame | history
tile_functor.h 4 KB -rwxr-xr-x view | raw | blame | history
tile_functor_cpu.cc 3 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu.h 3 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_bool.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_complex128.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_complex64.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_double.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_float.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_half.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_int16.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_int32.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_functor_gpu_int64.cu.cc 1 KB -rwxr-xr-x view | raw | blame | history
tile_ops.cc 24 KB -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl.h 2 KB -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_1.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_2.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_3.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_4.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_5.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_6.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_cpu_impl_7.cc 780 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl.h 2 KB -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_1.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_2.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_3.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_4.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_5.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_6.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_7.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_gpu_impl_8.cu.cc 782 b -rwxr-xr-x view | raw | blame | history
tile_ops_impl.h 2 KB -rwxr-xr-x view | raw | blame | history
topk_op.cc 11 KB -rwxr-xr-x view | raw | blame | history
topk_op.h 2 KB -rwxr-xr-x view | raw | blame | history
topk_op_gpu.h 19 KB -rwxr-xr-x view | raw | blame | history
topk_op_gpu_double.cu.cc 960 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_float.cu.cc 959 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_half.cu.cc 965 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_int16.cu.cc 959 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_int32.cu.cc 959 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_int64.cu.cc 959 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_int8.cu.cc 958 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_uint16.cu.cc 960 b -rwxr-xr-x view | raw | blame | history
topk_op_gpu_uint8.cu.cc 959 b -rwxr-xr-x view | raw | blame | history
training_op_helpers.cc 1 KB -rwxr-xr-x view | raw | blame | history
training_op_helpers.h 10 KB -rwxr-xr-x view | raw | blame | history
training_ops.cc 177 KB -rwxr-xr-x view | raw | blame | history
training_ops.h 9 KB -rwxr-xr-x view | raw | blame | history
training_ops_gpu.cu.cc 16 KB -rwxr-xr-x view | raw | blame | history
training_ops_test.cc 9 KB -rwxr-xr-x view | raw | blame | history
transpose_functor.h 9 KB -rwxr-xr-x view | raw | blame | history
transpose_functor_cpu.cc 8 KB -rwxr-xr-x view | raw | blame | history
transpose_functor_gpu.cu.cc 9 KB -rwxr-xr-x view | raw | blame | history
transpose_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
transpose_op.h 4 KB -rwxr-xr-x view | raw | blame | history
transpose_util_test.cc 5 KB -rwxr-xr-x view | raw | blame | history
tridiagonal_solve_op.cc 6 KB -rwxr-xr-x view | raw | blame | history
typed_conditional_accumulator_base.h 4 KB -rwxr-xr-x view | raw | blame | history
typed_queue.h 4 KB -rwxr-xr-x view | raw | blame | history
unary_ops_composition.cc 17 KB -rwxr-xr-x view | raw | blame | history
unary_ops_composition_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
unicode_ops.cc 21 KB -rwxr-xr-x view | raw | blame | history
unicode_script_op.cc 2 KB -rwxr-xr-x view | raw | blame | history
unique_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
unique_op_test.cc 6 KB -rwxr-xr-x view | raw | blame | history
unpack_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
unravel_index_op.cc 5 KB -rwxr-xr-x view | raw | blame | history
variable_ops.cc 10 KB -rwxr-xr-x view | raw | blame | history
variable_ops.h 2 KB -rwxr-xr-x view | raw | blame | history
variable_ops_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
where_op.cc 13 KB -rwxr-xr-x view | raw | blame | history
where_op.h 2 KB -rwxr-xr-x view | raw | blame | history
where_op_gpu.cu.h 13 KB -rwxr-xr-x view | raw | blame | history
where_op_gpu_impl_1.cu.cc 774 b -rwxr-xr-x view | raw | blame | history
where_op_gpu_impl_2.cu.cc 774 b -rwxr-xr-x view | raw | blame | history
where_op_gpu_impl_3.cu.cc 774 b -rwxr-xr-x view | raw | blame | history
where_op_gpu_impl_4.cu.cc 774 b -rwxr-xr-x view | raw | blame | history
where_op_gpu_impl_5.cu.cc 774 b -rwxr-xr-x view | raw | blame | history
whole_file_read_ops.cc 6 KB -rwxr-xr-x view | raw | blame | history
winograd_transform.h 11 KB -rwxr-xr-x view | raw | blame | history
word2vec_kernels.cc 12 KB -rwxr-xr-x view | raw | blame | history
xent_op.cc 7 KB -rwxr-xr-x view | raw | blame | history
xent_op.h 5 KB -rwxr-xr-x view | raw | blame | history
xent_op_gpu.cu.cc 2 KB -rwxr-xr-x view | raw | blame | history
xent_op_test.cc 2 KB -rwxr-xr-x view | raw | blame | history
xsmm_conv2d.cc 18 KB -rwxr-xr-x view | raw | blame | history
xsmm_conv2d.h 2 KB -rwxr-xr-x view | raw | blame | history
xsmm_conv2d_test.cc 10 KB -rwxr-xr-x view | raw | blame | history