Searched refs:ComputeSums (Results 1 – 1 of 1) sorted by relevance
73 XlaOp ComputeSums(XlaOp operand, XlaOp init_value, in ComputeSums() function164 auto pooled = ComputeSums(padded_operand, init_value, kernel_size, stride, in AvgPool()