Searched refs:dims_w (Results 1 – 1 of 1) sorted by relevance
1587 auto dims_w = weights.shape_; in BinaryTensorOpWeight() local1605 VLOG(2) << "weights dims: " << DebugString(dims_w) in BinaryTensorOpWeight()1608 if (dims_w.nbDims == dims_t.nbDims + 1) { in BinaryTensorOpWeight()1609 if (dims_w.d[0] == 1) { in BinaryTensorOpWeight()1610 for (int i = 1; i < dims_w.nbDims; i++) { in BinaryTensorOpWeight()1611 dims_w.d[i - 1] = dims_w.d[i]; in BinaryTensorOpWeight()1613 dims_w.nbDims--; in BinaryTensorOpWeight()1620 if (dims_w.nbDims == dims_t.nbDims && dims_w.d[0] == dims_t.d[0]) { in BinaryTensorOpWeight()1623 for (int i = 1; i < dims_w.nbDims; i++) { in BinaryTensorOpWeight()1624 if (dims_w.d[i] != dims_t.d[i]) { in BinaryTensorOpWeight()[all …]