Searched refs:add_mkl_shape (Results 1 – 2 of 2) sorted by relevance
144 MklDnnShape add_mkl_shape; in Compute() local145 GetMklShape(ctx, kInputIndex_Add, &add_mkl_shape, native_format); in Compute()165 add_mkl_shape.IsMklTensor() in Compute()166 ? add_mkl_shape.GetMklLayout() in Compute()
878 MklDnnShape add_mkl_shape; in AllocateOutputTensor() local879 GetMklShape(context, kInputIndex_Add, &add_mkl_shape, native_format); in AllocateOutputTensor()888 if (!native_format && add_mkl_shape == *output_mkl_shape && in AllocateOutputTensor()891 add_mkl_shape, false)) { in AllocateOutputTensor()903 add_mkl_shape.IsMklTensor() in AllocateOutputTensor()904 ? add_mkl_shape.GetMklLayout() in AllocateOutputTensor()