Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.cc52 const int64 leaf_count = output_buffers->leaf_count(); in ExecuteOnStream() local
53 TF_RET_CHECK(source_slices_.size() == leaf_count) in ExecuteOnStream()
55 << ") and outputs (" << leaf_count << ")"; in ExecuteOnStream()
58 for (int64 index = 0; index < leaf_count; ++index) { in ExecuteOnStream()
/external/libvpx/libvpx/vpx_dsp/
Dprob.h33 #define TREE_SIZE(leaf_count) (2 * (leaf_count)-2) argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc110 flattened_list.reserve(sub_shardings.leaf_count()); in Tuple()
142 int64 leaf_count = RequiredLeaves(tuple_shape); in SingleTuple() local
144 flattened_list.resize(leaf_count, sharding); in SingleTuple()
312 const int64 leaf_count = ShapeUtil::GetLeafCount(shape); in RequiredLeaves() local
313 return (leaf_count == 0) ? 1 : leaf_count; in RequiredLeaves()
Dhlo_sharding_metadata.cc284 TF_RET_CHECK(user_sharding_tree.leaf_count() == 1) in ApplyShardingFromUsers()
304 TF_RET_CHECK(sharding_tree.leaf_count() == 1); in ApplyShardingFromUsers()
Dhlo_parser.cc2619 int leaf_count = ShapeUtil::GetLeafCount(instruction->shape()); in ParseInstructionRhs() local
2622 if (leaf_count != replicated.size()) { in ParseInstructionRhs()
2624 StrCat("parameter has ", leaf_count, in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/python/
Dpytree.cc195 int leaf_count = 0; in Unflatten() local
205 leaf_count)); in Unflatten()
209 ++leaf_count; in Unflatten()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc122 EXPECT_EQ(0, shape_tree.leaf_count()); in TEST_F()
Dshape_tree.h238 int64 leaf_count() const { return std::distance(leaf_begin(), leaf_end()); } in leaf_count() function
/external/e2fsprogs/lib/blkid/
Dprobe.h557 __u32 leaf_count; member
Dprobe.c1299 leaf_node_count = blkid_be32(bnode->leaf_count); in probe_hfsplus()