|
..
|
|
batching_util
|
|
data
|
|
fuzzing
|
|
image
|
|
linalg
|
|
mkl
|
|
mlir_generated
|
|
risc
|
|
rnn
|
|
sparse
|
|
special_math
|
|
spectrogram_test_data
|
|
BUILD
|
|
aggregate_ops.cc
|
|
aggregate_ops.h
|
|
aggregate_ops_cpu.h
|
|
aggregate_ops_gpu.cu.cc
|
|
argmax_op.cc
|
|
argmax_op.h
|
|
argmax_op_gpu.cu.cc
|
|
as_string_op.cc
|
|
as_string_op_test.cc
|
|
assign_op.h
|
|
autotune_conv_impl.h
|
|
avgpooling_op.cc
|
|
avgpooling_op.h
|
|
avgpooling_op_gpu.cu.cc
|
|
barrier_ops.cc
|
|
base64_ops.cc
|
|
basic_ops_benchmark_test.cc
|
|
batch_kernel_test_util.cc
|
|
batch_kernel_test_util.h
|
|
batch_kernels.cc
|
|
batch_kernels.h
|
|
batch_kernels_env_test.cc
|
|
batch_kernels_test.cc
|
|
batch_norm_op.cc
|
|
batch_norm_op.h
|
|
batch_norm_op_gpu.cu.cc
|
|
batch_norm_op_test.cc
|
|
batch_util.h
|
|
batchtospace_op.cc
|
|
bcast_ops.cc
|
|
betainc_op.cc
|
|
betainc_op.h
|
|
betainc_op_gpu.cu.cc
|
|
bias_op.cc
|
|
bias_op.h
|
|
bias_op_gpu.cu.cc
|
|
bias_op_gpu.h
|
|
bias_op_test.cc
|
|
bincount_op.cc
|
|
bincount_op.h
|
|
bincount_op_gpu.cu.cc
|
|
bincount_op_test.cc
|
|
broadcast_to_op.cc
|
|
broadcast_to_op.h
|
|
broadcast_to_op_gpu.cu.cc
|
|
broadcast_to_op_test.cc
|
|
bucketize_op.cc
|
|
bucketize_op.h
|
|
bucketize_op_gpu.cu.cc
|
|
candidate_sampler_ops.cc
|
|
cast_op.cc
|
|
cast_op.h
|
|
cast_op_gpu.cu.cc
|
|
cast_op_impl.h
|
|
cast_op_impl_bfloat.cc
|
|
cast_op_impl_bool.cc
|
|
cast_op_impl_complex128.cc
|
|
cast_op_impl_complex64.cc
|
|
cast_op_impl_double.cc
|
|
cast_op_impl_float.cc
|
|
cast_op_impl_float8.cc
|
|
cast_op_impl_half.cc
|
|
cast_op_impl_int16.cc
|
|
cast_op_impl_int32.cc
|
|
cast_op_impl_int64.cc
|
|
cast_op_impl_int8.cc
|
|
cast_op_impl_uint16.cc
|
|
cast_op_impl_uint32.cc
|
|
cast_op_impl_uint64.cc
|
|
cast_op_impl_uint8.cc
|
|
cast_op_test.cc
|
|
check_numerics_op.cc
|
|
check_numerics_op_gpu.cu.cc
|
|
checkpoint_callback_manager.cc
|
|
checkpoint_callback_manager.h
|
|
checkpoint_callback_manager_test.cc
|
|
clustering_ops.cc
|
|
clustering_ops_test.cc
|
|
collective_nccl.cc
|
|
collective_nccl.h
|
|
collective_nccl_all_to_all.cc
|
|
collective_nccl_all_to_all.h
|
|
collective_nccl_broadcaster.cc
|
|
collective_nccl_broadcaster.h
|
|
collective_nccl_gatherer.cc
|
|
collective_nccl_gatherer.h
|
|
collective_nccl_reducer.cc
|
|
collective_nccl_reducer.h
|
|
collective_nccl_test.cc
|
|
collective_ops.cc
|
|
composite_tensor_ops.cc
|
|
composite_tensor_variant.cc
|
|
composite_tensor_variant.h
|
|
composite_tensor_variant_test.cc
|
|
concat_lib.h
|
|
concat_lib_cpu.cc
|
|
concat_lib_cpu.h
|
|
concat_lib_gpu.cc
|
|
concat_lib_gpu.h
|
|
concat_lib_gpu_impl.cu.cc
|
|
concat_op.cc
|
|
concat_op_test.cc
|
|
conditional_accumulator.h
|
|
conditional_accumulator_base.cc
|
|
conditional_accumulator_base.h
|
|
conditional_accumulator_base_op.cc
|
|
conditional_accumulator_base_op.h
|
|
conditional_accumulator_op.cc
|
|
constant_op.cc
|
|
constant_op.h
|
|
constant_op_test.cc
|
|
control_flow_ops.cc
|
|
control_flow_ops.h
|
|
control_flow_ops_test.cc
|
|
conv_2d.h
|
|
conv_2d_gpu.h
|
|
conv_2d_gpu_bfloat16.cu.cc
|
|
conv_2d_gpu_double.cu.cc
|
|
conv_2d_gpu_float.cu.cc
|
|
conv_2d_gpu_half.cu.cc
|
|
conv_2d_gpu_int.cu.cc
|
|
conv_2d_gpu_int_spatial_convolution.cu.cc
|
|
conv_2d_gpu_int_spatial_convolution_backward.cu.cc
|
|
conv_2d_gpu_uint16.cu.cc
|
|
conv_2d_gpu_uint32.cu.cc
|
|
conv_2d_gpu_uint64.cu.cc
|
|
conv_2d_gpu_uint8.cu.cc
|
|
conv_3d.h
|
|
conv_grad_filter_ops.cc
|
|
conv_grad_filter_ops_3d.cc
|
|
conv_grad_filter_ops_benchmark_test.cc
|
|
conv_grad_filter_ops_launcher.cc
|
|
conv_grad_input_ops.cc
|
|
conv_grad_input_ops.h
|
|
conv_grad_input_ops_3d.cc
|
|
conv_grad_input_ops_benchmark_test.cc
|
|
conv_grad_input_ops_bfloat16.cc
|
|
conv_grad_input_ops_double.cc
|
|
conv_grad_input_ops_float.cc
|
|
conv_grad_input_ops_half.cc
|
|
conv_grad_input_ops_int32.cc
|
|
conv_grad_ops.h
|
|
conv_grad_shape_utils.cc
|
|
conv_grad_shape_utils.h
|
|
conv_ops.cc
|
|
conv_ops.h
|
|
conv_ops_3d.cc
|
|
conv_ops_benchmark_test.cc
|
|
conv_ops_bfloat16.cc
|
|
conv_ops_double.cc
|
|
conv_ops_float.cc
|
|
conv_ops_fused_double.cc
|
|
conv_ops_fused_float.cc
|
|
conv_ops_fused_half.cc
|
|
conv_ops_fused_image_transform.cc
|
|
conv_ops_fused_impl.h
|
|
conv_ops_fused_int8.cc
|
|
conv_ops_gpu.cc
|
|
conv_ops_gpu.h
|
|
conv_ops_half.cc
|
|
conv_ops_impl.h
|
|
conv_ops_int32.cc
|
|
conv_ops_test.cc
|
|
conv_ops_using_gemm.cc
|
|
count_ops.cc
|
|
count_ops_test.cc
|
|
count_up_to_op.cc
|
|
cross_op.cc
|
|
cross_op.h
|
|
cross_op_gpu.cu.cc
|
|
cross_op_test.cc
|
|
ctc_decoder_ops.cc
|
|
ctc_loss_op.cc
|
|
cudnn_pooling_gpu.cc
|
|
cudnn_pooling_gpu.h
|
|
cudnn_rnn_ops.cc
|
|
cwise_op_abs.cc
|
|
cwise_op_acos.cc
|
|
cwise_op_acosh.cc
|
|
cwise_op_add_1.cc
|
|
cwise_op_add_2.cc
|
|
cwise_op_arg.cc
|
|
cwise_op_asin.cc
|
|
cwise_op_asinh.cc
|
|
cwise_op_atan.cc
|
|
cwise_op_atan2.cc
|
|
cwise_op_atanh.cc
|
|
cwise_op_bitwise_and.cc
|
|
cwise_op_bitwise_or.cc
|
|
cwise_op_bitwise_xor.cc
|
|
cwise_op_ceil.cc
|
|
cwise_op_clip.cc
|
|
cwise_op_clip.h
|
|
cwise_op_clip_gpu.cu.cc
|
|
cwise_op_complex.cc
|
|
cwise_op_conj.cc
|
|
cwise_op_cos.cc
|
|
cwise_op_cosh.cc
|
|
cwise_op_digamma.cc
|
|
cwise_op_div.cc
|
|
cwise_op_equal_to_1.cc
|
|
cwise_op_equal_to_2.cc
|
|
cwise_op_erf.cc
|
|
cwise_op_erfc.cc
|
|
cwise_op_exp.cc
|
|
cwise_op_expm1.cc
|
|
cwise_op_floor.cc
|
|
cwise_op_floor_div.cc
|
|
cwise_op_floor_mod.cc
|
|
cwise_op_gpu_abs.cu.cc
|
|
cwise_op_gpu_acos.cu.cc
|
|
cwise_op_gpu_acosh.cu.cc
|
|
cwise_op_gpu_add.cu.cc
|
|
cwise_op_gpu_arg.cu.cc
|
|
cwise_op_gpu_asin.cu.cc
|
|
cwise_op_gpu_asinh.cu.cc
|
|
cwise_op_gpu_atan.cu.cc
|
|
cwise_op_gpu_atan2.cu.cc
|
|
cwise_op_gpu_atanh.cu.cc
|
|
cwise_op_gpu_bitwise_and.cu.cc
|
|
cwise_op_gpu_bitwise_or.cu.cc
|
|
cwise_op_gpu_bitwise_xor.cu.cc
|
|
cwise_op_gpu_ceil.cu.cc
|
|
cwise_op_gpu_complex.cu.cc
|
|
cwise_op_gpu_conj.cu.cc
|
|
cwise_op_gpu_cos.cu.cc
|
|
cwise_op_gpu_cosh.cu.cc
|
|
cwise_op_gpu_digamma.cu.cc
|
|
cwise_op_gpu_div_shard1.cu.cc
|
|
cwise_op_gpu_div_shard2.cu.cc
|
|
cwise_op_gpu_div_shard3.cu.cc
|
|
cwise_op_gpu_div_shard4.cu.cc
|
|
cwise_op_gpu_equal_to.cu.cc
|
|
cwise_op_gpu_erf.cu.cc
|
|
cwise_op_gpu_erfc.cu.cc
|
|
cwise_op_gpu_exp.cu.cc
|
|
cwise_op_gpu_expm1.cu.cc
|
|
cwise_op_gpu_floor.cu.cc
|
|
cwise_op_gpu_floor_div.cu.cc
|
|
cwise_op_gpu_floor_mod.cu.cc
|
|
cwise_op_gpu_greater.cu.cc
|
|
cwise_op_gpu_greater_equal.cu.cc
|
|
cwise_op_gpu_igammas_double.cu.cc
|
|
cwise_op_gpu_igammas_float.cu.cc
|
|
cwise_op_gpu_imag.cu.cc
|
|
cwise_op_gpu_inverse.cu.cc
|
|
cwise_op_gpu_invert.cu.cc
|
|
cwise_op_gpu_isfinite.cu.cc
|
|
cwise_op_gpu_isinf.cu.cc
|
|
cwise_op_gpu_isnan.cu.cc
|
|
cwise_op_gpu_left_shift.cu.cc
|
|
cwise_op_gpu_less.cu.cc
|
|
cwise_op_gpu_less_equal.cu.cc
|
|
cwise_op_gpu_lgamma.cu.cc
|
|
cwise_op_gpu_log.cu.cc
|
|
cwise_op_gpu_log1p.cu.cc
|
|
cwise_op_gpu_logical_and.cu.cc
|
|
cwise_op_gpu_logical_not.cu.cc
|
|
cwise_op_gpu_logical_or.cu.cc
|
|
cwise_op_gpu_maximum.cu.cc
|
|
cwise_op_gpu_minimum.cu.cc
|
|
cwise_op_gpu_mod.cu.cc
|
|
cwise_op_gpu_mul_shard0.cu.cc
|
|
cwise_op_gpu_mul_shard1.cu.cc
|
|
cwise_op_gpu_mul_shard2.cu.cc
|
|
cwise_op_gpu_neg.cu.cc
|
|
cwise_op_gpu_not_equal_to.cu.cc
|
|
cwise_op_gpu_polygamma.cu.cc
|
|
cwise_op_gpu_pow.cu.cc
|
|
cwise_op_gpu_random_grad.cu.cc
|
|
cwise_op_gpu_real.cu.cc
|
|
cwise_op_gpu_right_shift.cu.cc
|
|
cwise_op_gpu_rint.cu.cc
|
|
cwise_op_gpu_round.cu.cc
|
|
cwise_op_gpu_rsqrt.cu.cc
|
|
cwise_op_gpu_select.cu.cc
|
|
cwise_op_gpu_sigmoid.cu.cc
|
|
cwise_op_gpu_sign.cu.cc
|
|
cwise_op_gpu_sin.cu.cc
|
|
cwise_op_gpu_sinh.cu.cc
|
|
cwise_op_gpu_sqrt.cu.cc
|
|
cwise_op_gpu_square.cu.cc
|
|
cwise_op_gpu_squared_difference.cu.cc
|
|
cwise_op_gpu_sub.cu.cc
|
|
cwise_op_gpu_tan.cu.cc
|
|
cwise_op_gpu_tanh.cu.cc
|
|
cwise_op_gpu_xdivy.cu.cc
|
|
cwise_op_gpu_xlog1py.cu.cc
|
|
cwise_op_gpu_xlogy.cu.cc
|
|
cwise_op_gpu_zeta.cu.cc
|
|
cwise_op_greater.cc
|
|
cwise_op_greater_equal.cc
|
|
cwise_op_igammas.cc
|
|
cwise_op_imag.cc
|
|
cwise_op_invert.cc
|
|
cwise_op_isfinite.cc
|
|
cwise_op_isinf.cc
|
|
cwise_op_isnan.cc
|
|
cwise_op_leakyrelu.cc
|
|
cwise_op_left_shift.cc
|
|
cwise_op_less.cc
|
|
cwise_op_less_equal.cc
|
|
cwise_op_lgamma.cc
|
|
cwise_op_log.cc
|
|
cwise_op_log1p.cc
|
|
cwise_op_logical_and.cc
|
|
cwise_op_logical_not.cc
|
|
cwise_op_logical_or.cc
|
|
cwise_op_maximum.cc
|
|
cwise_op_minimum.cc
|
|
cwise_op_mod.cc
|
|
cwise_op_mul_1.cc
|
|
cwise_op_mul_2.cc
|
|
cwise_op_ndtri.cc
|
|
cwise_op_ndtri_double.cu.cc
|
|
cwise_op_ndtri_float.cu.cc
|
|
cwise_op_neg_1.cc
|
|
cwise_op_neg_2.cc
|
|
cwise_op_not_equal_to_1.cc
|
|
cwise_op_not_equal_to_2.cc
|
|
cwise_op_pow.cc
|
|
cwise_op_random_grad.cc
|
|
cwise_op_real.cc
|
|
cwise_op_reciprocal.cc
|
|
cwise_op_right_shift.cc
|
|
cwise_op_rint.cc
|
|
cwise_op_round.cc
|
|
cwise_op_rsqrt.cc
|
|
cwise_op_select.cc
|
|
cwise_op_sigmoid.cc
|
|
cwise_op_sign.cc
|
|
cwise_op_sin.cc
|
|
cwise_op_sinh.cc
|
|
cwise_op_sqrt.cc
|
|
cwise_op_square.cc
|
|
cwise_op_squared_difference.cc
|
|
cwise_op_sub.cc
|
|
cwise_op_tan.cc
|
|
cwise_op_tanh.cc
|
|
cwise_op_xdivy.cc
|
|
cwise_op_xlog1py.cc
|
|
cwise_op_xlogy.cc
|
|
cwise_op_zeta.cc
|
|
cwise_ops.h
|
|
cwise_ops_common.cc
|
|
cwise_ops_common.h
|
|
cwise_ops_gpu_common.cu.h
|
|
cwise_ops_gpu_gradients.cu.h
|
|
cwise_ops_gradients.h
|
|
cwise_ops_test.cc
|
|
data_format_ops.cc
|
|
data_format_ops.h
|
|
data_format_ops_gpu.cu.cc
|
|
debug_ops.cc
|
|
debug_ops.h
|
|
debug_ops_gpu.cu.cc
|
|
debug_ops_test.cc
|
|
decode_compressed_op.cc
|
|
decode_csv_op.cc
|
|
decode_padded_raw_op.cc
|
|
decode_proto_op.cc
|
|
decode_raw_op.cc
|
|
decode_wav_op.cc
|
|
decode_wav_op_test.cc
|
|
deep_conv2d.cc
|
|
deep_conv2d.h
|
|
deep_conv2d_test.cc
|
|
dense_update_functor.cc
|
|
dense_update_functor.h
|
|
dense_update_functor_gpu.cu.cc
|
|
dense_update_ops.cc
|
|
depthtospace_op.cc
|
|
depthtospace_op.h
|
|
depthtospace_op_gpu.cu.cc
|
|
depthwise_conv_grad_op.cc
|
|
depthwise_conv_op.cc
|
|
depthwise_conv_op.h
|
|
depthwise_conv_op_gpu.h
|
|
depthwise_conv_op_gpu_bfloat16.cu.cc
|
|
depthwise_conv_op_gpu_double.cu.cc
|
|
depthwise_conv_op_gpu_float.cu.cc
|
|
depthwise_conv_op_gpu_half.cu.cc
|
|
depthwise_conv_ops_test.cc
|
|
dequantize_op.cc
|
|
dequantize_op_test.cc
|
|
deserialize_sparse_string_op.cc
|
|
deserialize_sparse_variant_op.cc
|
|
diag_op.cc
|
|
diag_op.h
|
|
diag_op_gpu.cu.cc
|
|
diag_op_test.cc
|
|
dilation_ops.cc
|
|
dilation_ops.h
|
|
dilation_ops_gpu.cu.cc
|
|
dynamic_partition_op.cc
|
|
dynamic_partition_op_gpu.cu.cc
|
|
dynamic_partition_op_test.cc
|
|
dynamic_stitch_op.cc
|
|
dynamic_stitch_op_gpu.cu.cc
|
|
dynamic_stitch_op_test.cc
|
|
edit_distance_op.cc
|
|
eigen_activations.h
|
|
eigen_activations_test.cc
|
|
eigen_attention.h
|
|
eigen_attention_test.cc
|
|
eigen_backward_cuboid_convolutions.h
|
|
eigen_backward_cuboid_convolutions_test.cc
|
|
eigen_backward_spatial_convolutions.h
|
|
eigen_backward_spatial_convolutions_test.cc
|
|
eigen_benchmark.h
|
|
eigen_benchmark_cpu_test.cc
|
|
eigen_cuboid_convolution.h
|
|
eigen_cuboid_convolutions_test.cc
|
|
eigen_mkldnn_contraction_kernel_test.cc
|
|
eigen_pooling.h
|
|
eigen_pooling_test.cc
|
|
encode_proto_op.cc
|
|
encode_wav_op.cc
|
|
encode_wav_op_test.cc
|
|
example_parsing_ops.cc
|
|
example_parsing_ops_test.cc
|
|
fact_op.cc
|
|
fake_quant_ops.cc
|
|
fake_quant_ops_functor.h
|
|
fake_quant_ops_gpu.cu.cc
|
|
fake_quant_ops_test.cc
|
|
fft_impl.cc
|
|
fft_impl.h
|
|
fft_ops.cc
|
|
fifo_queue.cc
|
|
fifo_queue.h
|
|
fifo_queue_op.cc
|
|
filesystem_ops.cc
|
|
fill_empty_rows_functor.cc
|
|
fill_empty_rows_functor.h
|
|
fill_empty_rows_functor_gpu.cu.cc
|
|
fill_functor.cc
|
|
fill_functor.cu.cc
|
|
fill_functor.h
|
|
fingerprint_op.cc
|
|
fingerprint_op_test.cc
|
|
fixed_length_record_reader_op.cc
|
|
fractional_avg_pool_op.cc
|
|
fractional_max_pool_op.cc
|
|
fractional_pool_common.cc
|
|
fractional_pool_common.h
|
|
function_ops.cc
|
|
function_ops.h
|
|
functional_ops.cc
|
|
fused_batch_norm_ex_op_test.cc
|
|
fused_batch_norm_op.cc
|
|
fused_batch_norm_op.cu.cc
|
|
fused_batch_norm_op.h
|
|
fused_batch_norm_op_test.cc
|
|
fused_eigen_output_kernels.cc
|
|
fused_eigen_output_kernels.h
|
|
gather_functor.cc
|
|
gather_functor.h
|
|
gather_functor_batched.cc
|
|
gather_functor_batched.h
|
|
gather_functor_batched_gpu.cu.cc
|
|
gather_functor_batched_gpu.cu.h
|
|
gather_functor_gpu.cu.cc
|
|
gather_functor_gpu.cu.h
|
|
gather_nd_op.cc
|
|
gather_nd_op.h
|
|
gather_nd_op_cpu_impl.h
|
|
gather_nd_op_cpu_impl_0.cc
|
|
gather_nd_op_cpu_impl_1.cc
|
|
gather_nd_op_cpu_impl_2.cc
|
|
gather_nd_op_cpu_impl_3.cc
|
|
gather_nd_op_cpu_impl_4.cc
|
|
gather_nd_op_cpu_impl_5.cc
|
|
gather_nd_op_cpu_impl_6.cc
|
|
gather_nd_op_cpu_impl_7.cc
|
|
gather_nd_op_gpu.cu.cc
|
|
gather_nd_op_test.cc
|
|
gather_op.cc
|
|
gather_op_test.cc
|
|
gemm_functors.h
|
|
generate_vocab_remapping_op.cc
|
|
gpu_device_array.h
|
|
gpu_device_array_gpu.h
|
|
gpu_prim.h
|
|
gpu_prim_helpers.h
|
|
gpu_prim_helpers_test.cu.cc
|
|
gpu_utils.cc
|
|
gpu_utils.h
|
|
guarantee_const_op.cc
|
|
guarantee_const_op_test.cc
|
|
hinge-loss.h
|
|
histogram_op.cc
|
|
histogram_op.h
|
|
histogram_op_gpu.cu.cc
|
|
host_constant_op.cc
|
|
host_constant_op.h
|
|
identity_n_op.cc
|
|
identity_n_op.h
|
|
identity_n_op_test.cc
|
|
identity_op.cc
|
|
identity_op.h
|
|
identity_op_test.cc
|
|
identity_reader_op.cc
|
|
immutable_constant_op.cc
|
|
immutable_constant_op.h
|
|
immutable_constant_op_test.cc
|
|
in_topk_op.cc
|
|
in_topk_op.h
|
|
in_topk_op_gpu.cu.cc
|
|
in_topk_op_test.cc
|
|
initializable_lookup_table.cc
|
|
initializable_lookup_table.h
|
|
inplace_ops.cc
|
|
inplace_ops_functor.h
|
|
inplace_ops_functor_gpu.cu.cc
|
|
isotonic_regression_op.cc
|
|
isotonic_regression_op_test.cc
|
|
kernel_platform_strings.h
|
|
l2loss_op.cc
|
|
l2loss_op.h
|
|
l2loss_op_gpu.cu.cc
|
|
linalg_ops_common.h
|
|
list_kernels.cc
|
|
list_kernels.cu.cc
|
|
list_kernels.h
|
|
listdiff_op.cc
|
|
lmdb_reader_op.cc
|
|
load_and_remap_matrix_op.cc
|
|
logging_ops.cc
|
|
logging_ops.h
|
|
logging_ops_test.cc
|
|
logistic-loss.h
|
|
lookup_ops_test.cc
|
|
lookup_table_init_op.cc
|
|
lookup_table_init_op.h
|
|
lookup_table_op.cc
|
|
lookup_table_op.h
|
|
lookup_util.cc
|
|
lookup_util.h
|
|
loss.h
|
|
loss_test.cc
|
|
lrn_op.cc
|
|
lrn_op_test.cc
|
|
map_kernels.cc
|
|
map_kernels.h
|
|
map_stage_op.cc
|
|
matching_files_op.cc
|
|
matmul_op.h
|
|
matmul_op_complex.cc
|
|
matmul_op_fused.cc
|
|
matmul_op_impl.h
|
|
matmul_op_real.cc
|
|
matmul_op_test.cc
|
|
matmul_util.cc
|
|
matmul_util.h
|
|
maxpooling_op.cc
|
|
maxpooling_op.h
|
|
maxpooling_op_gpu.cu.cc
|
|
maxpooling_op_gpu.h
|
|
merge_v2_checkpoints_op_test.cc
|
|
meta_support.cc
|
|
meta_support.h
|
|
mfcc.cc
|
|
mfcc.h
|
|
mfcc_dct.cc
|
|
mfcc_dct.h
|
|
mfcc_dct_test.cc
|
|
mfcc_mel_filterbank.cc
|
|
mfcc_mel_filterbank.h
|
|
mfcc_mel_filterbank_test.cc
|
|
mfcc_op.cc
|
|
mfcc_op_test.cc
|
|
mfcc_test.cc
|
|
multinomial_op.cc
|
|
multinomial_op.h
|
|
multinomial_op_gpu.cu.cc
|
|
multinomial_op_test.cc
|
|
mutex_ops.cc
|
|
nccl_ops.cc
|
|
nextafter_op.cc
|
|
nextafter_op.h
|
|
nextafter_op_gpu.cu.cc
|
|
nn_ops_test.cc
|
|
no_op.cc
|
|
no_op.h
|
|
nth_element_op.cc
|
|
nth_element_op.h
|
|
numeric_options_utils.h
|
|
one_hot_op.cc
|
|
one_hot_op.h
|
|
one_hot_op_gpu.cu.cc
|
|
one_hot_op_test.cc
|
|
ops_testutil.cc
|
|
ops_testutil.h
|
|
ops_testutil_test.cc
|
|
ops_util.h
|
|
ops_util_test.cc
|
|
pack_op.cc
|
|
pad_op.cc
|
|
pad_op.h
|
|
pad_op_gpu.cu.cc
|
|
padding_fifo_queue.cc
|
|
padding_fifo_queue.h
|
|
padding_fifo_queue_op.cc
|
|
parameterized_truncated_normal_op.cc
|
|
parameterized_truncated_normal_op.h
|
|
parameterized_truncated_normal_op_gpu.cu.cc
|
|
parameterized_truncated_normal_op_test.cc
|
|
parse_tensor_op.cc
|
|
parse_tensor_test.cc
|
|
partitioned_function_ops.cc
|
|
partitioned_function_ops.h
|
|
poisson-loss.h
|
|
pooling_ops_3d.cc
|
|
pooling_ops_3d.h
|
|
pooling_ops_3d_gpu.cu.cc
|
|
pooling_ops_3d_gpu.h
|
|
pooling_ops_common.cc
|
|
pooling_ops_common.h
|
|
pooling_ops_common_gpu.h
|
|
population_count_op.cc
|
|
population_count_op.h
|
|
population_count_op_gpu.cu.cc
|
|
priority_queue.cc
|
|
priority_queue.h
|
|
priority_queue_op.cc
|
|
quantization_utils.cc
|
|
quantization_utils.h
|
|
quantization_utils_test.cc
|
|
quantize_and_dequantize_op.cc
|
|
quantize_and_dequantize_op.h
|
|
quantize_and_dequantize_op_gpu.cu.cc
|
|
quantize_and_dequantize_op_test.cc
|
|
quantize_down_and_shrink_range.cc
|
|
quantize_down_and_shrink_range_op_test.cc
|
|
quantize_op.cc
|
|
quantize_op_test.cc
|
|
quantized_activation_ops.cc
|
|
quantized_activation_ops_test.cc
|
|
quantized_add_op.cc
|
|
quantized_add_op_test.cc
|
|
quantized_batch_norm_op.cc
|
|
quantized_batch_norm_op_test.cc
|
|
quantized_bias_add_op.cc
|
|
quantized_bias_add_op_test.cc
|
|
quantized_concat_op.cc
|
|
quantized_concat_op_test.cc
|
|
quantized_conv_ops.cc
|
|
quantized_conv_ops_test.cc
|
|
quantized_instance_norm.cc
|
|
quantized_instance_norm_test.cc
|
|
quantized_matmul_op.cc
|
|
quantized_matmul_op_test.cc
|
|
quantized_mul_op.cc
|
|
quantized_mul_op_test.cc
|
|
quantized_pooling_ops.cc
|
|
quantized_pooling_ops_test.cc
|
|
quantized_reshape_op.cc
|
|
quantized_reshape_op_test.cc
|
|
quantized_resize_bilinear_op.cc
|
|
quantized_resize_bilinear_op_test.cc
|
|
queue_base.cc
|
|
queue_base.h
|
|
queue_op.cc
|
|
queue_op.h
|
|
queue_ops.cc
|
|
ragged_cross_op.cc
|
|
ragged_fill_empty_rows_op.cc
|
|
ragged_fill_empty_rows_op_test.cc
|
|
ragged_gather_op.cc
|
|
ragged_gather_op_test.cc
|
|
ragged_range_op.cc
|
|
ragged_range_op_test.cc
|
|
ragged_tensor_from_variant_op.cc
|
|
ragged_tensor_from_variant_op_test.cc
|
|
ragged_tensor_to_sparse_kernel.cc
|
|
ragged_tensor_to_sparse_kernel_test.cc
|
|
ragged_tensor_to_tensor_op.cc
|
|
ragged_tensor_to_tensor_op_test.cc
|
|
ragged_tensor_to_variant_op.cc
|
|
ragged_tensor_to_variant_op_large_data_test.cc
|
|
ragged_tensor_to_variant_op_test.cc
|
|
ragged_tensor_to_variant_op_test.h
|
|
ragged_tensor_variant.cc
|
|
ragged_tensor_variant.h
|
|
random_binomial_op.cc
|
|
random_binomial_op.h
|
|
random_binomial_op_test.cc
|
|
random_index_shuffle.cc
|
|
random_index_shuffle.h
|
|
random_index_shuffle_ops.cc
|
|
random_index_shuffle_test.cc
|
|
random_op.cc
|
|
random_op.h
|
|
random_op_cpu.h
|
|
random_op_gpu.cu.cc
|
|
random_op_gpu.h
|
|
random_op_test.cc
|
|
random_ops_util.h
|
|
random_poisson_op.cc
|
|
random_poisson_op.h
|
|
random_poisson_op_test.cc
|
|
random_shuffle_op.cc
|
|
random_shuffle_queue_op.cc
|
|
range_sampler.cc
|
|
range_sampler.h
|
|
range_sampler_test.cc
|
|
reader_ops.cc
|
|
record_input_op.cc
|
|
record_yielder.cc
|
|
record_yielder.h
|
|
reduce_join_op.cc
|
|
reduction_gpu_kernels.cu.h
|
|
reduction_ops.h
|
|
reduction_ops_all.cc
|
|
reduction_ops_any.cc
|
|
reduction_ops_common.cc
|
|
reduction_ops_common.h
|
|
reduction_ops_common_gpu.h
|
|
reduction_ops_euclidean.cc
|
|
reduction_ops_gpu_bfloat16.cu.cc
|
|
reduction_ops_gpu_bool.cu.cc
|
|
reduction_ops_gpu_complex128.cu.cc
|
|
reduction_ops_gpu_complex64.cu.cc
|
|
reduction_ops_gpu_double.cu.cc
|
|
reduction_ops_gpu_float.cu.cc
|
|
reduction_ops_gpu_int.cu.cc
|
|
reduction_ops_half_mean_sum.cu.cc
|
|
reduction_ops_half_prod_max_min.cu.cc
|
|
reduction_ops_max.cc
|
|
reduction_ops_mean.cc
|
|
reduction_ops_min.cc
|
|
reduction_ops_prod.cc
|
|
reduction_ops_sum.cc
|
|
reduction_ops_test.cc
|
|
redux_functor.h
|
|
reference_gemm.h
|
|
regex_full_match_op.cc
|
|
regex_replace_op.cc
|
|
regex_replace_op_test.cc
|
|
relu_op.cc
|
|
relu_op.h
|
|
relu_op_functor.h
|
|
relu_op_gpu.cu.cc
|
|
requantization_range_op.cc
|
|
requantization_range_op_test.cc
|
|
requantize.cc
|
|
requantize_op_test.cc
|
|
reshape_op.cc
|
|
reshape_op.h
|
|
reshape_util.cc
|
|
reshape_util.h
|
|
reshape_util_gpu.cu.cc
|
|
resource_ops_test.cc
|
|
resource_variable_ops.cc
|
|
resource_variable_ops.h
|
|
resource_variable_util.cc
|
|
resource_variable_util.h
|
|
restore_op.cc
|
|
restore_op_test.cc
|
|
restore_v2_op_test.cc
|
|
reverse_op.cc
|
|
reverse_op.h
|
|
reverse_op_gpu.cu.cc
|
|
reverse_op_test.cc
|
|
reverse_sequence_op.cc
|
|
reverse_sequence_op.h
|
|
reverse_sequence_op_gpu.cu.cc
|
|
roll_op.cc
|
|
roll_op.h
|
|
roll_op_gpu.cu.cc
|
|
roll_op_test.cc
|
|
save_op.cc
|
|
save_op_test.cc
|
|
save_restore_tensor.cc
|
|
save_restore_tensor.h
|
|
save_restore_v2_ops.cc
|
|
save_v2_op_test.cc
|
|
scan_ops.cc
|
|
scan_ops.h
|
|
scan_ops_gpu.h
|
|
scan_ops_gpu_bfloat16.cu.cc
|
|
scan_ops_gpu_double.cu.cc
|
|
scan_ops_gpu_float.cu.cc
|
|
scan_ops_gpu_half.cu.cc
|
|
scan_ops_gpu_int.cu.cc
|
|
scan_ops_test.cc
|
|
scatter_functor.cc
|
|
scatter_functor.h
|
|
scatter_functor_gpu.cu.cc
|
|
scatter_functor_gpu.cu.h
|
|
scatter_nd_op.cc
|
|
scatter_nd_op.h
|
|
scatter_nd_op_cpu_impl.h
|
|
scatter_nd_op_cpu_impl_0.cc
|
|
scatter_nd_op_cpu_impl_1.cc
|
|
scatter_nd_op_cpu_impl_2.cc
|
|
scatter_nd_op_cpu_impl_3.cc
|
|
scatter_nd_op_cpu_impl_4.cc
|
|
scatter_nd_op_cpu_impl_5.cc
|
|
scatter_nd_op_cpu_impl_6.cc
|
|
scatter_nd_op_cpu_impl_7.cc
|
|
scatter_nd_op_gpu.cu.cc
|
|
scatter_nd_op_test.cc
|
|
scatter_nd_util.cc
|
|
scatter_nd_util.h
|
|
scatter_op.cc
|
|
scatter_op_gpu.cu.cc
|
|
scatter_op_test.cc
|
|
scoped_allocator_ops.cc
|
|
scoped_allocator_ops_test.cc
|
|
sdca_internal.cc
|
|
sdca_internal.h
|
|
sdca_ops.cc
|
|
sdca_ops_test.cc
|
|
searchsorted_op.cc
|
|
searchsorted_op.h
|
|
searchsorted_op_gpu.cu.cc
|
|
segment_reduction_ops.h
|
|
segment_reduction_ops_gpu.cu.h
|
|
segment_reduction_ops_gpu_0.cu.cc
|
|
segment_reduction_ops_gpu_1.cu.cc
|
|
segment_reduction_ops_gpu_2.cu.cc
|
|
segment_reduction_ops_impl.h
|
|
segment_reduction_ops_impl_1.cc
|
|
segment_reduction_ops_impl_2.cc
|
|
segment_reduction_ops_impl_3.cc
|
|
segment_reduction_ops_impl_3b.cc
|
|
segment_reduction_ops_impl_4.cc
|
|
segment_reduction_ops_impl_5.cc
|
|
segment_reduction_ops_test.cc
|
|
sendrecv_ops.cc
|
|
sendrecv_ops.h
|
|
sendrecv_ops_test.cc
|
|
sequence_ops.cc
|
|
sequence_ops.h
|
|
sequence_ops_gpu.cu.cc
|
|
sequence_ops_test.cc
|
|
serialize_sparse_op.cc
|
|
session_ops.cc
|
|
set_kernels.cc
|
|
shape_ops.cc
|
|
shape_ops.h
|
|
shape_ops_test.cc
|
|
shuffle_common.h
|
|
slice_op.cc
|
|
slice_op.h
|
|
slice_op_cpu_impl.h
|
|
slice_op_cpu_impl_1.cc
|
|
slice_op_cpu_impl_2.cc
|
|
slice_op_cpu_impl_3.cc
|
|
slice_op_cpu_impl_4.cc
|
|
slice_op_cpu_impl_5.cc
|
|
slice_op_cpu_impl_6.cc
|
|
slice_op_cpu_impl_7.cc
|
|
slice_op_cpu_impl_8.cc
|
|
slice_op_gpu.cu.cc
|
|
slice_op_test.cc
|
|
smooth-hinge-loss.h
|
|
snapshot_op.cc
|
|
snapshot_op.h
|
|
snapshot_op_gpu.cu.cc
|
|
sobol_op.cc
|
|
softmax_op.cc
|
|
softmax_op_functor.h
|
|
softmax_op_gpu.cu.cc
|
|
softplus_op.cc
|
|
softplus_op.h
|
|
softplus_op_gpu.cu.cc
|
|
softsign_op.cc
|
|
softsign_op.h
|
|
softsign_op_gpu.cu.cc
|
|
spacetobatch_benchmark_test.cc
|
|
spacetobatch_functor.cc
|
|
spacetobatch_functor.h
|
|
spacetobatch_functor_gpu.cu.cc
|
|
spacetobatch_op.cc
|
|
spacetodepth_op.cc
|
|
spacetodepth_op.h
|
|
spacetodepth_op_gpu.cu.cc
|
|
sparse_add_grad_op.cc
|
|
sparse_add_op.cc
|
|
sparse_add_op_test.cc
|
|
sparse_concat_op.cc
|
|
sparse_concat_op.h
|
|
sparse_concat_op_gpu.cu.cc
|
|
sparse_conditional_accumulator.h
|
|
sparse_conditional_accumulator_op.cc
|
|
sparse_cross_op.cc
|
|
sparse_dense_binary_op_shared.cc
|
|
sparse_dense_binary_op_shared_test.cc
|
|
sparse_fill_empty_rows_op.cc
|
|
sparse_fill_empty_rows_op_test.cc
|
|
sparse_matmul_op.cc
|
|
sparse_matmul_op.h
|
|
sparse_matmul_op_test.cc
|
|
sparse_reduce_op.cc
|
|
sparse_reduce_sum_op_test.cc
|
|
sparse_reorder_op.cc
|
|
sparse_reorder_op.h
|
|
sparse_reorder_op_gpu.cu.cc
|
|
sparse_reshape_op.cc
|
|
sparse_slice_grad_op.cc
|
|
sparse_slice_grad_op.h
|
|
sparse_slice_grad_op_gpu.cu.cc
|
|
sparse_slice_op.cc
|
|
sparse_slice_op.h
|
|
sparse_slice_op_gpu.cu.cc
|
|
sparse_softmax_op.cc
|
|
sparse_sparse_binary_op_shared.cc
|
|
sparse_split_op.cc
|
|
sparse_split_op.h
|
|
sparse_split_op_gpu.cu.cc
|
|
sparse_tensor_dense_add_op.cc
|
|
sparse_tensor_dense_add_op.h
|
|
sparse_tensor_dense_matmul_op.cc
|
|
sparse_tensor_dense_matmul_op.h
|
|
sparse_tensor_dense_matmul_op_gpu.cu.cc
|
|
sparse_tensor_dense_matmul_op_test.cc
|
|
sparse_tensors_map_ops.cc
|
|
sparse_to_dense_op.cc
|
|
sparse_to_dense_op_gpu.cu.cc
|
|
sparse_to_dense_op_gpu.h
|
|
sparse_to_dense_op_test.cc
|
|
sparse_utils.cc
|
|
sparse_utils.h
|
|
sparse_utils_test.cc
|
|
sparse_xent_op.cc
|
|
sparse_xent_op.h
|
|
sparse_xent_op_gpu.cu.cc
|
|
sparse_xent_op_test.cc
|
|
spectrogram.cc
|
|
spectrogram.h
|
|
spectrogram_convert_test_data.cc
|
|
spectrogram_op.cc
|
|
spectrogram_op_test.cc
|
|
spectrogram_test.cc
|
|
spectrogram_test_utils.cc
|
|
spectrogram_test_utils.h
|
|
split_lib.h
|
|
split_lib_cpu.cc
|
|
split_lib_gpu.cu.cc
|
|
split_lib_gpu.h
|
|
split_op.cc
|
|
split_op_test.cc
|
|
split_v_op.cc
|
|
split_v_op_test.cc
|
|
squared-loss.h
|
|
stack.cc
|
|
stack.h
|
|
stack_ops.cc
|
|
stage_op.cc
|
|
stateful_random_ops.cc
|
|
stateful_random_ops.h
|
|
stateful_random_ops_cpu_gpu.h
|
|
stateful_random_ops_gpu.cu.cc
|
|
stateless_random_gamma_op.cc
|
|
stateless_random_gamma_op.h
|
|
stateless_random_gamma_op_gpu.cu.cc
|
|
stateless_random_ops.cc
|
|
stateless_random_ops.h
|
|
stateless_random_ops_v2.cc
|
|
stateless_random_ops_v2.h
|
|
stateless_random_ops_v2_util.h
|
|
stateless_shuffle.cc
|
|
stochastic_cast_op.cc
|
|
stochastic_cast_op.h
|
|
stochastic_cast_op_test.cc
|
|
strided_slice_op.cc
|
|
strided_slice_op.h
|
|
strided_slice_op_define_grad.cc
|
|
strided_slice_op_gpu_bool.cu.cc
|
|
strided_slice_op_gpu_complex.cu.cc
|
|
strided_slice_op_gpu_impl.h
|
|
strided_slice_op_gpu_int.cu.cc
|
|
strided_slice_op_gpu_number_types.cu.cc
|
|
strided_slice_op_impl.h
|
|
strided_slice_op_inst_0.cc
|
|
strided_slice_op_inst_1.cc
|
|
strided_slice_op_inst_2.cc
|
|
strided_slice_op_inst_3.cc
|
|
strided_slice_op_inst_4.cc
|
|
strided_slice_op_inst_5.cc
|
|
strided_slice_op_inst_6.cc
|
|
strided_slice_op_inst_7.cc
|
|
strided_slice_op_inst_8.cc
|
|
strided_slice_op_test.cc
|