Searched refs:device_arrangement_ (Results 1 – 2 of 2) sorted by relevance
34 buffer << std::endl << std::string("device arrangement = " + device_arrangement_.ToString()); in StandardToString()53 device_arrangement_ = device_arrangement; in Init()144 (void)device_arrangement_.Init(device_arrangement_shape); in RemoveElementEqualToOneInDeviceArrangement()156 return static_cast<int64_t>(device_arrangement_.GetDimSize()) - 1 - tensor_map_.GetDimByIdx(idx); in GetSliceDeviceDimensionByTensorDimensionIndex()161 …return device_arrangement_.GetDimByIdx(static_cast<uint64_t>(GetSliceDeviceDimensionByTensorDimens… in GetSliceNumByTensorDimensionIndex()192 for (int64_t i = static_cast<int64_t>(device_arrangement_.GetDimSize()) - 1; i >= 0; i--) { in ComputeArrangementByExpandedShape()200 device_arrangement_.GetExpandedShapeByExpandListRemoveLeft(re_map_expand_list); in ComputeArrangementByExpandedShape()227 Status status = tensor_layout_new.Init(device_arrangement_, *tensor_map_new_ptr, expanded_shape); in ExpandTensorShapeWithoutExtendDeviceArrangement()273 device_arrangement_.GetExpandShapeListPair(expanded_arrangement); in ExpandDeviceArrangement()330 int64_t divisor = device_arrangement_.GetDimByReverseIdx(LongToUlong(dim)); in slice_shape()[all …]
61 Arrangement device_arrangement() const { return device_arrangement_; } in device_arrangement()78 return (device_arrangement_ == tensor_layout.device_arrangement()); in IsSameDeviceArrangement()139 Arrangement device_arrangement_; variable