Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc209 int64 shape_leaves = RequiredLeaves(shape); in CheckLeafCount() local
210 TF_RET_CHECK(shape_leaves == tuple_elements_.size()) in CheckLeafCount()
211 << "Shape " << ShapeUtil::HumanString(shape) << " has " << shape_leaves in CheckLeafCount()