Searched refs:layout_transfer (Results 1 – 2 of 2) sorted by relevance
80 RedistributionLayoutTransfer layout_transfer; in InferTensorRedistributionOperatorList() local81 Status status = layout_transfer.Init(from_, to_); in InferTensorRedistributionOperatorList()87 if (layout_transfer.IsDynamicShape()) { in InferTensorRedistributionOperatorList()88 from_layout = layout_transfer.from_in(); in InferTensorRedistributionOperatorList()89 to_layout = layout_transfer.to_in(); in InferTensorRedistributionOperatorList()91 …std::shared_ptr<ReshapeLayoutTransfer> ptr = layout_transfer.UnifyDeviceArrangementAndTensorShape(… in InferTensorRedistributionOperatorList()
49 bool layout_transfer() const { return layout_transfer_; } in layout_transfer() function