Searched refs:tensor_map_origin_ (Results 1 – 2 of 2) sorted by relevance
43 buffer << std::endl << std::string("tensor map origin = " + tensor_map_origin_.ToString()); in OriginToString()51 tensor_map_origin_ = tensor_map; in Init()76 if (tensor_map_origin_.Init(tensor_map) != SUCCESS) { in InitFromVector()82 if (Init(device_arrangement_origin_, tensor_map_origin_, tensor_shape_origin_) != SUCCESS) { in InitFromVector()89 …if (tensor_map_origin_.GetMaxItem() >= static_cast<int64_t>(device_arrangement_origin_.GetDimSize(… in IsValidTensorLayout()93 if (tensor_map_origin_.GetDimSize() != tensor_shape_origin_.GetDimSize()) { in IsValidTensorLayout()126 Shape tensor_map_shape = tensor_map_origin_.array(); in RemoveElementEqualToOneInDeviceArrangement()410 Shape tensor_map(tensor_map_origin_.GetDimSize(), -1); in TransferRepeatLayout()
67 Map origin_tensor_map() const { return tensor_map_origin_; } in origin_tensor_map()142 Map tensor_map_origin_; variable