Searched refs:GetSliceDeviceDimensionByTensorDimensionIndex (Results 1 – 2 of 2) sorted by relevance
132 int64_t GetSliceDeviceDimensionByTensorDimensionIndex(uint64_t idx) const;
155 int64_t TensorLayout::GetSliceDeviceDimensionByTensorDimensionIndex(uint64_t idx) const { in GetSliceDeviceDimensionByTensorDimensionIndex() function in mindspore::parallel::TensorLayout161 …device_arrangement_.GetDimByIdx(static_cast<uint64_t>(GetSliceDeviceDimensionByTensorDimensionInde… in GetSliceNumByTensorDimensionIndex()