Searched refs:XlaBuilderFriend (Results 1 – 3 of 3) sorted by relevance
1077 xla::XlaOp fusion = xla::internal::XlaBuilderFriend::BuildFusion( in ExportXlaOp()1095 value_map[op] = xla::internal::XlaBuilderFriend::BuildBitcast( in ExportXlaOp()1220 auto* shape = xla::internal::XlaBuilderFriend::GetInstruction(xla_op) in Lower()1235 xla::internal::XlaBuilderFriend::GetInstruction(xla_op) in Lower()
54 struct XlaBuilderFriend { struct118 friend struct internal::XlaBuilderFriend;1475 friend struct internal::XlaBuilderFriend;
154 XlaOp XlaBuilderFriend::BuildFusion(XlaBuilder* builder, in BuildFusion()171 XlaOp XlaBuilderFriend::BuildBitcast(XlaBuilder* builder, XlaOp operand, in BuildBitcast()181 HloInstructionProto* XlaBuilderFriend::GetInstruction(XlaOp op) { in GetInstruction()