Home
last modified time | relevance | path

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

/external/pytorch/torch/distributed/tensor/_ops/
D_embedding_ops.py85 offset_dim: int = 0 variable in _MaskPartial
/external/tensorflow/tensorflow/compiler/xla/service/
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
Dshape_inference.cc3351 int64_t offset_dim = dim_numbers.offset_dims(i); in ValidateGatherDimensionNumbers() local