Searched refs:iota_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | iota_test.cc | 95 const int64 iota_dim = std::get<2>(spec); in XLA_TEST_P() local
103 dimensions.insert(dimensions.begin() + iota_dim, num_elements); in XLA_TEST_P()
104 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in XLA_TEST_P()
127 const int64 iota_dim = std::get<2>(spec); in XLA_TEST_P() local
135 dimensions.insert(dimensions.begin() + iota_dim, num_elements); in XLA_TEST_P()
136 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3931 auto iota_dim = absl::c_find_if( in HandleReshape() local
3940 if (iota_dim != common_factors.end() && next_dim != common_factors.end()) { in HandleReshape()
3944 for (int64 dim = (iota_dim + 1)->second - 1; dim >= iota_dim->second; in HandleReshape()
|
| D | algebraic_simplifier_test.cc | 3185 const int64 iota_dim = in TEST_F() local
3188 EXPECT_THAT(iota_dim, ::testing::AnyOf(1, 2, 3)); in TEST_F()
|