| mkl_maxpooling_op.cc | 120 Tensor* workspace_tensor; variable
128 AllocateOutputSetMklShape(context, 1, &workspace_tensor, workspace_shape,
132 static_cast<const void*>(workspace_tensor->flat<T>().data()));
591 Tensor* workspace_tensor = nullptr; local
600 &workspace_tensor, workspace_tf_shape,
602 CHECK_NOTNULL(workspace_tensor);
603 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
628 const Tensor& workspace_tensor = variable
638 grad_tensor, workspace_tensor, orig_input_mkl_shape,
695 ConfigureWorkspace(workspace_tensor,
[all...] |
