HomeSort by relevance Sort by last modified time
    Searched defs:workspace_enabled_ (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 111 : OpKernel(context), workspace_enabled_(false) {
134 context->GetAttr("workspace_enabled", &this->workspace_enabled_);
183 bool workspace_enabled_; member in class:tensorflow::MklPoolingOpBase
mkl_lrn_op.cc 90 workspace_enabled_ = false;
91 context->GetAttr("workspace_enabled", &workspace_enabled_);
124 workspace_enabled_ = true;
151 workspace_enabled_);
156 workspace_enabled_);
215 Tensor** workspace, bool workspace_enabled_) {
233 if (workspace_enabled_) {
256 bool workspace_enabled_) {
270 if (workspace_enabled_) {
337 bool workspace_enabled_; member in class:tensorflow::MklLRNOp
728 bool workspace_enabled_; member in class:tensorflow::MklLRNGradOp
    [all...]
mkl_maxpooling_op.cc 68 workspace_enabled_ = false;
72 context->GetAttr("workspace_enabled", &workspace_enabled_);
194 bool workspace_enabled_; member in class:tensorflow::MklMaxPoolingOp
225 workspace_enabled_ = false;
229 context->GetAttr("workspace_enabled", &workspace_enabled_);
268 mkl_context.MklCreatePrimitives(context, workspace_enabled_);
271 mkl_context.MklPrepareInputs(context, workspace_enabled_);
298 mkl_context.MklCleanup(workspace_enabled_);
494 bool workspace_enabled_; member in class:tensorflow::MklMaxPoolingGradOp
506 // So we set workspace_enabled_ to true
    [all...]

Completed in 76 milliseconds