Home
last modified time | relevance | path

Searched refs:do_not_cache (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc248 const MklConvBwdInputParams& convBwdInputDims, bool do_not_cache) { in Get() argument
251 if (do_not_cache) { // Always allocate primitive. in Get()
429 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled() && in Compute() local
435 do_not_cache); in Compute()
491 if (do_not_cache) { in Compute()
Dmkl_conv_grad_filter_ops.cc298 const MklConvBwdFilterParams& convBwdFilterDims, bool do_not_cache) { in Get() argument
301 if (do_not_cache) { /* Create new primitive always */ in Get()
473 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled(); in Compute() local
477 do_not_cache); in Compute()
605 if (do_not_cache) delete conv_bwd_filter; in Compute()
Dmkl_matmul_ops_common.h295 const MklDnnMatMulFwdParams& mkldnn_matmul_fwd_dims, bool do_not_cache) { in Get() argument
299 if (do_not_cache) { in Get()
639 bool do_not_cache) { in Get() argument
642 if (do_not_cache) { in Get()
Dmkl_conv_ops.cc324 const MklConvFwdParams& convFwdDims, bool do_not_cache) { in Get() argument
327 if (do_not_cache) { in Get()
640 bool do_not_cache = in Compute() local
662 convFwdDims, do_not_cache); in Compute()
739 if (do_not_cache) delete conv_fwd; in Compute()
Dmkl_concat_op.cc386 const std::vector<memory::desc>& srcs_md, bool do_not_cache) { in Get() argument
389 if (do_not_cache) { in Get()