Searched refs:compute_operand_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 434 self.compute_operand_shape(
602 self.compute_operand_shape(out_op_id, out_dim, flex_name(in_op_id, in_dim))
604 def compute_operand_shape(self, op_id, dim, expr): member in _NnapiSerializer
1219 self.compute_operand_shape(out_id, idx, shape)
1738 self.compute_operand_shape(out_id, dim, size_arg[dim - 2])
1740 self.compute_operand_shape(
2187 self.compute_operand_shape(
2193 self.compute_operand_shape(
2200 self.compute_operand_shape(
2206 self.compute_operand_shape(
|