Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc742 value_map[op] = xla::internal::XlaBuilderFriend::BuildAddDependency( in ExportXlaOp()
880 valueMap[op] = xla::internal::XlaBuilderFriend::BuildDomain( in ExportXlaOp()
1244 xla::internal::XlaBuilderFriend::BuildPartitionId(ctx.builder, shape); in ExportXlaOp()
1408 xla::internal::XlaBuilderFriend::BuildRngGetAndUpdateState( in ExportXlaOp()
1702 xla::XlaOp fusion = xla::internal::XlaBuilderFriend::BuildFusion( in ExportXlaOp()
1720 xla::XlaOp bitcast = xla::internal::XlaBuilderFriend::BuildBitcast( in ExportXlaOp()
1729 xla::internal::XlaBuilderFriend::GetInstruction(bitcast); in ExportXlaOp()
1731 xla::internal::XlaBuilderFriend::GetInstruction(operand); in ExportXlaOp()
2013 auto* shape = xla::internal::XlaBuilderFriend::GetInstruction(xla_op) in Lower()
2065 xla::internal::XlaBuilderFriend::GetInstruction(xla_gte_op); in Lower()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc102 XlaOp XlaBuilderFriend::BuildAddDependency(XlaBuilder* builder, XlaOp operand, in BuildAddDependency()
112 XlaOp XlaBuilderFriend::BuildFusion(XlaBuilder* builder, in BuildFusion()
129 XlaOp XlaBuilderFriend::BuildBitcast(XlaBuilder* builder, XlaOp operand, in BuildBitcast()
139 XlaOp XlaBuilderFriend::BuildDomain(XlaBuilder* builder, XlaOp operand, in BuildDomain()
152 XlaOp XlaBuilderFriend::BuildPartitionId(XlaBuilder* builder, in BuildPartitionId()
161 XlaOp XlaBuilderFriend::BuildRngGetAndUpdateState(XlaBuilder* builder, in BuildRngGetAndUpdateState()
174 HloInstructionProto* XlaBuilderFriend::GetInstruction(XlaOp op) { in GetInstruction()
179 HloInstructionProto* XlaBuilderFriend::GetInstructionByHandle( in GetInstructionByHandle()
Dxla_builder.h59 struct XlaBuilderFriend { struct
138 friend struct internal::XlaBuilderFriend;
1566 friend struct internal::XlaBuilderFriend;
Dxla_builder_test.cc1425 internal::XlaBuilderFriend builder_friend; in TEST_F()