Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc268 memory::desc workspace_pd = lrn_fwd_prim_desc.workspace_desc(); in AllocateWorkspaceTensor() local
493 memory::desc workspace_pd, in ConfigureWorkspace()
H A Dmkl_maxpooling_op.cc214 memory::desc workspace_pd = pool_fwd_prim_desc.workspace_desc(); in AllocateWorkspaceTensor() local