Searched refs:SetOpMemDesc (Results 1 – 3 of 3) sorted by relevance
134 src_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc); in Compute()138 dst_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc); in Compute()389 orig_input_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc); in Compute()393 output_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc); in Compute()479 input_grad_dnn_data->SetOpMemDesc(orig_input_dims, memory::format::nhwc); in ConfigureInputGradient()
633 dnn_data_input->SetOpMemDesc(mkldnn_sizes, this->data_format_mkldnn_); in ConfigureInput()
1844 inline void SetOpMemDesc(const memory::dims& dim, memory::format fm) { in SetOpMemDesc() function