Searched refs:leaf_replication (Results 1 – 1 of 1) sorted by relevance
514 std::vector<bool>* leaf_replication);1445 std::vector<bool>* leaf_replication) { in SetEntryTupleShapesAndLeafReplication() argument1447 leaf_replication->reserve(block->getNumArguments()); in SetEntryTupleShapesAndLeafReplication()1472 leaf_replication->push_back( in SetEntryTupleShapesAndLeafReplication()1520 std::vector<bool> leaf_replication; in LowerBasicBlockAsFunction() local1523 &leaf_replication))) in LowerBasicBlockAsFunction()1532 xla::Parameter(builder, 0, input_shape, "arg_tuple", leaf_replication); in LowerBasicBlockAsFunction()