Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h190 iterator leaf_begin() { in leaf_begin() function
198 const_iterator leaf_begin() const { in leaf_begin() function
208 return tensorflow::gtl::make_range(leaf_begin(), leaf_end()); in leaves()
211 return tensorflow::gtl::make_range(leaf_begin(), leaf_end()); in leaves()
215 reverse_iterator leaf_rend() { return reverse_iterator(leaf_begin()); } in leaf_rend()
220 return const_reverse_iterator(leaf_begin()); in leaf_rend()
237 int64 leaf_count() const { return std::distance(leaf_begin(), leaf_end()); } in leaf_count()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_metadata.cc305 instruction->set_sharding(sharding_tree.leaf_begin()->second); in ApplyShardingFromUsers()