Home
last modified time | relevance | path

Searched defs:src_mkl_shape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_aggregate_ops.cc43 MklDnnShape src_mkl_shape; in GetTensorShape() local
74 MklDnnShape src_mkl_shape; in FindMKLInputIndex() local
184 MklDnnShape src_mkl_shape; in Compute() local
H A Dmkl_matmul_op_fused.cc67 MklDnnShape src_mkl_shape; in Compute() local
H A Dmkl_softmax_op.cc220 MklDnnShape src_mkl_shape; in Compute() local
H A Dmkl_qmatmul_op.cc167 MklDnnShape src_mkl_shape, weight_mkl_shape; in Compute() local
H A Dmkl_quantize_op.cc341 MklDnnShape src_mkl_shape; in Compute() local
H A Dmkl_conv_grad_input_ops.cc318 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
H A Dmkl_conv_grad_filter_ops.cc370 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
H A Dmkl_conv_ops.cc636 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute() local