Home
last modified time | relevance | path

Searched defs:workspace_mkl_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc240 MklDnnShape workspace_mkl_shape; in MklDefaultToEigen() local
270 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor() local
H A Dmkl_maxpooling_op.cc216 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor() local