Searched refs:src2_tensor (Results 1 – 1 of 1) sorted by relevance
53 const Tensor& src2_tensor = MklGetInput(ctx, src2_idx); in Compute() local63 : src2_tensor.dims(); in Compute()69 : src2_tensor.shape(); in Compute()87 float user_i2 = (src2_tensor.scalar<T>()()); in Compute()142 if (src2_tensor.dims() == 4) { in Compute()143 src2_dims = TFShapeToMklDnnDimsInNCHW(src2_tensor.shape(), in Compute()146 src2_dims = TFShapeToMklDnnDimsInNCDHW(src2_tensor.shape(), in Compute()173 src2.SetUsrMem(md2, &src2_tensor); in Compute()