Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_maxpooling_op.cc209 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
Dmkl_fused_batch_norm_op.cc725 size_t workspace_bytes = 0; in Compute() local
788 size_t workspace_bytes = workspace_md.get_size(); in Compute() local
801 size_t workspace_bytes = 0; in Compute() local
Dmkl_lrn_op.cc268 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local