Searched defs:base_shape (Results 1 – 3 of 3) sorted by relevance
361 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
159 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape()1775 Shape base_shape = in InferConvolveShape() local
2824 const Shape& window_shape, const Window& window, const Shape& base_shape, in IterateThroughWindow()