MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetMutableSubshape

Method GetMutableSubshape

tensorflow/compiler/xla/shape_util.cc:807–815  ·  view source on GitHub ↗

static */

Source from the content-addressed store, hash-verified

805}
806
807/* static */ Shape* ShapeUtil::GetMutableSubshape(Shape* shape,
808 ShapeIndexView index) {
809 Shape* return_shape = shape;
810 for (auto i : index) {
811 CHECK(return_shape->IsTuple());
812 return_shape = return_shape->mutable_tuple_shapes(i);
813 }
814 return return_shape;
815}
816
817/* static */
818bool ShapeUtil::IsLeafIndex(const Shape& shape, const ShapeIndex& index) {

Callers

nothing calls this directly

Calls 2

mutable_tuple_shapesMethod · 0.80
IsTupleMethod · 0.45

Tested by

no test coverage detected