Home
last modified time | relevance | path

Searched defs:workspace_bytes (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op.cc786 size_t workspace_bytes = 0; in Compute() local
849 size_t workspace_bytes = workspace_md.get_size(); in Compute() local
862 size_t workspace_bytes = 0; in Compute() local
H A Dmkl_maxpooling_op.cc215 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
H A Dmkl_lrn_op.cc269 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local