Home
last modified time | relevance | path

Searched refs:shape_with_layout (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc172 absl::Span<GlobalData* const> arguments, const Shape* shape_with_layout) { in ComputeAndCompareLiteral() argument
174 shape_with_layout)); in ComputeAndCompareLiteral()
180 const Shape* shape_with_layout) { in ComputeAndCompareLiteral() argument
182 error, shape_with_layout)); in ComputeAndCompareLiteral()
275 const Shape* shape_with_layout) { in ComputeAndTransfer() argument
293 return ExecuteAndTransfer(computation, arguments, shape_with_layout); in ComputeAndTransfer()
299 const Shape* shape_with_layout) { in ComputeAndCompareLiteralWithStatus() argument
329 if (shape_with_layout != nullptr) { in ComputeAndCompareLiteralWithStatus()
330 layout_shape = *shape_with_layout; in ComputeAndCompareLiteralWithStatus()
337 shape_with_layout = &layout_shape; in ComputeAndCompareLiteralWithStatus()
[all …]
Dclient_library_test_base.h185 const Shape* shape_with_layout = nullptr);
189 const Shape* shape_with_layout = nullptr);
196 const Shape* shape_with_layout = nullptr);
202 const Shape* shape_with_layout = nullptr);
206 const Shape* shape_with_layout = nullptr);
Dlocal_client_execute_test.cc264 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
269 options.set_result_layout(shape_with_layout); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc360 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
361 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
370 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
378 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
386 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
388 shape_with_layout, instruction, /*operand_no=*/4)); in PropagateBufferConstraint()
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc51 const Shape& shape_with_layout) { in set_result_layout() argument
53 result_layout_ = shape_with_layout; in set_result_layout()
Dclient.cc42 const Shape* shape_with_layout) { in Transfer() argument
45 if (shape_with_layout != nullptr) { in Transfer()
46 *request.mutable_shape_with_layout() = shape_with_layout->ToProto(); in Transfer()
119 const Shape* shape_with_layout, int64 replica_id, in TransferFromOutfeed() argument
126 if (shape_with_layout != nullptr) { in TransferFromOutfeed()
127 *request.mutable_shape_with_layout() = shape_with_layout->ToProto(); in TransferFromOutfeed()
Dclient.h135 const Shape* shape_with_layout = nullptr);
161 const Shape* shape_with_layout, int64 replica_id = 0,
Dxla_builder.cc1441 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed() argument
1449 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1453 if (!ShapeUtil::Compatible(*operand_shape, shape_with_layout)) { in Outfeed()
1456 ShapeUtil::HumanStringWithLayout(shape_with_layout), in Outfeed()
1459 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in Outfeed()
1494 const Shape& shape_with_layout, in OutfeedWithToken() argument
1502 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1506 if (!ShapeUtil::Compatible(*operand_shape, shape_with_layout)) { in OutfeedWithToken()
1509 ShapeUtil::HumanStringWithLayout(shape_with_layout), in OutfeedWithToken()
1512 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in OutfeedWithToken()
[all …]
Dxla_builder.h472 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
475 const Shape& shape_with_layout,
483 const Shape& shape_with_layout, const string& opaque,
518 const absl::optional<Shape>& shape_with_layout = absl::nullopt);
595 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
859 friend void Outfeed(XlaOp operand, const Shape& shape_with_layout,
868 absl::Span<const XlaOp> operands, const Shape& shape_with_layout,
927 const absl::optional<Shape>& shape_with_layout);
1028 const Shape& shape_with_layout,
1035 const Shape& shape_with_layout,
[all …]
Dexecutable_build_options.h46 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc207 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() argument
213 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetOperandLayout()
219 shape_with_layout, /*minor_to_major_only=*/true)) { in SetOperandLayout()
229 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetOperandLayout()
247 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
252 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
271 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout() argument
274 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetResultLayout()
279 shape_with_layout, /*minor_to_major_only=*/true)) { in SetResultLayout()
284 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetResultLayout()
[all …]
Dlayout_assignment.h171 Status SetOperandLayout(const Shape& shape_with_layout,
174 Status SetResultLayout(const Shape& shape_with_layout, bool dfs = true);
186 Status SetInstructionLayout(const Shape& shape_with_layout,
468 const Shape& shape_with_layout, HloInstruction* instruction);
Dhlo_cost_analysis_test.cc688 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
690 Shape shape_without_layout = shape_with_layout; in TEST_F()
702 shape_with_layout, HloOpcode::kAdd, c1, broadcast)); in TEST_F()
Dservice.cc967 return_shape = Shape(arg->shape_with_layout()); in TransferToClient()
1080 auto literal = Literal::CreateFromShape(Shape(arg->shape_with_layout())); in TransferFromOutfeed()
1084 executor, Shape(arg->shape_with_layout()), &literal)); in TransferFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc360 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
361 *shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major); in ReshapeSlice()
368 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i); in ReshapeSlice()
Dliteral.cc584 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout()
585 CHECK(ShapeUtil::Compatible(shape_with_layout, shape())) in Relayout()
586 << "Given shape_with_layout " << ShapeUtil::HumanString(shape_with_layout) in Relayout()
589 Literal result = CreateFromShape(shape_with_layout); in Relayout()
Dxla.proto328 ShapeProto shape_with_layout = 2; field
355 ShapeProto shape_with_layout = 1; field
Dliteral.h278 Literal Relayout(const Shape& shape_with_layout) const;
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1198 shape_with_layout, argument
1217 list(operands), shape_with_layout,
Dxla_client_test.py345 shape_with_layout=xla_client.Shape.array_shape(