| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_softmax_op.cc | 41 memory::dims src_dims; member in tensorflow::MklSoftmaxParams
47 MklSoftmaxParams(memory::dims src_dims, MklTensorFormat src_fmt, int axis) in MklSoftmaxParams()
229 memory::dims src_dims; in Compute() local
|
| H A D | mkl_relu_op.cc | 47 memory::dims src_dims; member in tensorflow::MklEltwiseFwdParams
53 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
223 memory::dims src_dims; member in tensorflow::MklEltwiseBwdParams
232 MklEltwiseBwdParams(const memory::dims& src_dims, in MklEltwiseBwdParams()
464 memory::dims src_dims; in Compute() local
622 memory::dims src_dims = {}; in Compute() local
|
| H A D | mkl_concat_op.cc | 243 std::vector<memory::dims> src_dims; member
638 memory::dims src_dims(src_md.data.dims, in Compute() local
650 memory::dims src_dims = TFShapeToMklDnnDims(s_shape); in Compute() local
897 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat() local
|
| H A D | mkl_eltwise_activation_base_op.h | 61 MklEltwiseFwdActivationParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdActivationParams()
267 memory::dims src_dims; in Compute() local
|
| H A D | mkl_fused_batch_norm_op.cc | 48 memory::dims src_dims; member
392 memory::dims src_dims; member
828 memory::dims src_dims = in Compute() local
1252 memory::dims src_dims = in Compute() local
|
| H A D | mkl_quantize_op.cc | 62 memory::dims src_dims; member
345 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
|
| H A D | mkl_dequantize_op.cc | 67 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute() local
|
| H A D | mkl_matmul_ops_common.h | 60 memory::dims src_dims; member
85 : src_dims(src_dims), in src_dims() function
|
| H A D | mkl_layer_norm_op.cc | 66 memory::dims src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute() local
|
| H A D | mkl_matmul_op_fused.cc | 116 memory::dims src_dims = memory::dims({batch, k}); in Compute() local
|
| H A D | mkl_aggregate_ops.cc | 194 memory::dims src_dims; in Compute() local
|
| H A D | mkl_avgpooling_op.cc | 88 memory::dims src_dims = in Compute() local
|
| H A D | mkl_maxpooling_op.cc | 114 memory::dims src_dims = in Compute() local
|
| H A D | mkl_qmatmul_op.cc | 178 memory::dims src_dims, weight_dims; in Compute() local
|
| H A D | mkl_conv_grad_filter_ops.cc | 43 memory::dims src_dims; member
|
| H A D | mkl_conv_ops.cc | 43 memory::dims src_dims; member
648 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
|
| H A D | mkl_pooling_ops_common.h | 43 memory::dims src_dims; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | MKLDNNConversions.cpp | 153 ideep::dims src_dims = ideep::dims(); in mkldnn_reorder_conv2d_weight() local
210 ideep::dims src_dims = ideep::dims(); in mkldnn_reorder_conv3d_weight() local
294 const ideep::dims& src_dims) { in get_conv_transpose_expected_weights_desc()
345 ideep::dims src_dims = ideep::dims(); in mkldnn_reorder_conv_transpose_weight() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 1186 auto src_dims = act_contig.sizes().vec(); in apply_impl() local
1579 auto src_dims = act_contig.sizes().vec(); in _quantized_convolution_onednn() local
|
| H A D | qlinear_prepack.cpp | 256 auto src_dims = ideep::dims(); // Unknown when prepacking in prepack() local
|
| H A D | qlinear.cpp | 1004 std::vector<int64_t> src_dims = {M, K}; in linear_int8_with_onednn_weight() local
|