Home
last modified time | relevance | path

Searched defs:shape_with_layout (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc175 absl::Span<const XlaOp> operands, const Shape& shape_with_layout, in BuildOpsSubmodule()
181 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
Dxla.cc145 Shape shape_with_layout = shape; in PYBIND11_MODULE() local
Doutfeed_receiver.cc429 Shape shape_with_layout = builder->GetShape(data).ValueOrDie(); in AddOutfeedToBuilder() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc172 absl::Span<GlobalData* const> arguments, const Shape* shape_with_layout) { in ComputeAndCompareLiteral()
180 const Shape* shape_with_layout) { in ComputeAndCompareLiteral()
275 const Shape* shape_with_layout) { in ComputeAndTransfer()
299 const Shape* shape_with_layout) { in ComputeAndCompareLiteralWithStatus()
360 const Shape* shape_with_layout) { in ComputeAndCompareLiteralWithStatus()
Dlocal_client_execute_test.cc264 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc52 const Shape& shape_with_layout) { in set_result_layout()
Dclient.cc42 const Shape* shape_with_layout) { in Transfer()
119 const Shape* shape_with_layout, int64_t replica_id, in TransferFromOutfeed()
Dxla_builder.cc1780 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed()
1833 const Shape& shape_with_layout, in OutfeedWithToken()
1853 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()
2813 const absl::optional<Shape>& shape_with_layout) { in AllReduce()
3328 const Shape& shape_with_layout, in SendToHost()
4265 void Outfeed(const XlaOp operand, const Shape& shape_with_layout, in Outfeed()
4523 const absl::optional<Shape>& shape_with_layout) { in AllReduce()
4773 const Shape& shape_with_layout, const ChannelHandle& handle) { in SendToHost()
4789 const Shape& shape_with_layout, in OutfeedWithToken()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc408 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto426 ShapeProto shape_with_layout = 2; field
453 ShapeProto shape_with_layout = 1; field
Dliteral_util.cc361 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc223 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
310 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout()
336 const Shape& shape_with_layout, const HloInstruction* instruction, in SetInstructionLayout()
887 const Shape& shape_with_layout, HloInstruction* instruction) { in CreateCopyWithNewLayout()
Dhlo_cost_analysis_test.cc723 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc497 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()