Searched refs:src_dnn_shape (Results 1 – 1 of 1) sorted by relevance
98 MklDnnShape src_dnn_shape; in Compute() local99 GetMklShape(context, kIdxInput, &src_dnn_shape); in Compute()108 if (!src_dnn_shape.IsMklTensor()) { in Compute()111 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()115 ConvertMklToTF<T>(context, src_tensor, src_dnn_shape, in Compute()130 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute()131 memory::dims input_dims = src_dnn_shape.GetSizesAsMklDnnDims(); in Compute()155 auto input_format = src_dnn_shape.GetTfDataFormat(); in Compute()284 MklDnnShape src_dnn_shape; in SanityCheckInputs() local285 GetMklShape(context, kIdxInput, &src_dnn_shape); in SanityCheckInputs()[all …]