Searched refs:weights_width (Results 1 – 2 of 2) sorted by relevance
39 int weights_height, int weights_width, in ComputeSpaceToBatchParams() argument51 dilation_factors_h_w[1] * (weights_width - 1) + 1; in ComputeSpaceToBatchParams()
817 const int weights_width = attr.weights.shape.i; in Parse() local821 if (input->tensor.shape.DimensionsProduct() / batch_size != weights_width) { in Parse()833 BHWC(input->tensor.shape.b, 1, 1, weights_width); in Parse()