Searched refs:add_shape_compute_mapping (Results 1 – 1 of 1) sorted by relevance
1212 def add_shape_compute_mapping(operator_schema: str, func: Callable): function1226 add_shape_compute_mapping(1230 add_shape_compute_mapping(1233 add_shape_compute_mapping(1236 add_shape_compute_mapping(1240 add_shape_compute_mapping(1243 add_shape_compute_mapping("prim::NumToTensor.bool(bool a) -> Tensor", zero_dim_tensor)1244 add_shape_compute_mapping(1248 add_shape_compute_mapping(1252 add_shape_compute_mapping([all …]