Home
last modified time | relevance | path

Searched refs:SingleTuple (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.h99 static HloSharding SingleTuple(const Shape& tuple_shape,
Dhlo_sharding.cc138 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple() function in xla::HloSharding
150 return shape.IsTuple() ? SingleTuple(shape, sharding) : sharding; in Single()
Dsharding_propagation.cc122 HloSharding::SingleTuple(instruction->shape(), HloSharding::Replicate())); in SetDefaultTupleSharding()
901 HloSharding::SingleTuple(instruction->shape(), operand->sharding()), in InferShardingFromOperands()
1280 : HloSharding::SingleTuple(instruction.shape(), in GetShardingFromUser()
Dhlo_sharding_test.cc155 HloSharding sharding = HloSharding::SingleTuple(ShapeUtil::MakeTupleShape({}), in TEST_F()