Lines Matching defs:shape
133 %shape = memref.alloc() : memref<2x2xf64>
136 linalg.fill ins(%v1 : f64) outs(%shape : memref<2x2xf64>)
146 call @pooling_on_buffers(%input, %shape, %output) :
406 def pooling_on_buffers(input, shape, output):
408 input, shape, outs=[output], strides=[2, 4], dilations=[1, 2]
440 def pooling_on_buffers(input, shape, output):
443 shape,
480 def pooling_on_buffers(input, shape, output):
481 linalg.pooling_nhwc_min(input, shape, outs=[output], strides=[2, 4])
512 def pooling_on_buffers(input, shape, output):
514 input, shape, outs=[output], strides=[2, 4], emit_generic=True