Searched refs:MklDnnDims3D (Results 1 – 4 of 4) sorted by relevance
165 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_batch; in GetInputSizeInMklOrder()166 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_depth; in GetInputSizeInMklOrder()167 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_planes; in GetInputSizeInMklOrder()168 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_rows; in GetInputSizeInMklOrder()169 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_cols; in GetInputSizeInMklOrder()266 mkldnn_sizes[MklDnnDims3D::Dim3d_O] = filter_out_depth; in GetFilterSizeInMklOrder()267 mkldnn_sizes[MklDnnDims3D::Dim3d_I] = filter_in_depth; in GetFilterSizeInMklOrder()268 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = filter_planes; in GetFilterSizeInMklOrder()269 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = filter_rows; in GetFilterSizeInMklOrder()270 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = filter_cols; in GetFilterSizeInMklOrder()[all …]
522 {bwd_output_dims[MklDnnDims3D::Dim3d_D], in Compute()523 bwd_output_dims[MklDnnDims3D::Dim3d_H], in Compute()524 bwd_output_dims[MklDnnDims3D::Dim3d_W], in Compute()525 bwd_output_dims[MklDnnDims3D::Dim3d_I], in Compute()526 bwd_output_dims[MklDnnDims3D::Dim3d_O]}); in Compute()
628 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_md.data.dims[0]; in ConfigureInput()629 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_md.data.dims[1]; in ConfigureInput()630 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_md.data.dims[2]; in ConfigureInput()631 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_md.data.dims[3]; in ConfigureInput()632 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_md.data.dims[4]; in ConfigureInput()
112 } MklDnnDims3D; typedef545 return MklDnnDims3D::Dim3d_N; in GetMklDnnTensor3DDimIndex()547 return MklDnnDims3D::Dim3d_C; in GetMklDnnTensor3DDimIndex()549 return MklDnnDims3D::Dim3d_D; in GetMklDnnTensor3DDimIndex()551 return MklDnnDims3D::Dim3d_H; in GetMklDnnTensor3DDimIndex()553 return MklDnnDims3D::Dim3d_W; in GetMklDnnTensor3DDimIndex()681 MklDnnDims3D::Dim3d_D; in SetTfDimOrder()683 MklDnnDims3D::Dim3d_H; in SetTfDimOrder()685 MklDnnDims3D::Dim3d_W; in SetTfDimOrder()687 MklDnnDims3D::Dim3d_C; in SetTfDimOrder()[all …]