Searched defs:tensor_redistribution (Results 1 – 7 of 7) sorted by relevance
86 TensorRedistribution tensor_redistribution; in TEST_F() local139 TensorRedistribution tensor_redistribution; in TEST_F() local192 TensorRedistribution tensor_redistribution; in TEST_F() local
64 RedistributionLayoutTransfer tensor_redistribution; in RedistributionLayoutTransferTestFunction() local
64 ReshapeLayoutTransfer tensor_redistribution; in InferUnifiedLayout() local
117 TensorRedistribution tensor_redistribution(!is_generating_costs_, true); in ComputeReplaceOp() local
119 TensorRedistribution tensor_redistribution(false); in GetRedistributionCost() local
672 TensorRedistribution tensor_redistribution(false, true); in GetForwardCommCost() local710 TensorRedistribution tensor_redistribution(false, true); in GetForwardComputationCost() local
430 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution, in Redistribution()544 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node) { in StepRedistribution()2103 TensorRedistribution tensor_redistribution; in InferSensRedistribution() local2548 TensorRedistribution tensor_redistribution; in ParallelCommunication() local