Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_tfconv_op.h105 if (input.IsReorderNeeded(output_tf_pd)) { in ConvertMklToTf()
Dmkl_pooling_ops_common.h741 input_gradient_dnn_data->IsReorderNeeded(original_output_format); in ConfigureInputGradient()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h855 if (input.IsReorderNeeded(output_tf_pd)) { in ConvertMklToTF()
1858 inline bool IsReorderNeeded(const memory::primitive_desc& op_pd) const { in IsReorderNeeded() function
1869 inline bool IsReorderNeeded(const memory::format& target_format) const { in IsReorderNeeded() function
1901 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
1915 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
1944 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
1960 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
2015 if (IsReorderNeeded(op_pd)) { in PrepareReorderToUserMemIfReq()