Searched refs:input_elems (Results 1 – 2 of 2) sorted by relevance
442 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in XLA_TEST_F() local443 iota(input_elems.begin(), input_elems.end(), 1.0f); in XLA_TEST_F()444 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in XLA_TEST_F()505 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local506 iota_int_init_value(input_elems, 1); in RunTest()507 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest()565 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local566 iota_int_init_value(input_elems, 1); in RunTest()567 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest()631 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local[all …]
1396 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in TEST_P() local1397 std::iota(input_elems.begin(), input_elems.end(), -7); in TEST_P()1398 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in TEST_P()