Searched refs:dnn_shape_src (Results 1 – 2 of 2) sorted by relevance
444 MklDnnShape dnn_shape_src; in Compute() local445 GetMklShape(context, src_index, &dnn_shape_src); in Compute()466 if (dnn_shape_src.IsMklTensor()) { in Compute()467 src_md = dnn_shape_src.GetMklLayout(); in Compute()468 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()498 if (is_src_reordered || dnn_shape_src.IsMklTensor()) { in Compute()503 if (dnn_shape_src.IsMklTensor()) { in Compute()504 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()505 dnn_shape_src.GetSizesAsMklDnnDims(), in Compute()506 dnn_shape_src.GetTfDataFormat()); in Compute()[all …]
692 MklDnnShape dnn_shape_src; in Compute() local693 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()695 if (dnn_shape_src.IsMklTensor()) { in Compute()696 tf_shape_src = dnn_shape_src.GetTfShape(); in Compute()697 OP_REQUIRES(context, dnn_shape_src.GetDimension() == 4, in Compute()732 if (dnn_shape_src.IsMklTensor()) in Compute()733 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()753 if (dnn_shape_src.IsMklTensor()) { in Compute()754 if (dnn_shape_src.IsTensorInNCHWFormat()) { in Compute()768 dnn_shape_src.IsMklTensor() in Compute()[all …]