Searched refs:iota_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | iota_test.cc | 70 const int64 iota_dim = std::get<2>(spec); in TEST_P() local
73 dimensions.insert(dimensions.begin() + iota_dim, num_elements); in TEST_P()
74 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
97 const int64 iota_dim = std::get<2>(spec); in TEST_P() local
100 dimensions.insert(dimensions.begin() + iota_dim, num_elements); in TEST_P()
101 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
122 const int64 iota_dim = GetParam(); in TEST_P() local
125 dimensions.insert(dimensions.begin() + iota_dim, num_elements); in TEST_P()
126 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 2505 const int64 iota_dim = in TEST_F() local
2508 EXPECT_THAT(iota_dim, ::testing::AnyOf(1, 2, 3)); in TEST_F()
|