Searched refs:src2_dims (Results 1 – 1 of 1) sorted by relevance
141 memory::dims src2_dims; in Compute() local143 src2_dims = TFShapeToMklDnnDimsInNCHW(src2_tensor.shape(), in Compute()146 src2_dims = TFShapeToMklDnnDimsInNCDHW(src2_tensor.shape(), in Compute()149 md2 = memory::desc(src2_dims, MklDnnType<T>(), src1_mkl_data_format); in Compute()