Searched defs:scales_shape (Results 1 – 2 of 2) sorted by relevance
22 const TensorShape& scales_shape, in QuantizationAxisAndShapeValid()
78 const std::vector<int> scales_shape = {1, 1, 1, in ProcessPerChannelQuantizedWeights() local