Searched defs:all_gather_dimension (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_instructions.h | 398 int64 all_gather_dimension() const { return all_gather_dimension_; } in all_gather_dimension() function
|
D | hlo_instruction.cc | 422 int64 all_gather_dimension = proto.dimensions(0); in CreateFromProto() local 1044 const Shape& shape, HloInstruction* operand, int64 all_gather_dimension, in CreateAllGather()
|
D | shape_inference.cc | 2035 const Shape& operand_shape, int64 all_gather_dimension, int64 shard_count) { in InferAllGatherShape()
|
D | hlo_instructions.cc | 619 const Shape& shape, HloInstruction* operand, int64 all_gather_dimension, in HloAllGatherInstruction()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner_util.cc | 1659 int64 channel_id, int64 all_gather_dimension) { in GetPerGroupCollectiveOpsCreator()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 2714 XlaOp XlaBuilder::AllGather(XlaOp operand, int64 all_gather_dimension, in AllGather() 4653 XlaOp AllGather(const XlaOp operand, int64 all_gather_dimension, in AllGather()
|