File tree

• test
  • CMakeLists.txt
  • asp
    • CMakeLists.txt
  • autograd
    • CMakeLists.txt
  • book
    • CMakeLists.txt
  • collective/fleet
    • CMakeLists.txt
  • contrib
    • CMakeLists.txt
  • cpp_extension
    • CMakeLists.txt
  • cpp/prim
    • CMakeLists.txt
  • custom_op
    • CMakeLists.txt
  • custom_runtime
    • CMakeLists.txt
  • deprecated
    • CMakeLists.txt
    • amp
      • CMakeLists.txt
      • test_collect_operator_stats.py
    • asp
      • CMakeLists.txt
      • test_asp_customized_pruning.py
      • test_asp_optimize_dynamic.py
      • test_asp_optimize_static.py
      • test_asp_pruning_dynamic.py
      • test_asp_pruning_static.py
      • test_asp_save_load.py
    • autograd
      • CMakeLists.txt
      • config.py
      • test_autograd_functional_static.py
      • utils.py
    • book
      • CMakeLists.txt
      • test_fit_a_line.py
      • test_image_classification.py
      • test_recognize_digits.py
      • test_recommender_system.py
      • test_word2vec_book.py
    • collective
      • CMakeLists.txt
      • fleet
        • CMakeLists.txt
        • test_communicator_sync.py
        • test_fleet_fp16_allreduce_meta_optimizer.py
        • test_fleet_meta_optimizer_base.py
        • test_fleet_static_mp_layers.py
        • test_fleet_utils.py
    • contrib
      • CMakeLists.txt
      • test_bf16_utils.py
      • test_image_classification_fp16.py
    • cpp_extension
      • CMakeLists.txt
      • custom_power.h
      • mix_relu_and_extension.cc
      • mix_relu_and_extension_setup.py
      • test_mixed_extension_setup.py
    • cpp
      • CMakeLists.txt
      • prim
        • CMakeLists.txt
        • test_static_prim.cc
    • custom_op
      • CMakeLists.txt
      • custom_cast_op.cc
      • custom_raw_op_kernel_op.cc
      • custom_raw_op_kernel_op.cu
      • custom_raw_op_kernel_op.h
      • custom_raw_op_kernel_op_setup.py
      • test_custom_cast_op_jit.py
      • test_custom_raw_op_kernel_op.py
      • utils.py
    • custom_runtime
      • CMakeLists.txt
      • custom_op.cc
      • test_custom_cpu_plugin.py
      • test_custom_cpu_to_static.py
      • test_custom_op_setup.py
    • distributed_passes
      • CMakeLists.txt
      • test_ps_trainer_pass.py
    • distribution
      • CMakeLists.txt
      • parameterize.py
      • test_distribution.py
      • test_distribution_bernoulli_static.py
      • test_distribution_beta_static.py
      • test_distribution_binomial_static.py
      • test_distribution_categorical.py
      • test_distribution_cauchy_static.py
      • test_distribution_continuous_bernoulli_static.py
      • test_distribution_exponential_static.py
      • test_distribution_gamma_static.py
      • test_distribution_geometric_static.py
      • test_distribution_gumbel_static.py
      • test_distribution_multinomial.py
      • test_distribution_multinomial_static.py
      • test_distribution_multivariate_normal_static.py
      • test_distribution_poisson_static.py
      • test_distribution_transform_static.py
      • test_distribution_uniform.py
    • fft
      • CMakeLists.txt
      • test_spectral_op.py
    • ir
      • CMakeLists.txt
      • inference
        • CMakeLists.txt
        • test_mul_gru_fuse_pass.py
        • test_mul_lstm_fuse_pass.py
      • pass_test.py
      • pir
        • CMakeLists.txt
        • test_build_op.py
        • test_ir_backward.py
        • test_ir_pybind.py
        • test_ir_vjp.py
        • test_pass_manager.py
        • test_special_op_translator.py
        • test_standalone_pir.py
        • translator
          • CMakeLists.txt
          • test_all_reduce_translator.py
          • test_barrier_translator.py
          • test_c_allreduce_min_translator.py
          • test_c_allreduce_prod_translator.py
          • test_c_reduce_max_translator.py
          • test_c_reduce_min_translator.py
          • test_c_reduce_prod_translator.py
          • test_c_scatter_translator.py
          • test_c_split_translator.py
          • test_dgc_momentum_translator.py
          • test_distributed_fused_lamb.py
          • test_distributed_fused_lamb_init.py
          • test_distributed_lookup_table_translate.py
          • test_distributed_push_sparse_translator.py
          • test_global_scatter_translator.py
          • test_limit_by_capacity_translator.py
          • test_nop_translator.py
          • test_op_translator.py
          • test_partial_allgather_translator.py
          • test_partial_recv_translator.py
          • test_partial_send_translator.py
          • test_prune_gate_by_capacity_translator.py
          • test_push_dense_translator.py
          • test_random_routing_translator.py
      • test_ir_embedding_eltwise_layernorm_fuse_pass.py
      • test_ir_fc_fuse_pass.py
      • test_ir_generate_pass.py
      • test_ir_graph_to_program_pass.py
      • test_ir_preln_residual_bias_fuse_pass.py
      • test_ir_skip_layernorm_pass.py
      • test_ir_yolo_box_pass.py
      • test_op_input_grad_semantic.py
    • legacy_test
      • CMakeLists.txt
      • auto_parallel_op_test.py
      • check_nan_inf_backward_stack.py
      • check_nan_inf_backward_static_stack.py
      • check_nan_inf_base.py
      • check_nan_inf_base_dygraph.py
      • dist_fleet_ctr.py
      • dist_test.sh
      • run_server_for_communicator_geo.py
      • test_accuracy_op.py
      • test_adam_op.py
      • test_adamax_api.py
      • test_adamw_op.py
      • test_adaptive_avg_pool2d.py
      • test_adaptive_max_pool1d.py
      • test_adaptive_max_pool2d.py
      • test_adaptive_max_pool3d.py
      • test_add_position_encoding_op.py
      • test_add_reader_dependency.py
      • test_addmm_op.py
      • test_affine_channel_op.py
      • test_affine_grid_op.py
      • test_allclose_layer.py
      • test_allclose_op.py
      • test_apply.py
      • test_apply_pass_to_program.py
      • test_arg_min_max_op.py
      • test_arg_min_max_v2_op.py
      • test_argsort_op.py
      • test_array_read_write_op.py
      • test_assign_op.py
      • test_atan2_op.py
      • test_attribute_var.py
      • test_auc_op.py
      • test_auto_parallel_completion.py
      • test_auto_parallel_completion_gpt.py
      • test_auto_parallel_cost_model.py
      • test_auto_parallel_dist_tensor.py
      • test_auto_parallel_partitioner.py
      • test_auto_parallel_partitioner_gpt.py
      • test_auto_parallel_reshard.py
      • test_auto_parallel_reshard_dpmppp.py
      • test_auto_parallel_reshard_mppp.py
      • test_auto_parallel_searcher.py
      • test_auto_search_dist_matmul_op.py
      • test_auto_search_dist_op.py
      • test_avoid_twice_initialization.py
      • test_backward.py
      • test_backward_infer_var_data_type_shape.py
      • test_base_layer.py
      • test_batch_norm_op.py
      • test_bce_loss.py
      • test_bfgs.py
      • test_bicubic_interp_op.py
      • test_bicubic_interp_v2_op.py
      • test_bilateral_slice_op.py
      • test_bilinear_interp_op.py
      • test_bilinear_tensor_product_op.py
      • test_bincount_op.py
      • test_bitwise_shift_op.py
      • test_block_rename_var.py
      • test_bmm_op.py
      • test_broadcast_tensors_op.py
      • test_broadcast_to_op.py
      • test_calc_gradient.py
      • test_callback_early_stop.py
      • test_cast_op.py
      • test_channel_shuffle.py
      • test_cholesky_solve_op.py
      • test_clip_grad_norm_.py
      • test_clip_grad_value_.py
      • test_communicator_async.py
      • test_communicator_geo.py
      • test_compare_op.py
      • test_compiled_program.py
      • test_complex_abs.py
      • test_complex_op.py
      • test_complex_variable.py
      • test_complex_view_op.py
      • test_conditional_block.py
      • test_conj_op.py
      • test_conv1d_transpose_layer.py
      • test_conv2d_api.py
      • test_conv2d_layer.py
      • test_conv2d_op_depthwise_conv.py
      • test_conv2d_transpose_layer.py
      • test_conv2d_transpose_op_depthwise_conv.py
      • test_conv3d_layer.py
      • test_conv3d_transpose_layer.py
      • test_conv3d_transpose_part2_op.py
      • test_conv_nn_grad.py
      • test_copysign_op.py
      • test_cost_model.py
      • test_crop_op.py
      • test_crop_tensor_op.py
      • test_cross_entropy2_op.py
      • test_cross_entropy_op.py
      • test_cross_op.py
      • test_cummax_op.py
      • test_cummin_op.py
      • test_cumprod_op.py
      • test_cumsum_op.py
      • test_data_feeder.py
      • test_data_norm_op.py
      • test_dataloader_early_reset.py
      • test_dataloader_keep_order.py
      • test_dataloader_unkeep_order.py
      • test_dataset.py
      • test_dataset_dataloader.py
      • test_decoupled_py_reader.py
      • test_decoupled_py_reader_data_check.py
      • test_deform_conv2d.py
      • test_deformable_conv_op.py
      • test_deformable_conv_v1_op.py
      • test_deprecated_memory_optimize_interfaces.py
      • test_desc_clone.py
      • test_detection.py
      • test_determinant_op.py
      • test_device_guard.py
      • test_diag_v2.py
      • test_diagonal_op.py
      • test_digamma_op.py
      • test_dist_fleet_a_sync_optimizer_async.py
      • test_dist_fleet_a_sync_optimizer_auto.py
      • test_dist_fleet_a_sync_optimizer_auto_async.py
      • test_dist_fleet_a_sync_optimizer_auto_geo.py
      • test_dist_fleet_a_sync_optimizer_geo.py
      • test_dist_fleet_a_sync_optimizer_sync.py
      • test_dist_fleet_decay.py
      • test_dist_fleet_geo.py
      • test_dist_fleet_heter_program.py
      • test_dist_fleet_ps.py
      • test_dist_fleet_ps10.py
      • test_dist_fleet_ps13.py
      • test_dist_fleet_ps2.py
      • test_dist_fleet_ps3.py
      • test_dist_fleet_ps4.py
      • test_dist_fleet_ps5.py
      • test_dist_fleet_ps6.py
      • test_dist_fleet_ps7.py
      • test_dist_fleet_ps8.py
      • test_dist_fleet_ps9.py
      • test_dist_fleet_trainer_desc_config.py
      • test_dist_sparse_tensor_load_adagrad.py
      • test_dist_sparse_tensor_load_adam.py
      • test_dist_sparse_tensor_load_ftrl.py
      • test_dist_sparse_tensor_load_momentum.py
      • test_dist_sparse_tensor_load_rmsprop.py
      • test_dist_sparse_tensor_load_sgd.py
      • test_dist_tree_index.py
      • test_downpoursgd.py
      • test_dygraph_multi_forward.py
      • test_eager_deletion_delete_vars.py
      • test_eager_run_program.py
      • test_eigh_op.py
      • test_eigvalsh_op.py
      • test_einsum_op.py
      • test_elementwise_floordiv_op.py
      • test_elementwise_gradient_op.py
      • test_elementwise_heaviside_op.py
      • test_elementwise_mod_op.py
      • test_elementwise_mul_op.py
      • test_elementwise_pow_op.py
      • test_ema.py
      • test_ema_fleet.py
      • test_embedding_id_stop_gradient.py
      • test_entry_attr.py
      • test_entry_attr2.py
      • test_erf_op.py
      • test_error_clip.py
      • test_executor_and_mul.py
      • test_executor_and_use_program_cache.py
      • test_executor_check_feed.py
      • test_executor_check_fetch_list.py
      • test_executor_feed_non_tensor.py
      • test_expand_as_v2_op.py
      • test_expand_op.py
      • test_expand_v2_op.py
      • test_eye_op.py
      • test_fc_op.py
      • test_feed_data_check_shape_type.py
      • test_fetch_lod_tensor_array.py
      • test_fetch_var.py
      • test_fill_any_op.py
      • test_fill_constant_op.py
      • test_fill_diagonal_tensor_op.py
      • test_fill_zeros_like2_op.py
      • test_flatten2_op.py
      • test_flatten_contiguous_range_op.py
      • test_fleet.py
      • test_fleet_base.py
      • test_fleet_base_2.py
      • test_fleet_base_3.py
      • test_fleet_metric.py
      • test_fleet_nocvm_1.py
      • test_fleet_unitaccessor.py
      • test_fleet_util.py
      • test_flip.py
      • test_fmax_op.py
      • test_fmin_op.py
      • test_fold_op.py
      • test_fractional_max_pool2d_api.py
      • test_fractional_max_pool2d_op.py
      • test_fractional_max_pool3d_api.py
      • test_fractional_max_pool3d_op.py
      • test_frame_op.py
      • test_full_like_op.py
      • test_functional_conv2d.py
      • test_functional_conv2d_transpose.py
      • test_functional_conv3d.py
      • test_functional_conv3d_transpose.py
      • test_fuse_bn_act_pass.py
      • test_fuse_elewise_add_act_pass.py
      • test_gammaln_op.py
      • test_gather_nd_op.py
      • test_gather_tree_op.py
      • test_gaussian_random_op.py
      • test_generator_dataloader.py
      • test_get_inputs_outputs_in_block.py
      • test_get_tensor_from_selected_rows_op.py
      • test_gradient_clip.py
      • test_graph_send_recv_op.py
      • test_graph_send_ue_recv_op.py
      • test_graph_send_uv_op.py
      • test_grid_sampler_op.py
      • test_gru_op.py
      • test_gru_rnn_op.py
      • test_gru_unit_op.py
      • test_gumbel_softmax_op.py
      • test_hinge_loss_op.py
      • test_histogramdd_op.py
      • test_householder_product.py
      • test_hsigmoid_op.py
      • test_huber_loss_op.py
      • test_hypot.py
      • test_identity_loss_op.py
      • test_iinfo_and_finfo.py
      • test_im2sequence_op.py
      • test_image_classification_layer.py
      • test_imperative_base.py
      • test_imperative_double_grad.py
      • test_imperative_framework.py
      • test_imperative_gan.py
      • test_imperative_load_static_param.py
      • test_imperative_lod_tensor_to_selected_rows.py
      • test_imperative_mnist.py
      • test_imperative_mnist_sorted_gradient.py
      • test_imperative_ocr_attention_model.py
      • test_imperative_optimizer_v2.py
      • test_imperative_ptb_rnn.py
      • test_imperative_ptb_rnn_sorted_gradient.py
      • test_imperative_recurrent_usage.py
      • test_imperative_reinforcement.py
      • test_imperative_selected_rows_to_lod_tensor.py
      • test_imperative_star_gan_with_gradient_penalty.py
      • test_imperative_transformer_sorted_gradient.py
      • test_index_add_op.py
      • test_index_fill.py
      • test_index_sample_op.py
      • test_index_select_op.py
      • test_infer_no_need_buffer_slots.py
      • test_inference_api.py
      • test_inference_model_io.py
      • test_initializer.py
      • test_initializer_nn.py
      • test_inplace.py
      • test_inplace_addto_strategy.py
      • test_inplace_softmax_with_cross_entropy.py
      • test_input_spec.py
      • test_install_check.py
      • test_instance_norm_op.py
      • test_instance_norm_op_v2.py
      • test_inverse_op.py
      • test_io_save_load.py
      • test_is_integer.py
      • test_isclose_op.py
      • test_jit_layer.py
      • test_kldiv_loss_op.py
      • test_kron_op.py
      • test_kthvalue_op.py
      • test_l1_norm_op.py
      • test_label_smooth_op.py
      • test_layer_norm_op.py
      • test_layers.py
      • test_lazy_init.py
      • test_lbfgs.py
      • test_learning_rate_scheduler.py
      • test_lerp_op.py
      • test_lgamma_op.py
      • test_linalg_cond.py
      • test_linalg_matrix_exp.py
      • test_linear_interp_op.py
      • test_linear_interp_v2_op.py
      • test_linspace.py
      • test_load_state_dict_from_old_format.py
      • test_lod_reset_op.py
      • test_lod_tensor.py
      • test_lod_tensor_array.py
      • test_log_loss_op.py
      • test_log_softmax.py
      • test_logcumsumexp_op.py
      • test_logspace.py
      • test_logsumexp.py
      • test_lookup_table_bf16_op.py
      • test_lookup_table_op.py
      • test_lookup_table_v2_bf16_op.py
      • test_lookup_table_v2_op.py
      • test_lr_scheduler.py
      • test_lrn_op.py
      • test_lstm_op.py
      • test_lu_op.py
      • test_lu_unpack_op.py
      • test_masked_scatter.py
      • test_masked_select_op.py
      • test_math_op_patch.py
      • test_math_op_patch_var_base.py
      • test_matmul_op.py
      • test_matmul_v2_op.py
      • test_matrix_power_op.py
      • test_max_op.py
      • test_maxout_op.py
      • test_memory_reuse_exclude_feed_var.py
      • test_merged_momentum_op.py
      • test_meshgrid_op.py
      • test_metrics.py
      • test_min_op.py
      • test_model.py
      • test_modified_huber_loss_op.py
      • test_momentum_op.py
      • test_mul_op.py
      • test_multi_dot_op.py
      • test_multinomial_op.py
      • test_multiprocess_dataloader_static.py
      • test_multiprocess_reader_exception.py
      • test_mv_op.py
      • test_name_scope.py
      • test_nan_inf.py
      • test_nce.py
      • test_nearest_interp_op.py
      • test_nearest_interp_v2_op.py
      • test_nll_loss.py
      • test_nn_functional_embedding_static.py
      • test_nn_functional_hot_op.py
      • test_nn_matmul_v2_grad.py
      • test_nn_sigmoid_op.py
      • test_nonzero_api.py
      • test_norm_all.py
      • test_one_hot_v2_op.py
      • test_ops_nms.py
      • test_optimizer.py
      • test_optimizer_in_control_flow.py
      • test_overlap_add_op.py
      • test_pad3d_op.py
      • test_paddle_save_load_binary.py
      • test_parameter.py
      • test_partial_concat_op.py
      • test_partial_sum_op.py
      • test_pass_builder.py
      • test_pixel_shuffle_op.py
      • test_pixel_unshuffle.py
      • test_pool2d_op.py
      • test_pool3d_op.py
      • test_pow.py
      • test_prelu_op.py
      • test_pretrained_model.py
      • test_print_op.py
      • test_prod_op.py
      • test_program.py
      • test_program_code.py
      • test_program_converter.py
      • test_program_prune_backward.py
      • test_program_to_string.py
      • test_prune.py
      • test_psroi_pool_op.py
      • test_pull_gpups_sparse_op.py
      • test_put_along_axis_op.py
      • test_py_func_op.py
      • test_py_reader_combination.py
      • test_py_reader_return_list.py
      • test_py_reader_sample_generator.py
      • test_pyramid_hash_op.py
      • test_python_operator_overriding.py
      • test_qr_op.py
      • test_quantile_and_nanquantile.py
      • test_randn_op.py
      • test_random_seed.py
      • test_reader_reset.py
      • test_real_imag_op.py
      • test_reduce_op.py
      • test_regularizer.py
      • test_regularizer_api.py
      • test_repeat_interleave_op.py
      • test_reshape_op.py
      • test_reverse_op.py
      • test_rnn_cell_api.py
      • test_rnn_decode_api.py
      • test_rnn_op.py
      • test_roi_align_op.py
      • test_roi_pool_op.py
      • test_roll_op.py
      • test_row_conv_op.py
      • test_rrelu_op.py
      • test_run_program_op.py
      • test_save_inference_model_conditional_op.py
      • test_save_model_without_var.py
      • test_scale_op.py
      • test_scatter_nd_op.py
      • test_scatter_op.py
      • test_seed_op.py
      • test_segment_ops.py
      • test_select_input_output_op.py
      • test_selu_op.py
      • test_set_bool_attr.py
      • test_set_value_op.py
      • test_sgd_op.py
      • test_shuffle_batch_op.py
      • test_shuffle_channel_op.py
      • test_sigmoid_cross_entropy_with_logits_op.py
      • test_sign_op.py
      • test_signal.py
      • test_simple_rnn_op.py
      • test_slice_op.py
      • test_slice_scatter.py
      • test_slice_var.py
      • test_softmax_op.py
      • test_solve_op.py
      • test_sparse_conv_op.py
      • test_sparse_elementwise_op.py
      • test_sparse_isnan_op.py
      • test_sparse_norm_op.py
      • test_sparse_slice_op.py
      • test_sparse_softmax_op.py
      • test_sparse_sum_op.py
      • test_spectral_norm_op.py
      • test_split_op.py
      • test_split_program.py
      • test_squared_l2_norm_op.py
      • test_squeeze2_op.py
      • test_static_pylayer.py
      • test_static_pylayer_block.py
      • test_static_save_load.py
      • test_static_save_load_large.py
      • test_static_shape_inferrence_for_shape_tensor.py
      • test_stft_op.py
      • test_stride.py
      • test_svd_op.py
      • test_swiglu.py
      • test_switch.py
      • test_switch_autotune.py
      • test_tdm_child_op.py
      • test_tdm_sampler_op.py
      • test_temporal_shift_op.py
      • test_tensor.py
      • test_tensor_array_to_tensor.py
      • test_tensor_scalar_type_promotion_static.py
      • test_tensor_type_promotion.py
      • test_top_k_op.py
      • test_top_k_v2_op.py
      • test_trace_op.py
      • test_trainable.py
      • test_transformer_api.py
      • test_transpose_op.py
      • test_triangular_solve_op.py
      • test_tril_triu_op.py
      • test_trilinear_interp_op.py
      • test_trilinear_interp_v2_op.py
      • test_trunc_op.py
      • test_truncated_gaussian_random_op.py
      • test_unbind_op.py
      • test_unfold_op.py
      • test_uniform_random_bf16_op.py
      • test_uniform_random_op.py
      • test_unique_consecutive_op.py
      • test_unpool3d_op.py
      • test_unpool_op.py
      • test_unsqueeze2_op.py
      • test_unstack_op.py
      • test_var_base.py
      • test_var_info.py
      • test_variable.py
      • test_warprnnt_op.py
      • test_weight_normalization.py
      • test_where_op.py
      • test_yolov3_loss_op.py
      • test_zero_dim_complex_api.py
      • test_zero_dim_distribution_loss_api.py
      • test_zero_dim_no_backward_api.py
      • test_zero_dim_sundry_dygraph_api.py
      • test_zero_dim_sundry_static_api_part1.py
      • test_zero_dim_sundry_static_api_part3.py
      • utils.py
    • prim
      • CMakeLists.txt
      • composite_ops
        • CMakeLists.txt
        • test_composite_batch_norm.py
        • test_composite_batch_norm_grad.py
        • test_composite_dropout.py
        • test_composite_gelu.py
        • test_composite_gelu_grad.py
        • test_composite_layer_norm.py
        • test_composite_layer_norm_grad.py
        • test_composite_mean.py
        • test_composite_mean_grad.py
        • test_composite_relu_custom_vjp.py
        • test_composite_softmax.py
        • test_composite_softmax_custom_vjp.py
        • test_composite_softmax_grad.py
      • pir_prim
        • CMakeLists.txt
        • test_custom_vjp_trait.py
        • test_decomp_op.py
        • test_decompose_op.py
        • test_vjp_prim.py
      • prim
        • CMakeLists.txt
        • flags
          • CMakeLists.txt
          • test_prim_flags.py
          • test_prim_flags_case.py
        • vjp
          • CMakeLists.txt
          • eager
            • CMakeLists.txt
            • test_comp_eager_cast_grad.py
            • test_comp_eager_pow_grad.py
          • static
            • CMakeLists.txt
            • test_comp_add_grad.py
            • test_comp_add_tanh_grad.py
            • test_comp_cast_grad.py
            • test_comp_div_grad.py
            • test_comp_exp_grad.py
            • test_comp_gather_grad.py
            • test_comp_matmul_double_grad.py
            • test_comp_reshape_grad.py
            • test_comp_sigmoid_grad.py
            • test_comp_sqrt_grad.py
            • test_comp_sub_grad.py
            • test_comp_tanh_grad.py
            • test_comp_transpose_grad.py
      • process
        • CMakeLists.txt
        • test_check_inputs.py
        • test_copy_op.py
      • test_comp_custom_vjp.py
      • test_comp_dispensable.py
      • test_comp_get_grad_op_desc_prim_enabled.py
      • test_comp_skip_op_set.py
    • ps
      • config_gpubox.yaml
      • cpu_async_ps_config.yaml
      • cpu_geo_ps_config.yaml
      • cpu_sync_ps_config.yaml
      • fl_async_ps_config.yaml
      • gpu_ps_config.yaml
      • heter_ps_config.yaml
      • ps_dnn_model.py
      • ps_dnn_trainer.py
    • quantization
      • CMakeLists.txt
      • test_graph.py
      • test_imperative_out_scale.py
      • test_imperative_qat.py
      • test_imperative_qat_channelwise.py
      • test_imperative_qat_fuse.py
      • test_imperative_skip_op.py
      • test_moving_average_abs_max_scale_op.py
      • test_post_training_quantization_while.py
      • test_ptq.py
      • test_quant2_int8_mkldnn_pass.py
      • test_quant_amp.py
      • test_quant_aware.py
      • test_quant_aware_user_defined.py
      • test_quant_post_quant_aware.py
      • test_quantization_mkldnn_pass.py
      • test_quantization_pass.py
      • test_quantization_scale_pass.py
      • test_trace_quanter.py
      • test_user_defined_quantization.py
      • test_weight_quantization_mobilenetv1.py
    • rnn
      • CMakeLists.txt
      • convert.py
      • test_rnn_api.py
      • test_rnn_cells_static.py
      • test_rnn_cudnn_params_packing.py
      • test_rnn_nets.py
      • test_rnn_nets_static.py
    • sequence
      • CMakeLists.txt
      • test_sequence_conv.py
      • test_sequence_expand.py
      • test_sequence_mask.py
      • test_sequence_pool.py
      • test_sequence_softmax_op.py
    • standalone_executor
      • CMakeLists.txt
      • test_standalone_dist_attr_run_time_set_get.py
      • test_standalone_executor.py
      • test_standalone_executor_1f1b_plan.py
      • test_standalone_executor_fthenb_plan.py
      • test_standalone_executor_multi_micro_batch.py
      • test_standalone_executor_plan.py
      • test_standalone_op_priority.py
      • test_standalone_sequentail_run.py
    • tokenizer
      • CMakeLists.txt
      • test_faster_tokenizer_op.py
  • ir/inference
    • CMakeLists.txt
  • legacy_test
    • CMakeLists.txt
    • run_server_for_communicator_geo.py
    • test_attention_lstm_op.py
    • test_conv2d_transpose_op.py
    • test_cross_entropy_loss.py
    • test_fused_embedding_fc_lstm_op.py
    • test_fused_fc_elementwise_layernorm_op.py
    • test_fusion_gru_op.py
    • test_fusion_lstm_op.py
    • test_fusion_repeated_fc_relu_op.py
    • test_fusion_seqconv_eltadd_relu_op.py
    • test_fusion_seqpool_concat_op.py
    • test_fusion_seqpool_cvm_concat_op.py
    • test_imperative_hook_for_layer.py
    • test_pad_op.py
    • test_pool2d_api.py
    • test_pool3d_api.py
    • test_softmax2d.py
    • test_softmax_with_cross_entropy_op.py
    • test_static_save_load_bf16.py
    • test_warpctc_op.py
  • mkldnn
    • test_batch_norm_mkldnn_op.py
    • test_elementwise_mul_onednn_op.py
    • test_gaussian_random_mkldnn_op.py
    • test_log_softmax_mkldnn_op.py
    • test_lrn_mkldnn_op.py
    • test_pool2d_bf16_mkldnn_op.py
    • test_pool2d_int8_mkldnn_op.py
    • test_pool2d_mkldnn_op.py
    • test_softmax_bf16_mkldnn_op.py
    • test_softmax_mkldnn_op.py
  • prim
    • composite_ops
      • CMakeLists.txt
    • pir_prim
      • CMakeLists.txt
    • prim
      • flags
        • CMakeLists.txt
      • vjp/static
        • CMakeLists.txt
  • quantization
    • CMakeLists.txt
  • rnn
    • CMakeLists.txt
  • sequence
    • CMakeLists.txt
  • standalone_executor
    • CMakeLists.txt
    • test_standalone_custom_stream.py
  • tokenizer
    • CMakeLists.txt
  • xpu
    • test_pad_op_xpu.py
    • test_pool2d_op_xpu.py
    • test_softmax_with_cross_entropy_op_xpu.py
    • test_warpctc_op_xpu.py