| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_maxpooling_op.cc | 121 Tensor* workspace_tensor; variable
129 AllocateOutputSetMklShape(context, 1, &workspace_tensor, workspace_shape,
133 static_cast<const void*>(workspace_tensor->flat<T>().data()));
659 Tensor* workspace_tensor = nullptr; local
668 &workspace_tensor, workspace_tf_shape,
670 CHECK_NOTNULL(workspace_tensor);
671 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
693 const Tensor& workspace_tensor = variable
764 const_cast<uint8*>(workspace_tensor.flat<uint8>().data()));
[all...] |
| mkl_lrn_op.cc | 244 Tensor* workspace_tensor = nullptr; local
249 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor,
251 CHECK_NOTNULL(workspace_tensor);
272 Tensor* workspace_tensor = nullptr; local
282 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor,
284 CHECK_NOTNULL(workspace_tensor);
285 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
423 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); variable
425 ConfigureWorkspace(workspace_tensor,
511 void ConfigureWorkspace(const Tensor& workspace_tensor,
610 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); local
[all...] |
| mkl_avgpooling_op.cc | 96 Tensor workspace_tensor; variable
98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace,
|
