Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc126 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute()
381 memory::desc original_output_md = orig_output_dnn_shape.GetCurLayout(); in Compute()
385 memory::desc orig_input_md = orig_input_dnn_shape.GetCurLayout(); in Compute()
476 memory::desc input_grad_md = input_grad_dnn_shape.GetCurLayout(); in ConfigureInputGradient()
Dmkl_maxpooling_op.cc162 lt_user_input = (dnnLayout_t)input_shape.GetCurLayout(); in MklCreateLayoutsAndPrimitives()
328 lt_input_user = (dnnLayout_t)input_shape.GetCurLayout(); in MklCreateLayouts()
338 lt_outbackprop_user = (dnnLayout_t)output_backprop_shape.GetCurLayout(); in MklCreateLayouts()
Dmkl_conv_grad_bias_ops.cc173 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout()); in MklCreateInputLayouts()
Dmkl_avgpooling_op.cc170 lt_user_input = (dnnLayout_t)input_shape.GetCurLayout(); in MklCreateLayoutsAndPrimitives()
385 lt_user_outbackprop = (dnnLayout_t)out_backprop_shape.GetCurLayout(); in MklCreateLayoutsAndPrimitives()
Dmkl_conv_ops.cc708 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout()); in MklCreateInputLayouts()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h238 const dnnLayout_t GetCurLayout() const { in GetCurLayout() function
662 inline const memory::desc GetCurLayout() const { in GetCurLayout() function