| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 95 EXPECT_TRUE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
96 ShapeUtil::MakeTokenShape())); in TEST()
97 EXPECT_FALSE(ShapeUtil::Compatible(ShapeUtil::MakeTokenShape(), in TEST()
100 ShapeUtil::MakeTokenShape())); in TEST()
102 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}), in TEST()
103 ShapeUtil::MakeTupleShape({ShapeUtil::MakeTokenShape()}))); in TEST()
107 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
108 ShapeUtil::MakeTokenShape())); in TEST()
109 EXPECT_FALSE(ShapeUtil::Equal(ShapeUtil::MakeTokenShape(), in TEST()
112 ShapeUtil::MakeTokenShape())); in TEST()
[all …]
|
| D | shape_test.cc | 39 const Shape token_ = ShapeUtil::MakeTokenShape();
148 *program_shape.add_parameters() = ShapeUtil::MakeTokenShape(); in TEST_F()
152 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 | 397 static Shape MakeTokenShape();
|
| D | shape_util.cc | 383 /* 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 | 429 auto token_shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 398 if (!ShapeUtil::Equal(token->shape(), ShapeUtil::MakeTokenShape())) { in CheckIsTokenOperand()
428 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()})); in HandleInfeed()
444 return CheckShape(outfeed, ShapeUtil::MakeTokenShape()); in HandleOutfeed()
1012 ShapeUtil::MakeTokenShape()}), in HandleSend()
1017 return CheckShape(send_done, ShapeUtil::MakeTokenShape()); in HandleSendDone()
1025 ShapeUtil::MakeShape(U32, {}), ShapeUtil::MakeTokenShape()}), in HandleRecv()
1034 ShapeUtil::MakeTokenShape()})); in HandleRecvDone()
1153 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 | 498 ShapeUtil::MakeTokenShape()}), in HloSendInstruction()
514 : HloSendRecvInstruction(HloOpcode::kSendDone, ShapeUtil::MakeTokenShape(), in HloSendDoneInstruction()
536 ShapeUtil::MakeTokenShape()}), in HloRecvInstruction()
556 ShapeUtil::MakeTokenShape()}), in HloRecvDoneInstruction()
2052 {infeed_shape, ShapeUtil::MakeTokenShape()})), in HloInfeedInstruction()
2091 : HloInstruction(HloOpcode::kOutfeed, ShapeUtil::MakeTokenShape()), in HloOutfeedInstruction()
|
| D | hlo_instruction.cc | 1164 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateAfterAll()
1173 new HloInstruction(HloOpcode::kAfterAll, ShapeUtil::MakeTokenShape())); in CreateToken()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 185 return ShapeUtil::MakeTokenShape(); in TypeToShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1670 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in Infeed()
1691 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Infeed()
1743 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeTokenShape()}); in InfeedWithToken()
1773 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1793 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()
1844 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in OutfeedWithTokenInternal()
1854 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in CreateToken()
1874 *instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in AfterAll()
3149 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Send()
3170 ShapeUtil::MakeTokenShape()}) in SendWithToken()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 202 .def_static("token_shape", []() { return ShapeUtil::MakeTokenShape(); }) in BuildXlaCompilerSubmodule()
|
| /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/tf2xla/ |
| D | xla_compiler.cc | 927 *xla_shape = xla::ShapeUtil::MakeTokenShape(); in XLAShapeForArgument()
|