Home
last modified time | relevance | path

Searched refs:GetSrcMemoryFormat (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_relu_op.cc94 memory::format GetSrcMemoryFormat() { return context_.src_fmt; } in GetSrcMemoryFormat() function in tensorflow::MklEltwiseFwdPrimitive
282 memory::format GetSrcMemoryFormat() { return context_.src_fmt; } in GetSrcMemoryFormat() function in tensorflow::MklEltwiseBwdPrimitive
492 if (src_md.data.format != eltwise_fwd->GetSrcMemoryFormat()) { in Compute()
495 {{src_dims}, MklDnnType<T>(), eltwise_fwd->GetSrcMemoryFormat()}, in Compute()
649 if (src_md.data.format != eltwise_bwd->GetSrcMemoryFormat()) { in Compute()
Dmkl_avgpooling_op.cc520 if (input_md.data.format != pooling_fwd->GetSrcMemoryFormat()) { in Compute()
523 {{src_dims}, MklDnnType<T>(), pooling_fwd->GetSrcMemoryFormat()}, in Compute()
Dmkl_conv_grad_filter_ops.cc139 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; } in GetSrcMemoryFormat() function in tensorflow::MklConvBwdFilterPrimitive
539 if (fwd_src_md.data.format != conv_bwd_filter->GetSrcMemoryFormat()) { in Compute()
Dmkl_maxpooling_op.cc597 if (input_md.data.format != pooling_fwd->GetSrcMemoryFormat()) { in Compute()
600 {{src_dims}, MklDnnType<T>(), pooling_fwd->GetSrcMemoryFormat()}, in Compute()
Dmkl_pooling_ops_common.h91 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; } in GetSrcMemoryFormat() function
Dmkl_conv_ops.cc161 memory::format GetSrcMemoryFormat() const { return context_.src_fmt; } in GetSrcMemoryFormat() function in tensorflow::MklConvFwdPrimitive
1100 if (src_md.data.format != conv_fwd->GetSrcMemoryFormat()) { in Compute()