Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_maxpooling_op.cc209 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
213 workspace_tf_shape.AddDim(workspace_bytes); in AllocateWorkspaceTensor()
Dmkl_fused_batch_norm_op.cc725 size_t workspace_bytes = 0; in Compute() local
726 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
788 size_t workspace_bytes = workspace_md.get_size(); in Compute() local
789 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
801 size_t workspace_bytes = 0; in Compute() local
802 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
Dmkl_lrn_op.cc268 size_t workspace_bytes = workspace_pd.get_size(); in AllocateWorkspaceTensor() local
274 workspace_tf_shape.AddDim(workspace_bytes); in AllocateWorkspaceTensor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3229 SE_ASSIGN_OR_RETURN(DeviceMemory<uint8> workspace_bytes, in DoBlasInternalImpl()
3231 *workspace = (void *)GpuMemoryMutable(&workspace_bytes); in DoBlasInternalImpl()