| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 92 EXPECT_TRUE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
93 ShapeUtil::MakeTokenShape())); in TEST()
94 EXPECT_FALSE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
97 ShapeUtil::MakeTokenShape())); in TEST()
99 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
100 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
104 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
105 ShapeUtil::MakeTokenShape())); in TEST()
106 EXPECT_FALSE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
109 ShapeUtil::MakeTokenShape())); in TEST()
[all …]
|
| D | shape_test.cc | 36 const Shape token_ = ShapeUtil::MakeTokenShape();
145 *program_shape.add_parameters() = ShapeUtil::MakeTokenShape(); in TEST_F()
149 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST_F()
|
| D | layout_util_test.cc | 222 Shape src = ShapeUtil::MakeTokenShape(); in TEST_F()
223 Shape dst = ShapeUtil::MakeTokenShape(); in TEST_F()
246 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), ShapeUtil::MakeTokenShape(), in TEST_F()
252 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), ShapeUtil::MakeTokenShape(), in TEST_F()
283 {ShapeUtil::MakeOpaqueShape(), ShapeUtil::MakeTokenShape()}) { in TEST_F()
|
| D | literal_util.cc | 102 return Literal(ShapeUtil::MakeTokenShape()); in CreateToken()
|
| D | shape_util.h | 349 static Shape MakeTokenShape();
|
| D | shape_util.cc | 255 /* static */ Shape ShapeUtil::MakeTokenShape() { in MakeTokenShape() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | token_hlo_test.cc | 77 HloInstruction::CreateParameter(1, ShapeUtil::MakeTokenShape(), "p1")); in XLA_TEST_F()
98 {ShapeUtil::MakeShape(F32, {1, 2, 3}), ShapeUtil::MakeTokenShape()}), in XLA_TEST_F()
|
| D | transfer_manager_test.cc | 263 auto device_buffer = AllocateDeviceBuffer(ShapeUtil::MakeTokenShape()); in XLA_TEST_F()
|
| D | concat_test.cc | 426 auto token_shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 223 if (!ShapeUtil::Equal(token->shape(), ShapeUtil::MakeTokenShape())) { in CheckIsTokenOperand()
253 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()})); in HandleInfeed()
269 return CheckShape(outfeed, ShapeUtil::MakeTokenShape()); in HandleOutfeed()
699 ShapeUtil::MakeTokenShape()})); in HandleSend()
703 return CheckShape(send_done, ShapeUtil::MakeTokenShape()); in HandleSendDone()
710 ShapeUtil::MakeShape(U32, {}), ShapeUtil::MakeTokenShape()})); in HandleRecv()
718 ShapeUtil::MakeTokenShape()})); in HandleRecvDone()
832 return CheckShape(token, ShapeUtil::MakeTokenShape()); in HandleAfterAll()
|
| D | while_loop_invariant_code_motion_test.cc | 254 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F()
303 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F()
|
| D | hlo_instructions.cc | 404 ShapeUtil::MakeTokenShape()}), in HloSendInstruction()
420 : HloSendRecvInstruction(HloOpcode::kSendDone, ShapeUtil::MakeTokenShape(), in HloSendDoneInstruction()
442 ShapeUtil::MakeTokenShape()}), in HloRecvInstruction()
462 ShapeUtil::MakeTokenShape()}), in HloRecvDoneInstruction()
1777 {infeed_shape, ShapeUtil::MakeTokenShape()})), in HloInfeedInstruction()
1816 : HloInstruction(HloOpcode::kOutfeed, ShapeUtil::MakeTokenShape()), in HloOutfeedInstruction()
|
| D | hlo_instruction.cc | 918 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateAfterAll()
927 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateToken()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1288 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in Infeed()
1310 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Infeed()
1364 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in InfeedWithToken()
1390 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1410 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1443 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in OutfeedWithToken()
1468 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in CreateToken()
1488 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in AfterAll()
2317 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Send()
2338 {shape, ShapeUtil::MakeShape(U32, {}), ShapeUtil::MakeTokenShape()}) in SendWithToken()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math_test.cc | 186 shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 452 xla::ShapeUtil::MakeTokenShape()}); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 692 *xla_shape = xla::ShapeUtil::MakeTokenShape(); in XLAShapeForArgument()
|