Searched refs:summand_mkl_shape (Results 1 – 1 of 1) sorted by relevance
1769 MklDnnShape summand_mkl_shape; in AllocateOutputTensor() local1770 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()1771 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor()1772 if (summand_mkl_shape.IsMklTensor()) { in AllocateOutputTensor()1778 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()1779 summand_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()1782 summand_mkl_shape); in AllocateOutputTensor()1800 MklDnnShape summand_mkl_shape; in AllocateOutputTensor() local1801 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()1822 summand_mkl_shape.IsMklTensor() in AllocateOutputTensor()[all …]