Home
last modified time | relevance | path

Searched refs:tensor_format_mkldnn_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_pooling_ops_common.h473 this->tensor_format_mkldnn_ = in MklPoolingOpBase()
478 MklTensorFormatToMklDnnDataFormat(this->tensor_format_mkldnn_); in MklPoolingOpBase()
607 MklTensorFormat tensor_format_mkldnn_; variable
H A Dmkl_avgpooling_op.cc121 this->tensor_format_mkldnn_, &output_tensor); in Compute()
268 this->tensor_format_mkldnn_, &output_tensor); in Compute()
H A Dmkl_maxpooling_op.cc143 this->tensor_format_mkldnn_, &output_tensor); in Compute()
319 this->tensor_format_mkldnn_, &output_tensor); in Compute()