| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 92 auto outfeed_shape = ShapeUtil::MakeShape(F32, {4, 8}); in TEST_F() local
95 LiteralUtil::CreateFromDimensions(F32, outfeed_shape.dimensions())); in TEST_F()
97 auto outfeed = OutfeedWithToken(data, token, outfeed_shape, ""); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 276 if (absl::optional<Shape> outfeed_shape = GetXfeedShape( in ReplayComputation() local
284 auto consume_outfeed = [client, outfeed_shape] { in ReplayComputation()
285 Literal outfeed(*outfeed_shape); in ReplayComputation()
289 << ShapeUtil::HumanStringWithLayout(*outfeed_shape); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 253 if (ShapeUtil::IsInitialized(options.outfeed_shape)) { in ExecuteReplicatedImpl()
279 if (ShapeUtil::IsInitialized(options.outfeed_shape)) { in ExecuteReplicatedImpl()
288 Literal literal(options.outfeed_shape); in ExecuteReplicatedImpl()
|
| D | hlo_runner_interface.h | 68 Shape outfeed_shape; member
|
| D | hlo_instruction.cc | 405 Shape outfeed_shape(proto.outfeed_shape()); in CreateFromProto() local
407 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape)); in CreateFromProto()
408 instruction = CreateOutfeed(outfeed_shape, operands(0), operands(1), in CreateFromProto()
1116 const Shape& outfeed_shape, HloInstruction* operand, in CreateOutfeed() argument
1119 outfeed_shape, operand, token_operand, outfeed_config); in CreateOutfeed()
4111 const Shape& HloInstruction::outfeed_shape() const { in outfeed_shape() function in xla::HloInstruction
4112 return Cast<HloOutfeedInstruction>(this)->outfeed_shape(); in outfeed_shape()
|
| D | hlo_instruction.h | 743 const Shape& outfeed_shape, HloInstruction* operand,
1807 const Shape& outfeed_shape() const;
|
| D | hlo_instructions.cc | 2087 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction() argument
2092 outfeed_shape_(outfeed_shape), in HloOutfeedInstruction()
2101 *proto.mutable_outfeed_shape() = outfeed_shape().ToProto(); in ToProto()
2130 outfeed_shape(), new_operands[0], new_operands[1], outfeed_config()); in CloneWithNewOperandsImpl()
|
| D | hlo_instructions.h | 1192 explicit HloOutfeedInstruction(const Shape& outfeed_shape,
1197 const Shape& outfeed_shape() const { return outfeed_shape_; } in outfeed_shape() function
|
| D | hlo_verifier.cc | 437 if (!ShapesSame(outfeed->outfeed_shape(), outfeed->operand(0)->shape())) { in HandleOutfeed()
442 StringifyShape(outfeed->outfeed_shape()), outfeed->ToString()); in HandleOutfeed()
|
| D | hlo_parser.cc | 2107 optional<Shape> outfeed_shape; in ParseInstructionRhs() local
2110 &outfeed_shape}; in ParseInstructionRhs()
2118 outfeed_shape.has_value() ? *outfeed_shape : outfeed_input->shape(); in ParseInstructionRhs()
|
| D | hlo_instruction_test.cc | 731 EXPECT_TRUE(ShapeUtil::Equal(clone01->outfeed_shape(), shape01)); in TEST_F()
732 EXPECT_TRUE(ShapeUtil::Equal(clone10->outfeed_shape(), shape10)); in TEST_F()
|
| D | hlo_parser_test.cc | 1226 outfeed = token[] outfeed(infeed.data, token0), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
1230 outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
|
| D | hlo.proto | 142 xla.ShapeProto outfeed_shape = 29; field
|
| D | layout_assignment.cc | 470 instruction->outfeed_shape(), instruction, 0)); in AddMandatoryConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2912 Shape outfeed_shape = operand->shape(); in HandleOutfeed() local
2913 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes(hlo->outfeed_shape(), in HandleOutfeed()
2914 &outfeed_shape)); in HandleOutfeed()
2917 outfeed_shape, operand, token, hlo->outfeed_config())); in HandleOutfeed()
3027 hlo->outfeed_shape(), &per_branch_partitioned_shapes[i])); in HandleOutfeed()
|
| D | spmd_partitioner_test.cc | 728 EXPECT_TRUE(LayoutUtil::Equal(root->outfeed_shape().tuple_shapes(0).layout(), in TEST_F()
730 EXPECT_TRUE(LayoutUtil::Equal(root->outfeed_shape().tuple_shapes(1).layout(), in TEST_F()
793 first_outfeed_instr->outfeed_shape().tuple_shapes(0).layout(), in TEST_F()
796 first_outfeed_instr->outfeed_shape().tuple_shapes(1).layout(), in TEST_F()
799 second_outfeed_instr->outfeed_shape().tuple_shapes(0).layout(), in TEST_F()
802 second_outfeed_instr->outfeed_shape().tuple_shapes(1).layout(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1816 outfeed_shape = xla_client.shape_from_pyval(
1818 ops.OutfeedWithToken(x, token, outfeed_shape)
1827 got = device.transfer_from_outfeed(outfeed_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 586 // CHECK: ROOT %[[RESULT:.*]] = token[] outfeed(s32[3] [[DATA]], token[] [[TOKEN]]), outfeed_shape…
|