Searched defs:inp_dim (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | dropout.h | 63 auto inp_dim = input.dim(); in _dropoutNd_helper() local
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 4251 def start_index(index, out_dim, inp_dim): argument
4254 def end_index(index, out_dim, inp_dim): argument
4335 def start_index(index, out_dim, inp_dim): argument
4338 def end_index(index, out_dim, inp_dim): argument
4498 def start_index(index, out_dim, inp_dim): argument
4501 def end_index(index, out_dim, inp_dim): argument
|