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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 235 return context_.bwd_pd;
275 std::shared_ptr<mkldnn::pooling_backward::primitive_desc> bwd_pd; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
295 bwd_pd(nullptr),
mkl_relu_op.cc 279 return context_.bwd_pd;
313 std::shared_ptr<mkldnn::eltwise_backward::primitive_desc> bwd_pd; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
334 bwd_pd(nullptr),
359 context_.bwd_pd.reset(new mkldnn::eltwise_backward::primitive_desc(
366 context_.bwd_pd.get()->diff_src_primitive_desc(), DummyData));
370 *context_.bwd_pd, *context_.src_mem, *context_.diff_dst_mem,
    [all...]

Completed in 794 milliseconds