Searched refs:input1_strategy (Results 1 – 3 of 3) sorted by relevance
62 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F() local63 ASSERT_EQ(input0_strategy, input1_strategy); in TEST_F()75 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F() local76 ASSERT_EQ(input0_strategy[0], input1_strategy[1]); in TEST_F()78 ASSERT_EQ(input1_strategy.size(), 2); in TEST_F()90 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F() local91 ASSERT_EQ(input0_strategy[1], input1_strategy[0]); in TEST_F()93 ASSERT_EQ(input1_strategy.size(), 1); in TEST_F()105 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F() local107 ASSERT_EQ(input1_strategy[0], 1); in TEST_F()[all …]
96 Dimensions input1_strategy = {input0_strategy.at(1)}; in GenerateOpStrategies() local99 tmp_strategy.push_back(input1_strategy); // input1 in GenerateOpStrategies()
1168 Dimensions input1_strategy = sp->GetInputDim()[1]; in GenerateStrategiesForBroadcastRight() local1172 (void)input1_strategy.erase(input1_strategy.begin(), in GenerateStrategiesForBroadcastRight()1173 input1_strategy.begin() + static_cast<different_type>(size_diff)); in GenerateStrategiesForBroadcastRight()1178 input1_strategy[i] = 1; in GenerateStrategiesForBroadcastRight()1185 tmp_strategy.push_back(input1_strategy); // input1 in GenerateStrategiesForBroadcastRight()1228 Dimensions input1_strategy = sp->GetInputDim()[1]; in GenerateStrategiesForBroadcastBoth() local1235 input1_strategy[i] = 1; in GenerateStrategiesForBroadcastBoth()1238 sp->ResetInputs({input0_strategy, input1_strategy}); in GenerateStrategiesForBroadcastBoth()