Home
last modified time | relevance | path

Searched defs:shape_with_layout (Results 1 – 14 of 14) 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()
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 replica_id, in TransferFromOutfeed()
Dxla_builder.cc1768 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed()
1821 const Shape& shape_with_layout, in OutfeedWithToken()
1841 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()
2784 const absl::optional<Shape>& shape_with_layout) { in AllReduce()
3239 const Shape& shape_with_layout, in SendToHost()
4453 void Outfeed(const XlaOp operand, const Shape& shape_with_layout, in Outfeed()
4672 const absl::optional<Shape>& shape_with_layout) { in AllReduce()
4911 const Shape& shape_with_layout, const ChannelHandle& handle) { in SendToHost()
4927 const Shape& shape_with_layout, in OutfeedWithToken()
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc137 const py::bytes& opaque, bool has_side_effect) -> XlaOp { in BuildOpsSubmodule()
Dxla.cc141 Shape shape_with_layout = shape; in PYBIND11_MODULE() local
Doutfeed_receiver.cc421 Shape shape_with_layout = builder->GetShape(data).ValueOrDie(); in AddOutfeedToBuilder() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc366 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto399 ShapeProto shape_with_layout = 2; field
426 ShapeProto shape_with_layout = 1; field
Dliteral_util.cc360 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc208 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
272 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout()
298 const Shape& shape_with_layout, const HloInstruction* instruction, in SetInstructionLayout()
800 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.cc480 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()