Searched defs:workspace_enabled (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_maxpooling_op.cc72 context->GetAttr("workspace_enabled", &workspace_enabled_);
229 context->GetAttr("workspace_enabled", &workspace_enabled_);
341 void MklCreatePrimitives(OpKernelContext* context, bool workspace_enabled) { argument
345 if (workspace_enabled == false) {
364 if (workspace_enabled == false) {
388 void MklPrepareInputs(OpKernelContext* context, bool workspace_enabled) { argument
397 if (workspace_enabled == false) {
469 void MklCleanup(bool workspace_enabled) { argument
472 if (workspace_enabled == false) {
483 if (workspace_enabled
[all...]

Completed in 108 milliseconds