Searched refs:MklShape (Results 1 – 7 of 7) sorted by relevance
153 class MklShape {155 MklShape() {} in MklShape() function156 TF_DISALLOW_COPY_AND_ASSIGN(MklShape); // Cannot copy158 ~MklShape() { in ~MklShape()788 typedef std::vector<MklShape> MklShapeList;804 const MklShape& mkl_shape) { in ConvertMklToTF()876 inline void GetMklShape(OpKernelContext* ctext, int n, MklShape* mklshape) { in GetMklShape()964 const MklShape& mkl_shape) { in AllocateOutputSetMklShape()999 const MklShape& mkl_shape) { in AllocateOutputSetMklShape()1090 const MklShape& mkl_shape, in MklSizesToTFSizes()[all …]
92 MklShape output_mkl_shape; in Compute()158 MklShape input_shape;
125 MklShape mkl_out_shape; in Compute()154 MklShape input_shape;315 MklShape mkl_out_shape; in Compute()346 MklShape out_backprop_shape;
106 MklShape mkl_out_shape, mkl_workspace_shape; in Compute()149 MklShape input_shape;278 MklShape mkl_output_shape; in Compute()306 MklShape input_shape, output_backprop_shape;
447 MklShape mkl_filter_shape; in Compute()542 MklShape mkl_output_mkl_shape; in Compute()551 MklShape mkl_output_mkl_shape; in Compute()626 MklShape mkl_output_mkl_shape; in Compute()643 MklShape mkl_filter_output_mkl_shape; in Compute()698 MklShape input_shape;
256 const MklShape* mklInputShape) { in Init()
430 TensorFormat data_format, const MklShape* mkl_in_shape);