Searched refs:summand_mkl_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 1752 MklDnnShape summand_mkl_shape; in AllocateOutputTensor() local
1753 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
1754 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor()
1762 summand_mkl_shape.SetMklLayout(&dst_md); in AllocateOutputTensor()
1763 summand_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
1769 context, summand_idx, 0, output_tensor, summand_mkl_shape, false), in AllocateOutputTensor()
1784 MklDnnShape summand_mkl_shape; in AllocateOutputTensor() local
1785 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
1817 summand_mkl_shape.IsMklTensor() in AllocateOutputTensor()
1818 ? summand_mkl_shape.GetMklLayout() in AllocateOutputTensor()
|