Home
last modified time | relevance | path

Searched refs:shape_with_layout (Results 1 – 24 of 24) 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.h187 const Shape* shape_with_layout = nullptr);
191 const Shape* shape_with_layout = nullptr);
198 const Shape* shape_with_layout = nullptr);
204 const Shape* shape_with_layout = nullptr);
208 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/python/
Doutfeed_receiver.cc421 Shape shape_with_layout = builder->GetShape(data).ValueOrDie(); in AddOutfeedToBuilder() local
423 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in AddOutfeedToBuilder()
429 << "; shape=" << shape_with_layout.ToString(); in AddOutfeedToBuilder()
434 if (!ShapeUtil::Equal(shape_with_layout, found->second)) { in AddOutfeedToBuilder()
438 shape_with_layout.DebugString(), found->second.DebugString(), in AddOutfeedToBuilder()
442 shape_registry_.insert({consumer_id, shape_with_layout}); in AddOutfeedToBuilder()
451 token = OutfeedWithToken(data, token, shape_with_layout, ""); in AddOutfeedToBuilder()
Dxla.cc141 Shape shape_with_layout = shape; in PYBIND11_MODULE() local
143 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in PYBIND11_MODULE()
148 literal = std::make_shared<Literal>(shape_with_layout); in PYBIND11_MODULE()
Dops.cc135 absl::Span<const XlaOp> operands, const Shape& shape_with_layout, in BuildOpsSubmodule()
139 builder, call_target_name, operands, shape_with_layout, in BuildOpsSubmodule()
Dxla_client_test.py335 shape_with_layout=xla_client.Shape.array_shape(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc366 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
367 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
376 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
384 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
392 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
394 shape_with_layout, instruction, /*operand_no=*/4)); in PropagateBufferConstraint()
/external/tensorflow/tensorflow/compiler/xla/client/
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()
Dexecutable_build_options.cc52 const Shape& shape_with_layout) { in set_result_layout() argument
54 result_layout_ = shape_with_layout; in set_result_layout()
Dclient.h135 const Shape* shape_with_layout = nullptr);
161 const Shape* shape_with_layout, int64 replica_id = 0,
Dxla_builder.cc1768 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed() argument
1776 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1780 if (!ShapeUtil::Compatible(*operand_shape, shape_with_layout)) { in Outfeed()
1783 ShapeUtil::HumanStringWithLayout(shape_with_layout), in Outfeed()
1786 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in Outfeed()
1821 const Shape& shape_with_layout, in OutfeedWithToken() argument
1825 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1829 if (!ShapeUtil::Compatible(*operand_shape, shape_with_layout)) { in OutfeedWithToken()
1832 ShapeUtil::HumanStringWithLayout(shape_with_layout), in OutfeedWithToken()
1835 return OutfeedWithTokenInternal(operand, token, shape_with_layout, in OutfeedWithToken()
[all …]
Dxla_builder.h641 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
644 const Shape& shape_with_layout,
647 XlaOp operand, XlaOp token, const Shape& shape_with_layout,
654 const Shape& shape_with_layout, const string& opaque,
666 const Shape& shape_with_layout, const string& opaque,
675 const XlaComputation& computation, const Shape& shape_with_layout,
744 const absl::optional<Shape>& shape_with_layout = absl::nullopt);
870 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
1211 friend void Outfeed(XlaOp operand, const Shape& shape_with_layout,
1231 absl::Span<const XlaOp> operands, const Shape& shape_with_layout,
[all …]
Dexecutable_build_options.h54 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc208 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() argument
214 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetOperandLayout()
220 shape_with_layout, /*minor_to_major_only=*/true)) { in SetOperandLayout()
230 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetOperandLayout()
248 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
253 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
272 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout() argument
275 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetResultLayout()
280 shape_with_layout, /*minor_to_major_only=*/true)) { in SetResultLayout()
285 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetResultLayout()
[all …]
Dlayout_assignment.h172 Status SetOperandLayout(const Shape& shape_with_layout,
175 Status SetResultLayout(const Shape& shape_with_layout, bool dfs = true);
187 Status SetInstructionLayout(const Shape& shape_with_layout,
472 const Shape& shape_with_layout, HloInstruction* instruction);
Dhlo_cost_analysis_test.cc723 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
725 Shape shape_without_layout = shape_with_layout; in TEST_F()
737 shape_with_layout, HloOpcode::kAdd, c1, broadcast)); in TEST_F()
Dservice.cc923 return_shape = Shape(arg->shape_with_layout()); in TransferToClient()
1036 auto literal = Literal::CreateFromShape(Shape(arg->shape_with_layout())); 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()
Dxla.proto399 ShapeProto shape_with_layout = 2; field
426 ShapeProto shape_with_layout = 1; field
Dliteral.cc720 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout()
721 CHECK(ShapeUtil::Compatible(shape_with_layout, shape())) in Relayout()
722 << "Given shape_with_layout " << ShapeUtil::HumanString(shape_with_layout) in Relayout()
725 Literal result = CreateFromShape(shape_with_layout); in Relayout()
Dliteral.h294 Literal Relayout(const Shape& shape_with_layout) const;
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h231 XlaOp operand, XlaOp token, const Shape& shape_with_layout,
Dmlir_hlo_builder.cc480 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal() argument