| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_avgpooling_op.cc | 129 MklDnnThreadPool eigen_tp(context); in Compute() local
263 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_maxpooling_op.cc | 151 MklDnnThreadPool eigen_tp(context); in Compute() local
313 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_transpose_op.cc | 87 MklDnnThreadPool eigen_tp(context); in MKLTransposeND() local
|
| H A D | mkl_dequantize_op.cc | 75 MklDnnThreadPool eigen_tp(ctx); in Compute() local
|
| H A D | mkl_requantize_per_channel_op.cc | 142 MklDnnThreadPool eigen_tp(ctx); in Compute() local
|
| H A D | mkl_concat_op.cc | 747 MklDnnThreadPool eigen_tp(context); in Compute() local
785 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_batch_matmul_op.cc | 162 MklDnnThreadPool eigen_tp(ctx); in Compute() local
|
| H A D | mkl_matmul_op.cc | 159 MklDnnThreadPool eigen_tp(ctx); in MklBlasGemm() local
|
| H A D | mkl_matmul_op_fused.cc | 267 MklDnnThreadPool eigen_tp(ctx, st ? 1 : -1); in Compute() local
|
| H A D | mkl_aggregate_ops.cc | 177 MklDnnThreadPool eigen_tp(ctx); in Compute() local
|
| H A D | mkl_einsum_op.cc | 122 MklDnnThreadPool eigen_tp(ctx); in MKLContractOperands() local
|
| H A D | mkl_softmax_op.cc | 324 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_fused_batch_norm_op.cc | 930 MklDnnThreadPool eigen_tp(context); in Compute() local
1367 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_relu_op.cc | 483 MklDnnThreadPool eigen_tp(context); in Compute() local
690 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_slice_op.cc | 463 MklDnnThreadPool eigen_tp(context); in ComputeMklSlice() local
|
| H A D | mkl_qmatmul_op.cc | 294 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_quantize_op.cc | 486 MklDnnThreadPool eigen_tp(ctx); in Compute() local
|
| H A D | mkl_conv_grad_input_ops.cc | 486 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_lrn_op.cc | 170 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_conv_grad_filter_ops.cc | 615 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| H A D | mkl_conv_ops.cc | 890 MklDnnThreadPool eigen_tp(context); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 228 inline dnnl::stream* CreateStream(MklDnnThreadPool* eigen_tp, in CreateStream()
613 MklDnnThreadPool eigen_tp; variable
1543 MklDnnThreadPool eigen_tp; variable
1610 MklDnnThreadPool eigen_tp; variable
1721 MklDnnThreadPool eigen_tp; variable
|