Searched refs:codegen_shape_tuple (Results 1 – 3 of 3) sorted by relevance
774 opaque_metadata_str = self.codegen_shape_tuple(opaque_metadata_list)1426 def codegen_shape_tuple(self, shape: Tuple[Expr, ...]) -> str: member in CppWrapperCpu1585 size = self.codegen_shape_tuple(shape)1586 stride = self.codegen_shape_tuple(orig_stride)1653 size = self.codegen_shape_tuple(shape)1654 stride = self.codegen_shape_tuple(stride)1681 self.codegen_shape_tuple(shape),1682 self.codegen_shape_tuple(stride),1692 size = self.codegen_shape_tuple(size_list)1693 stride = self.codegen_shape_tuple(stride_list)
183 wrapper.codegen_shape_tuple(sympy_grid),184 wrapper.codegen_shape_tuple(646 size = self.codegen_shape_tuple(buf.get_size())647 stride = self.codegen_shape_tuple(buf.get_stride())1087 def codegen_shape_tuple(self, shape: Tuple[Expr, ...]) -> str: member in WrapperCodeGen1097 self.codegen_shape_tuple(shape),1098 self.codegen_shape_tuple(stride),1116 size = self.codegen_shape_tuple(size)1117 stride = self.codegen_shape_tuple(stride)
4472 size = V.graph.wrapper_code.codegen_shape_tuple(self.get_size())4473 stride = V.graph.wrapper_code.codegen_shape_tuple(self.get_stride())