Searched refs:src1_dims (Results 1 – 1 of 1) sorted by relevance
155 memory::dims src1_dims; in Compute() local157 src1_dims = TFShapeToMklDnnDimsInNCHW(src1_tensor.shape(), in Compute()160 src1_dims = TFShapeToMklDnnDimsInNCDHW(src1_tensor.shape(), in Compute()163 md1 = memory::desc(src1_dims, MklDnnType<T>(), src2_mkl_data_format); in Compute()