Searched refs:SetOpMemDesc (Results 1 – 3 of 3) sorted by relevance
138 src_dnn_data.SetOpMemDesc(input_dims, memory::format_tag::nhwc); in Compute()143 dst_dnn_data.SetOpMemDesc(input_dims, memory::format_tag::nhwc); in Compute()385 orig_input_dnn_data.SetOpMemDesc(orig_input_dims, in Compute()391 output_dnn_data.SetOpMemDesc(orig_input_dims, memory::format_tag::nhwc); in Compute()486 input_grad_dnn_data->SetOpMemDesc(orig_input_dims, in ConfigureInputGradient()
635 dnn_data_input->SetOpMemDesc(mkldnn_sizes, this->data_format_mkldnn_); in ConfigureInput()
1427 inline void SetOpMemDesc(const memory::dims& dim, memory::format_tag fm) { in SetOpMemDesc() function