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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 126 std::shared_ptr<mkldnn::memory::desc> dst_md; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
143 dst_md(nullptr),
mkl_relu_op.cc 112 std::shared_ptr<memory::desc> dst_md; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
130 dst_md(nullptr),
    [all...]
mkl_conv_ops.cc 189 std::shared_ptr<mkldnn::memory::desc> dst_md; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
222 context_.dst_md.reset(new memory::desc(
233 *context_.filter_md, *context_.bias_md, *context_.dst_md,
239 *context_.filter_md, *context_.dst_md, convFwdDims.strides,
1257 auto dst_md = dst_pd.desc(); local
1771 auto dst_md = summand_mkl_shape.GetMklLayout(); variable
    [all...]

Completed in 61 milliseconds