Searched refs:manual_shapes (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 224 target='CPU', slice_mode='batch_slice', manual_shapes=None, argument
265 if not manual_shapes:
268 if not isinstance(manual_shapes, tuple):
271 for dim in manual_shapes:
273 self.gatherv2.add_prim_attr("manual_split", manual_shapes)
274 self.embeddinglookup.add_prim_attr("manual_split", manual_shapes)
|
| D | thor_layer.py | 721 target='CPU', slice_mode='batch_slice', manual_shapes=None, argument
764 if not manual_shapes:
767 if not isinstance(manual_shapes, tuple):
770 for dim in manual_shapes:
772 self.gatherv2.add_prim_attr("manual_split", manual_shapes)
773 self.embeddinglookup.add_prim_attr("manual_split", manual_shapes)
|