Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_maxpooling_op.cc210 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
Dmkl_fused_batch_norm_op.cc732 size_t workspace_bytes = 0; in Compute() local
795 size_t workspace_bytes = workspace_md.get_size(); in Compute() local
808 size_t workspace_bytes = 0; in Compute() local
Dmkl_lrn_op.cc269 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc347 int64_t workspace_bytes = 1LL << 32; // 4GB by default. in operator ()() local