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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_input_ops.cc 150 std::shared_ptr<memory::desc> diff_src_md; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
169 diff_src_md(nullptr),
177 context_.diff_src_md.reset(
188 convolution_direct, *context_.diff_src_md, *context_.filter_md,
194 prop_kind::forward, convolution_direct, *context_.diff_src_md,
mkl_pooling_ops_common.h 268 std::shared_ptr<mkldnn::memory::desc> diff_src_md; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
290 diff_src_md(nullptr),

Completed in 71 milliseconds