Home
last modified time | relevance | path

Searched defs:offset_dim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_embedding_ops.py85 offset_dim: int = 0 variable in _MaskPartial
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc932 int64_t offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughOperandToGatherOutputOrScatterUpdate() local
974 int64_t offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughGatherOutputOrScatterUpdateToOperand() local
H A Dshape_inference.cc3351 int64_t offset_dim = dim_numbers.offset_dims(i); in ValidateGatherDimensionNumbers() local