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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_filter_ops.cc 139 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; }
539 if (fwd_src_md.data.format != conv_bwd_filter->GetSrcMemoryFormat()) {
mkl_pooling_ops_common.h 91 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; }
mkl_relu_op.cc 94 memory::format GetSrcMemoryFormat() { return context_.src_fmt; }
282 memory::format GetSrcMemoryFormat() { return context_.src_fmt; }
492 if (src_md.data.format != eltwise_fwd->GetSrcMemoryFormat()) {
495 {{src_dims}, MklDnnType<T>(), eltwise_fwd->GetSrcMemoryFormat()},
649 if (src_md.data.format != eltwise_bwd->GetSrcMemoryFormat()) {
    [all...]
mkl_conv_ops.cc 161 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; }
    [all...]

Completed in 2131 milliseconds