| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 93 EXPECT_TRUE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
94 ShapeUtil::MakeTokenShape())); in TEST()
95 EXPECT_FALSE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
98 ShapeUtil::MakeTokenShape())); in TEST()
100 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
101 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
105 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
106 ShapeUtil::MakeTokenShape())); in TEST()
107 EXPECT_FALSE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
110 ShapeUtil::MakeTokenShape())); in TEST()
[all …]
|
| D | shape_test.cc | 38 const Shape token_ = ShapeUtil::MakeTokenShape();
147 *program_shape.add_parameters() = ShapeUtil::MakeTokenShape(); in TEST_F()
151 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST_F()
|
| D | layout_util_test.cc | 156 Shape src = ShapeUtil::MakeTokenShape(); in TEST_F()
157 Shape dst = ShapeUtil::MakeTokenShape(); in TEST_F()
180 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), ShapeUtil::MakeTokenShape(), in TEST_F()
186 MakeShapeWithLayout(F32, {42, 123}, {1, 0}), ShapeUtil::MakeTokenShape(), in TEST_F()
217 {ShapeUtil::MakeOpaqueShape(), ShapeUtil::MakeTokenShape()}) { in TEST_F()
|
| D | literal_util.cc | 122 return Literal(ShapeUtil::MakeTokenShape()); in CreateToken()
|
| D | shape_util.h | 351 static Shape MakeTokenShape();
|
| D | shape_util.cc | 282 /* static */ Shape ShapeUtil::MakeTokenShape() { in MakeTokenShape() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | token_hlo_test.cc | 76 HloInstruction::CreateParameter(1, ShapeUtil::MakeTokenShape(), "p1")); in XLA_TEST_F()
97 {ShapeUtil::MakeShape(F32, {1, 2, 3}), ShapeUtil::MakeTokenShape()}), in XLA_TEST_F()
|
| D | transfer_manager_test.cc | 305 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 | 310 if (!ShapeUtil::Equal(token->shape(), ShapeUtil::MakeTokenShape())) { in CheckIsTokenOperand()
340 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()})); in HandleInfeed()
356 return CheckShape(outfeed, ShapeUtil::MakeTokenShape()); in HandleOutfeed()
877 ShapeUtil::MakeTokenShape()}), in HandleSend()
882 return CheckShape(send_done, ShapeUtil::MakeTokenShape()); in HandleSendDone()
890 ShapeUtil::MakeShape(U32, {}), ShapeUtil::MakeTokenShape()}), in HandleRecv()
899 ShapeUtil::MakeTokenShape()})); in HandleRecvDone()
1015 return CheckShape(token, ShapeUtil::MakeTokenShape()); in HandleAfterAll()
|
| D | hlo_dce_test.cc | 335 ShapeUtil::MakeTokenShape(), {param}, nested_called_computation)); in TEST_F()
345 ShapeUtil::MakeTokenShape(), {param}, called_computation)); in TEST_F()
|
| 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 | 440 ShapeUtil::MakeTokenShape()}), in HloSendInstruction()
456 : HloSendRecvInstruction(HloOpcode::kSendDone, ShapeUtil::MakeTokenShape(), in HloSendDoneInstruction()
478 ShapeUtil::MakeTokenShape()}), in HloRecvInstruction()
498 ShapeUtil::MakeTokenShape()}), in HloRecvDoneInstruction()
1900 {infeed_shape, ShapeUtil::MakeTokenShape()})), in HloInfeedInstruction()
1939 : HloInstruction(HloOpcode::kOutfeed, ShapeUtil::MakeTokenShape()), in HloOutfeedInstruction()
|
| D | hlo_instruction.cc | 1005 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateAfterAll()
1014 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateToken()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1347 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in Infeed()
1368 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Infeed()
1421 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in InfeedWithToken()
1446 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1466 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1499 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in OutfeedWithToken()
1524 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in CreateToken()
1544 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in AfterAll()
2347 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Send()
2368 ShapeUtil::MakeTokenShape()}) in SendWithToken()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 170 return ShapeUtil::MakeTokenShape(); in TypeToShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math_test.cc | 198 shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 472 xla::ShapeUtil::MakeTokenShape()}); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 381 .def_static("token_shape", []() { return ShapeUtil::MakeTokenShape(); }) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 895 *xla_shape = xla::ShapeUtil::MakeTokenShape(); in XLAShapeForArgument()
|