OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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