Lines Matching full:shape

3 func @reduce_op_args_num_mismatch(%shape : !shape.shape, %init : !shape.size) {
5   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> !shape.size {
6     ^bb0(%index: index, %dim: !shape.size):
7       shape.yield %dim : !shape.size
14 func @reduce_op_arg0_wrong_type(%shape : !shape.shape, %init : !shape.size) {
16   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> !shape.size {
17     ^bb0(%index: f32, %dim: !shape.size, %acc: !shape.size):
18       %new_acc = "shape.add"(%acc, %dim)
19           : (!shape.size, !shape.size) -> !shape.size
20       shape.yield %new_acc : !shape.size
27 func @reduce_op_arg1_wrong_type(%shape : !shape.shape, %init : !shape.size) {
29   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> !shape.size {
30     ^bb0(%index: index, %dim: f32, %lci: !shape.size):
31       shape.yield
38 func @reduce_op_arg1_wrong_type(%shape : tensor<?xindex>, %init : index) {
40   %num_elements = shape.reduce(%shape, %init) : tensor<?xindex> -> index {
42       shape.yield
49 func @reduce_op_init_type_mismatch(%shape : !shape.shape, %init : f32) {
51   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> f32 {
52     ^bb0(%index: index, %dim: !shape.size, %lci: !shape.size):
53       shape.yield
60 func @yield_op_args_num_mismatch(%shape : !shape.shape, %init : !shape.size) {
62   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> !shape.size {
63     ^bb0(%index: index, %dim: !shape.size, %lci: !shape.size):
64       shape.yield %dim, %dim : !shape.size, !shape.size
71 func @yield_op_type_mismatch(%shape : !shape.shape, %init : !shape.size) {
73   %num_elements = shape.reduce(%shape, %init) : !shape.shape -> !shape.size {
74     ^bb0(%index: index, %dim: !shape.size, %lci: !shape.size):
76       shape.yield %c0 : index
85   %w0 = shape.assuming_all
91 func @shape_of(%value_arg : !shape.value_shape,
93 …ne of the operands can hold error values then the result must be of type `shape` to propagate them…
94   %0 = shape.shape_of %value_arg : !shape.value_shape -> tensor<?xindex>
100 func @rank(%arg : !shape.shape) {
102   %0 = shape.rank %arg : !shape.shape -> index
109   %c0 = shape.const_size 0
111   %result = shape.get_extent %arg, %c0 : tensor<?xindex>, !shape.size -> index
117 func @mul(%lhs : !shape.size, %rhs : index) -> index {
119   %result = shape.mul %lhs, %rhs : !shape.size, index -> index
125 func @num_elements(%arg : !shape.shape) -> index {
127   %result = shape.num_elements %arg : !shape.shape -> index
133 func @add(%lhs : !shape.size, %rhs : index) -> index {
135   %result = shape.add %lhs, %rhs : !shape.size, index -> index
141 func @broadcast(%arg0 : !shape.shape, %arg1 : !shape.shape) -> tensor<?xindex> {
142 …ne of the operands can hold error values then the result must be of type `shape` to propagate them…
143   %result = shape.broadcast %arg0, %arg1
144       : !shape.shape, !shape.shape -> tensor<?xindex>
151 func @broadcast(%arg0 : !shape.shape, %arg1 : tensor<?xindex>) -> tensor<?xindex> {
152 …ne of the operands can hold error values then the result must be of type `shape` to propagate them…
153   %result = shape.broadcast %arg0, %arg1
154       : !shape.shape, tensor<?xindex> -> tensor<?xindex>