Home
last modified time | relevance | path

Searched refs:VmapSymDimVector (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/
DLegacyVmapTransforms.h43 using VmapSymDimVector = SmallVector<c10::SymInt, kVmapStaticDimVecSize>; variable
/external/pytorch/aten/src/ATen/functorch/
DLegacyVmapTransforms.h50 using VmapSymDimVector = SmallVector<c10::SymInt, kVmapStaticDimVecSize>; variable
DBatchRulesRandomness.cpp137 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in unary_pointwise_random_batch_rule()
166 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in tensor_like_random_batch_rule()
195 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in native_dropout_batching_rule()
DBatchRulesHelper.cpp79 VmapSymDimVector new_sizes(tensor.sym_sizes().begin(), tensor.sym_sizes().end()); in maybePadToLogicalRank()
DBatchRulesBinaryOps.cpp70 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in apply()