Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc239 Tensor* workspace_tensor = nullptr; in MklDefaultToEigen() local
267 Tensor* workspace_tensor = nullptr; in AllocateWorkspaceTensor() local
422 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); in Compute() local
492 void ConfigureWorkspace(const Tensor& workspace_tensor, in ConfigureWorkspace()
595 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); in SanityCheckInputs() local
H A Dmkl_maxpooling_op.cc213 Tensor* workspace_tensor = nullptr; in AllocateWorkspaceTensor() local
247 const Tensor& workspace_tensor = in Compute() local