Searched refs:input_elems (Results 1 – 3 of 3) sorted by relevance
274 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in XLA_TEST_F() local275 iota(input_elems.begin(), input_elems.end(), 1.0f); in XLA_TEST_F()276 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in XLA_TEST_F()337 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local338 iota_int_init_value(input_elems, 1); in RunTest()339 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest()397 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local398 iota_int_init_value(input_elems, 1); in RunTest()399 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest()463 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local[all …]
107 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in TestImpl() local109 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in TestImpl()
1473 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in TEST_P() local1474 std::iota(input_elems.begin(), input_elems.end(), -7); in TEST_P()1475 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in TEST_P()