Searched refs:dnn_shape_diff_dst (Results 1 – 2 of 2) sorted by relevance
598 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute() local600 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute()625 if (!dnn_shape_src.IsMklTensor() && !dnn_shape_diff_dst.IsMklTensor()) { in Compute()631 !dnn_shape_diff_dst.IsMklTensor()) { in Compute()644 dnn_shape_diff_dst.IsMklTensor()) { in Compute()645 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()648 dnn_shape_diff_dst.GetTfDataFormat(); in Compute()662 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()672 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) { in Compute()709 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) { in Compute()[all …]
1080 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute() local1082 GetMklShape(context, kDiffDstIndex, &dnn_shape_diff_dst, native_format); in Compute()1085 if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()1086 tf_shape_diff_dst = dnn_shape_diff_dst.GetTfShape(); in Compute()1088 context, dnn_shape_diff_dst.GetDimension() == 4, in Compute()1135 } else if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()1136 depth_ = dnn_shape_diff_dst.DimSize(MklDnnDims::Dim_C); in Compute()1166 dnn_shape_diff_dst.IsMklTensor() in Compute()1167 ? dnn_shape_diff_dst.GetSizesAsMklDnnDims() in Compute()1177 dnn_shape_diff_dst.IsMklTensor() in Compute()[all …]