Searched refs:mkl_shape (Results 1 – 2 of 2) sorted by relevance
190 const MklDnnShape* mkl_shape; in Compute() local196 mkl_shape = &input_shape_0; in Compute()202 mkl_shape = &input_shape_1; in Compute()213 if (mkl_shape->GetTfShape().num_elements() == in Compute()225 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()226 mkl_shape->GetSizesAsMklDnnDims(), in Compute()227 mkl_shape->GetTfDataFormat()); in Compute()229 auto output_mkl_md = mkl_shape->GetMklLayout(); in Compute()
804 const MklShape& mkl_shape) { in ConvertMklToTF() argument808 for (size_t j = 0; j < mkl_shape.GetDimension(); j++) { in ConvertMklToTF()810 output_shape.AddDim(mkl_shape.GetSizes()[mkl_shape.tf_dim_idx(j)]); in ConvertMklToTF()816 dnnLayout_t output_layout = static_cast<dnnLayout_t>(mkl_shape.GetTfLayout()); in ConvertMklToTF()821 mkl_shape.GetConvertedFlatData(output_layout, input_buffer, output_buffer); in ConvertMklToTF()833 const MklDnnShape& mkl_shape) { in ConvertMklToTF() argument836 if (!mkl_shape.IsMklTensor()) in ConvertMklToTF()839 TensorShape output_shape = mkl_shape.GetTfShape(); in ConvertMklToTF()849 auto input_mkl_md = mkl_shape.GetMklLayout(); in ConvertMklToTF()850 auto output_tf_md = mkl_shape.GetTfLayout(); in ConvertMklToTF()[all …]