Searched defs:quant_dim (Results 1 – 3 of 3) sorted by relevance
82 ArrayRef<double> max, int quant_dim, in GetQuantizedType()156 Attribute max, int quant_dim, in GetQuantizedTypeAttr()207 Type target, int quant_dim) { in ResetAxisAndBroadcast()346 Type GetUniformQuantizedPerAxisTypeForWeight(ElementsAttr attr, int quant_dim, in GetUniformQuantizedPerAxisTypeForWeight()382 int32_t quant_dim = -1; in GetUniformQuantizedTypeForBias() local
101 int quant_dim = -1; in matchAndRewrite() local
219 int quant_dim = -1; in matchAndRewrite() local