Searched defs:orig_w (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | MKLDNNConversions.cpp | 617 const ideep::tensor orig_w = itensor_view_from_dense(weight_); in mkl_reorder_linear_weight() local
|
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_ir.py | 1126 def create(cls, x, packed_w, orig_w, B, batch_size): argument
|
| D | ir.py | 6279 def create(cls, x, packed_w, orig_w, B, batch_size): argument
|