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

Method SubShapeTree

tensorflow/compiler/xla/shape_tree.h:671–678  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

669
670template <typename T>
671StatusOr<ShapeTree<T>> ShapeTree<T>::SubShapeTree(
672 const ShapeIndex& index) const {
673 TF_ASSIGN_OR_RETURN(const Shape* sub_shape,
674 ShapeUtil::TryGetSubshape(shape(), index));
675 ShapeTree<T> sub_shape_tree(*sub_shape);
676 sub_shape_tree.CopySubtreeFrom(*this, index, {});
677 return std::move(sub_shape_tree);
678}
679
680template <typename T>
681bool ShapeTree<T>::operator==(const ShapeTree<T>& other) const {

Callers 1

SubShapedBufferMethod · 0.80

Calls 1

CopySubtreeFromMethod · 0.80

Tested by

no test coverage detected