Searched defs:input_elems (Results 1 – 2 of 2) sorted by relevance
442 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in XLA_TEST_F() local505 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local565 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local631 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local701 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in RunTest() local765 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in RunTest() local834 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in RunTest() local903 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in RunTest() local974 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest() local1049 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() local