Searched refs:VmapSymDimVector (Results 1 – 5 of 5) sorted by relevance
43 using VmapSymDimVector = SmallVector<c10::SymInt, kVmapStaticDimVecSize>; variable
50 using VmapSymDimVector = SmallVector<c10::SymInt, kVmapStaticDimVecSize>; variable
137 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()
79 VmapSymDimVector new_sizes(tensor.sym_sizes().begin(), tensor.sym_sizes().end()); in maybePadToLogicalRank()
70 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in apply()