Home | History | Annotate | Download | only in xla

Lines Matching refs:ShapeIndex

63   int_tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
72 [&num_nodes](const ShapeIndex& /*index*/, bool data) {
89 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
97 [&num_nodes](const ShapeIndex& /*index*/, int* data) {
105 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
157 [&sum](const ShapeIndex& /*index*/, int data) { sum += data; });
169 [&sum](const ShapeIndex& /*index*/, int* data) { *data = 0; });
374 t.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
443 std::vector<ShapeIndex> v;
447 EXPECT_EQ(v, (std::vector<ShapeIndex>{{},
461 std::vector<ShapeIndex> v;
465 EXPECT_EQ(v, (std::vector<ShapeIndex>{
481 std::vector<ShapeIndex> v;
485 EXPECT_EQ(v, (std::vector<ShapeIndex>{
491 std::vector<ShapeIndex> v;
495 EXPECT_EQ(v, (std::vector<ShapeIndex>{