Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_matmul_ops_common.h332 auto scratchpad_md = context_.fwd_pd->scratchpad_desc(); in Setup() local
767 auto scratchpad_md = context_.prim_desc->scratchpad_desc(); in Setup() local
H A Dmkl_conv_ops.cc382 auto scratchpad_md = context_.fwd_pd->scratchpad_desc(); in Setup() local