Searched refs:CreateOne (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 50 HloInstruction* CreateOne(const Shape& shape, SpmdBuilder* b);
|
| D | dot_handler.cc | 692 CreateOne(cw_data_partition_id->shape(), &body_b))); in PartitionBaseCase()
1191 i->shape(), HloOpcode::kAdd, i, CreateOne(i->shape(), &body_b))); in PartitionBaseCase()
1226 i->shape(), HloOpcode::kAdd, i, CreateOne(i->shape(), &body_b))); in PartitionBaseCase()
1331 CreateOne(real_i->shape(), &body_b))); in PartitionBaseCase()
|
| D | gather_scatter_handler.cc | 548 identity = CreateOne(operand.hlo()->shape(), &b_); in HandleScatter()
|
| D | spmd_partitioner_util.cc | 92 HloInstruction* CreateOne(const Shape& shape, SpmdBuilder* b) { in CreateOne() function
97 CreateOne(ShapeUtil::GetTupleElementShape(shape, i), b)); in CreateOne()
|