Home
last modified time | relevance | path

Searched defs:shapeVec (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesRandomness.cpp25 c10::SmallVector<SymInt> shapeVec(1, maybe_layer->batchSize()); in random_batching_rule() local
137 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in unary_pointwise_random_batch_rule() local
166 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in tensor_like_random_batch_rule() local
195 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in native_dropout_batching_rule() local
H A DBatchRulesBinaryOps.cpp70 VmapSymDimVector shapeVec(1, maybe_layer->batchSize()); in apply() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/
H A Dcodegen_utils.cc94 SmallVector<Value, 4> shapeVec; in calcMultiDimIndex() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupWrapper.cpp129 auto shapeVec = std::vector<int64_t>(); in deserialize_fingerprint() local
/aosp_15_r20/external/pytorch/android/pytorch_android/src/main/cpp/
H A Dpytorch_jni_common.cpp98 std::vector<int64_t> shapeVec{}; in newAtTensor() local