Searched refs:outfeed_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 244 if (absl::optional<Shape> outfeed_shape = GetXfeedShape( in ReplayComputation() local
252 auto consume_outfeed = [client, outfeed_shape] { in ReplayComputation()
254 client->TransferFromOutfeedLocal(*outfeed_shape, /*device_ordinal=*/0) in ReplayComputation()
257 << ShapeUtil::HumanStringWithLayout(*outfeed_shape); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 320 if (ShapeUtil::IsInitialized(options.outfeed_shape)) { in ExecuteReplicated()
346 if (ShapeUtil::IsInitialized(options.outfeed_shape)) { in ExecuteReplicated()
357 executor, options.outfeed_shape, &literal)); in ExecuteReplicated()
|
| D | hlo_runner.h | 71 Shape outfeed_shape; member
|
| D | hlo_instruction.cc | 368 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local
370 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape)); in CreateFromProto()
371 instruction = CreateOutfeed(outfeed_shape, operands(0), operands(1), in CreateFromProto()
870 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed() argument
873 outfeed_shape, operand, token_operand, outfeed_config); in CreateOutfeed()
3404 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
3405 return Cast<HloOutfeedInstruction>(this)->outfeed_shape(); in outfeed_shape()
|
| D | hlo_instruction.h | 544 const Shape& outfeed_shape, HloInstruction* operand,
1519 const Shape& outfeed_shape() const;
|
| D | hlo_instructions.cc | 1812 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction() argument
1817 outfeed_shape_(outfeed_shape), in HloOutfeedInstruction()
1826 *proto.mutable_outfeed_shape() = outfeed_shape().ToProto(); in ToProto()
1851 outfeed_shape(), new_operands[0], new_operands[1], outfeed_config()); in CloneWithNewOperandsImpl()
|
| D | hlo_verifier.cc | 262 if (!ShapesSame(outfeed->outfeed_shape(), outfeed->operand(0)->shape())) { in HandleOutfeed()
267 StringifyShape(outfeed->outfeed_shape()), outfeed->ToString()); in HandleOutfeed()
|
| D | hlo_instructions.h | 1000 explicit HloOutfeedInstruction(const Shape& outfeed_shape,
1005 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
|
| D | hlo.proto | 138 xla.ShapeProto outfeed_shape = 29; field
|
| D | hlo_instruction_test.cc | 729 EXPECT_TRUE(ShapeUtil::Equal(clone01->outfeed_shape(), shape01)); in TEST_F()
730 EXPECT_TRUE(ShapeUtil::Equal(clone10->outfeed_shape(), shape10)); in TEST_F()
|
| D | layout_assignment.cc | 454 instruction->outfeed_shape(), instruction, 0)); in AddMandatoryConstraints()
|