Home
last modified time | relevance | path

Searched defs:current_rank (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/distributed/_shard/sharded_tensor/
Dreshard.py19 def get_idx_from_placements(placements, current_rank) -> int: argument
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc400 const int current_rank = input_type.getRank(); in matchAndRewrite() local