Searched refs:dynamic_dimensions (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 568 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
590 ContainersEqual(result_shape.dynamic_dimensions(), {false, true, false})) in TEST_F()
611 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
634 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true})) in TEST_F()
662 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
703 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), in TEST_F()
743 ContainersEqual(result_shape.dynamic_dimensions(), {true, true, false})) in TEST_F()
768 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
795 ContainersEqual(result_shape.dynamic_dimensions(), {true, false, false})) in TEST_F()
834 ContainersEqual(result_shape.dynamic_dimensions(), {true, false, false})) in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 165 const std::vector<bool>& dynamic_dimensions) { in MakeShape() argument
166 return MakeValidatedShape(element_type, dimensions, dynamic_dimensions) in MakeShape()
180 const std::vector<bool>& dynamic_dimensions) { in MakeValidatedShape() argument
183 for (int i = 0; i < dynamic_dimensions.size(); ++i) { in MakeValidatedShape()
184 shape.set_dynamic_dimension(i, dynamic_dimensions[i]); in MakeValidatedShape()
|
| D | shape.h | 75 const std::vector<bool>& dynamic_dimensions() const { in dynamic_dimensions() function
|
| D | shape_util.h | 378 const std::vector<bool>& dynamic_dimensions);
387 const std::vector<bool>& dynamic_dimensions);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 298 std::vector<bool>* dynamic_dimensions);
3395 std::vector<bool>* dynamic_dimensions) { in ParseDimensionSizes() argument
3407 dynamic_dimensions->push_back(is_dynamic); in ParseDimensionSizes()
3552 std::vector<bool> dynamic_dimensions; in ParseShape() local
3553 if (!ParseDimensionSizes(&dimension_sizes, &dynamic_dimensions)) { in ParseShape()
3559 result->set_dynamic_dimension(i, dynamic_dimensions[i]); in ParseShape()
|