Home
last modified time | relevance | path

Searched refs:shape_with_layout (Results 1 – 22 of 22) 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.cc261 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
266 options.set_result_layout(shape_with_layout); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc333 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
334 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
343 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
351 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
359 shape_with_layout, instruction, /*operand_no=*/0)); in PropagateBufferConstraint()
361 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,
Dexecutable_build_options.h46 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
Dxla_builder.h426 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
429 const Shape& shape_with_layout,
437 const Shape& shape_with_layout, const string& opaque,
556 const Shape& shape_with_layout, const ChannelHandle& handle);
820 friend void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
829 absl::Span<const XlaOp> operands, const Shape& shape_with_layout,
994 const Shape& shape_with_layout,
1001 const Shape& shape_with_layout,
1409 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout,
1416 const Shape& shape_with_layout,
[all …]
Dxla_builder.cc1385 void XlaBuilder::Outfeed(const XlaOp& operand, const Shape& shape_with_layout, in Outfeed() argument
1393 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1397 if (!ShapeUtil::Compatible(operand_shape, shape_with_layout)) { in Outfeed()
1400 ShapeUtil::HumanStringWithLayout(shape_with_layout), in Outfeed()
1403 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in Outfeed()
1438 const Shape& shape_with_layout, in OutfeedWithToken() argument
1446 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1450 if (!ShapeUtil::Compatible(operand_shape, shape_with_layout)) { in OutfeedWithToken()
1453 ShapeUtil::HumanStringWithLayout(shape_with_layout), in OutfeedWithToken()
1456 *instr.mutable_outfeed_shape() = shape_with_layout.ToProto(); in OutfeedWithToken()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc198 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() argument
204 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetOperandLayout()
210 shape_with_layout)) { in SetOperandLayout()
220 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetOperandLayout()
238 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
243 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
262 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout() argument
265 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetResultLayout()
269 if (!curr_shape_layout->MatchesLayoutInShape(shape_with_layout)) { in SetResultLayout()
274 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetResultLayout()
[all …]
Dlayout_assignment.h170 Status SetOperandLayout(const Shape& shape_with_layout,
173 Status SetResultLayout(const Shape& shape_with_layout, bool dfs = true);
185 Status SetInstructionLayout(const Shape& shape_with_layout,
464 const Shape& shape_with_layout, HloInstruction* instruction);
Dhlo_cost_analysis_test.cc498 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
500 Shape shape_without_layout = shape_with_layout; in TEST_F()
512 shape_with_layout, HloOpcode::kAdd, c1, broadcast)); in TEST_F()
Dservice.cc946 return_shape = Shape(arg->shape_with_layout()); in TransferToClient()
1059 auto literal = Literal::CreateFromShape(Shape(arg->shape_with_layout())); in TransferFromOutfeed()
1063 executor, Shape(arg->shape_with_layout()), literal)); in TransferFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc308 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
309 *shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major); in ReshapeSlice()
316 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i); in ReshapeSlice()
Dxla.proto321 ShapeProto shape_with_layout = 2; field
348 ShapeProto shape_with_layout = 1; field
Dliteral.cc590 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout()
591 CHECK(ShapeUtil::Compatible(shape_with_layout, shape())) in Relayout()
592 << "Given shape_with_layout " << ShapeUtil::HumanString(shape_with_layout) in Relayout()
595 Literal result = CreateFromShape(shape_with_layout); in Relayout()
Dliteral.h271 Literal Relayout(const Shape& shape_with_layout) const;
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc99 const Literal& argument, const absl::optional<Shape>& shape_with_layout, in FromLiteral() argument
109 if (shape_with_layout) { in FromLiteral()
110 Literal relaid = argument.Relayout(shape_with_layout.value()); in FromLiteral()
620 const Shape& shape_with_layout, in CustomCall() argument
629 shape_with_layout, in CustomCall()
Dlocal_computation_builder.h78 const Literal& argument, const absl::optional<Shape>& shape_with_layout,
313 const Shape& shape_with_layout,
Dxla_client.py1421 shape_with_layout, argument
1440 shape_with_layout,
Dxla_client_test.py322 shape_with_layout=xla_client.Shape.array_shape(np.float32, (), ()),