Home
last modified time | relevance | path

Searched refs:GetShapeLayoutVector (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dshape_util.h69 StatusOr<std::vector<int>> GetShapeLayoutVector(const xla::Shape& shape);
H A Dshape_util.cc166 StatusOr<std::vector<int>> GetShapeLayoutVector(const xla::Shape& shape) { in GetShapeLayoutVector() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/
H A Draw_api_test.cc488 std::vector<int> layout = GetShapeLayoutVector(literal.shape()).ValueOrDie(); in TEST()
515 std::vector<int> layout = GetShapeLayoutVector(literal.shape()).ValueOrDie(); in TEST()