Searched defs:shuffle_size (Results 1 – 6 of 6) sorted by relevance
30 ShuffleNode::ShuffleNode(std::shared_ptr<DatasetNode> child, int32_t shuffle_size, bool reset_every… in ShuffleNode()
33 int64_t *shuffle_size) { in ComputeShuffleSize()65 int64_t shuffle_size = 0; in AddShuffleOp() local
41 ShuffleOp::ShuffleOp(int32_t shuffle_size, uint32_t shuffle_seed, int32_t op_connector_size, bool r… in ShuffleOp()
43 uint32_t shuffle_size = 32; in TEST_F() local
1350 int32_t shuffle_size = 10; in TEST_F() local
239 ….def(py::init([](std::shared_ptr<DatasetNode> self, int32_t shuffle_size, bool reset_every_epoch) { in __anonb78612491c02()