tensorflow/tensorflow/external/mkl_dnn_v1/src/cpu/ Source Tree

../
bfloat16.cpp
binary_injector_utils.cpp
binary_injector_utils.hpp
cpu_batch_normalization_list.cpp
cpu_batch_normalization_pd.hpp cpu_batch_normalization_bwd_pd_t cpu_batch_normalization_fwd_pd_t
cpu_batch_normalization_utils.cpp
cpu_batch_normalization_utils.hpp
cpu_binary_list.cpp
cpu_binary_pd.hpp cpu_binary_pd_t
cpu_concat.cpp
cpu_concat_pd.hpp cpu_concat_pd_t
cpu_convolution_list.cpp
cpu_convolution_pd.hpp cpu_convolution_bwd_data_pd_t cpu_convolution_bwd_weights_pd_t cpu_convolution_fwd_pd_t
cpu_deconvolution_list.cpp
cpu_deconvolution_pd.hpp cpu_deconvolution_bwd_data_pd_t cpu_deconvolution_bwd_weights_pd_t cpu_deconvolution_fwd_pd_t
cpu_eltwise_list.cpp
cpu_eltwise_pd.hpp cpu_eltwise_bwd_pd_t cpu_eltwise_fwd_pd_t
cpu_engine.cpp
cpu_engine.hpp cpu_engine_factory_t cpu_engine_impl_list_t cpu_engine_t
cpu_inner_product_list.cpp
cpu_inner_product_pd.hpp cpu_inner_product_bwd_data_pd_t cpu_inner_product_bwd_weights_pd_t cpu_inner_product_fwd_pd_t
cpu_layer_normalization_list.cpp
cpu_layer_normalization_pd.hpp cpu_layer_normalization_bwd_pd_t cpu_layer_normalization_fwd_pd_t
cpu_lrn_list.cpp
cpu_lrn_pd.hpp cpu_lrn_bwd_pd_t cpu_lrn_fwd_pd_t
cpu_memory_storage.hpp cpu_memory_storage_t
cpu_pooling_list.cpp
cpu_pooling_pd.hpp cpu_pooling_bwd_pd_t cpu_pooling_fwd_pd_t
cpu_prelu_list.cpp
cpu_prelu_pd.hpp cpu_prelu_bwd_pd_t cpu_prelu_fwd_pd_t
cpu_primitive.hpp
cpu_reduction_list.cpp
cpu_reduction_pd.hpp cpu_reduction_pd_t
cpu_resampling_list.cpp
cpu_resampling_pd.hpp cpu_resampling_bwd_pd_t cpu_resampling_fwd_pd_t
cpu_rnn_list.cpp
cpu_shuffle_list.cpp
cpu_shuffle_pd.hpp cpu_shuffle_pd_t
cpu_softmax_list.cpp
cpu_softmax_pd.hpp cpu_softmax_bwd_pd_t cpu_softmax_fwd_pd_t
cpu_stream.hpp cpu_stream_t
cpu_sum.cpp
cpu_sum_pd.hpp cpu_sum_pd_t
dw_convolution_utils.hpp
[+] gemm/
gemm_convolution.cpp
gemm_convolution.hpp gemm_convolution_bwd_data_t gemm_convolution_bwd_weights_t gemm_convolution_fwd_t
gemm_convolution_utils.cpp
gemm_convolution_utils.hpp conv_gemm_conf_t conv_gemm_loop_order_t single_gemm_conv_chunk_desc_t
gemm_inner_product.cpp gemm_inner_product_bwd_data_t gemm_inner_product_bwd_weights_t gemm_inner_product_fwd_t
gemm_inner_product.hpp gemm_inner_product_bwd_data_t gemm_inner_product_bwd_weights_t gemm_inner_product_fwd_t
gemm_inner_product_utils.cpp ref_pp_kernel_t
gemm_inner_product_utils.hpp pp_kernel_t
gemm_x8s8s32x_conv_zp_src_pad_comp.cpp
gemm_x8s8s32x_conv_zp_src_pad_comp.hpp
gemm_x8s8s32x_convolution.cpp
gemm_x8s8s32x_convolution.hpp gemm_x8s8s32x_convolution_bwd_data_t gemm_x8s8s32x_convolution_fwd_t
gemm_x8s8s32x_convolution_utils.cpp ref_pp_ker_t
gemm_x8s8s32x_convolution_utils.hpp pp_ker_t
gemm_x8s8s32x_inner_product.cpp
gemm_x8s8s32x_inner_product.hpp gemm_x8s8s32x_inner_product_fwd_t
[+] jit_utils/
[+] matmul/
nchw_pooling.cpp nchw_pooling_bwd_t nchw_pooling_fwd_t
nchw_pooling.hpp nchw_pooling_bwd_t nchw_pooling_fwd_t
ncsp_batch_normalization.cpp ncsp_batch_normalization_bwd_t ncsp_batch_normalization_fwd_t
ncsp_batch_normalization.hpp ncsp_batch_normalization_bwd_t ncsp_batch_normalization_fwd_t
nhwc_pooling.cpp nhwc_pooling_bwd_t nhwc_pooling_fwd_t
nhwc_pooling.hpp nhwc_pooling_bwd_t nhwc_pooling_fwd_t
nspc_batch_normalization.cpp nspc_batch_normalization_bwd_t nspc_batch_normalization_fwd_t
nspc_batch_normalization.hpp nspc_batch_normalization_bwd_t nspc_batch_normalization_fwd_t
platform.cpp
platform.hpp
primitive_attr_postops.cpp
primitive_attr_postops.hpp ref_binary_scalar_t ref_eltwise_scalar_fwd_t ref_post_ops_t
ref_batch_normalization.cpp ref_batch_normalization_bwd_t ref_batch_normalization_fwd_t
ref_batch_normalization.hpp ref_batch_normalization_bwd_t ref_batch_normalization_fwd_t
ref_binary.cpp
ref_binary.hpp ref_binary_t
ref_concat.hpp ref_concat_t
ref_convolution.cpp
ref_convolution.hpp ref_convolution_bwd_data_t ref_convolution_bwd_weights_t ref_convolution_fwd_t
ref_convolution_int8.cpp
ref_convolution_int8.hpp ref_convolution_int8_bwd_data_t ref_convolution_int8_fwd_t
ref_convolution_utils.hpp
ref_deconvolution.cpp
ref_deconvolution.hpp ref_deconvolution_bwd_data_t ref_deconvolution_bwd_weights_t ref_deconvolution_fwd_t
ref_eltwise.cpp ref_eltwise_bwd_t ref_eltwise_fwd_t
ref_eltwise.hpp ref_eltwise_bwd_t ref_eltwise_fwd_t
ref_fused_convolution.hpp ref_fused_convolution_fwd_t
ref_inner_product.cpp
ref_inner_product.hpp ref_inner_product_bwd_data_t ref_inner_product_bwd_weights_t ref_inner_product_fwd_t
ref_inner_product_int8.cpp
ref_inner_product_int8.hpp ref_inner_product_int8_fwd_t
ref_inner_product_utils.hpp
ref_io_helper.hpp
ref_layer_normalization.cpp ref_layer_normalization_bwd_t ref_layer_normalization_fwd_t
ref_layer_normalization.hpp ref_layer_normalization_bwd_t ref_layer_normalization_fwd_t
ref_lrn.cpp
ref_lrn.hpp ref_lrn_bwd_t ref_lrn_fwd_t
ref_pooling.cpp ref_pooling_bwd_t ref_pooling_fwd_t
ref_pooling.hpp ref_pooling_bwd_t ref_pooling_fwd_t
ref_prelu.cpp
ref_prelu.hpp ref_prelu_bwd_t ref_prelu_fwd_t
ref_reduction.cpp ref_reduction_t
ref_reduction.hpp ref_reduction_t
ref_resampling.cpp
ref_resampling.hpp ref_resampling_bwd_t ref_resampling_fwd_t
ref_shuffle.cpp
ref_shuffle.hpp ref_shuffle_t
ref_softmax.cpp
ref_softmax.hpp ref_softmax_bwd_t ref_softmax_fwd_t
ref_sum.hpp ref_sum_t
[+] reorder/
resampling_utils.hpp bwd_linear_coeffs_t linear_coeffs_t
[+] rnn/
simple_concat.cpp simple_concat_t
simple_concat.hpp simple_concat_t
simple_layer_normalization.cpp simple_layer_normalization_bwd_t simple_layer_normalization_fwd_t
simple_layer_normalization.hpp simple_layer_normalization_bwd_t simple_layer_normalization_fwd_t
simple_layer_normalization_kernels.cpp diff_data_kernel_t diff_ss_kernel_t stat_and_data_kernel_t
simple_layer_normalization_kernels.hpp diff_data_kernel_t diff_ss_kernel_t stat_and_data_kernel_t
simple_q10n.hpp qz qz_a1 qz_a1b0 qz_b0
simple_resampling.cpp
simple_resampling.hpp simple_resampling_base_t simple_resampling_bwd_t simple_resampling_fwd_t
simple_sum.cpp simple_sum_t
simple_sum.hpp simple_sum_t sum_bf16_params_t
[+] x64/
zero_point_utils.cpp
zero_point_utils.hpp zero_point_call_params_t zero_point_config_t zero_point_pad_comp_config_t

Browse the source code of tensorflow/tensorflow/external/mkl_dnn_v1/src/cpu/ online