Searched defs:view_shape (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/ |
| D | pattern_conversion.cpp | 244 std::vector<int64_t> view_shape(ind_size, 1); in ReshapeToAdvancedIndexingFormat() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesModules.cpp | 17 c10::SymDimVector view_shape(indices.dim(), 1); in getStepTensor() local
|
| D | BatchRulesViews.cpp | 470 c10::SmallVector<c10::SymInt> view_shape(size_.size(), /*init_value*/1); in expand_batch_rule() local
|
| /external/pytorch/aten/src/ATen/ |
| D | LegacyBatchingRegistrations.cpp | 182 VmapDimVector view_shape(size_physical.size(), 1); in expand_batching_rule() local
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 6339 std::array<int64_t, 3> view_shape = {1, N * groups, N ? -1 : 1}; in group_norm_mean_jvp() local
6352 std::vector<int64_t> view_shape = {1, N * groups, N ? -1 : 1}; in group_norm_invstd_jvp() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 3664 std::vector<c10::SymInt> view_shape(size, 1); in meshgrid() local
|