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.cc194 absl::Span<const XlaOp> operands, const Shape& shape_with_layout, in BuildOpsSubmodule()
200 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
Dxla.cc161 Shape shape_with_layout = shape; in PYBIND11_MODULE() local
Doutfeed_receiver.cc428 Shape shape_with_layout = builder->GetShape(data).ValueOrDie(); in AddOutfeedToBuilder() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc171 absl::Span<GlobalData* const> arguments, const Shape* shape_with_layout) { in ComputeAndCompareLiteral()
179 const Shape* shape_with_layout) { in ComputeAndCompareLiteral()
275 const Shape* shape_with_layout) { in ComputeAndTransfer()
299 const Shape* shape_with_layout) { in ComputeAndCompareLiteralWithStatus()
361 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.cc41 const Shape* shape_with_layout) { in Transfer()
118 const Shape* shape_with_layout, int64_t replica_id, in TransferFromOutfeed()
Dxla_builder.cc1854 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed()
1907 const Shape& shape_with_layout, in OutfeedWithToken()
1927 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()
2939 const std::optional<Shape>& shape_with_layout, in AllReduce()
3469 const Shape& shape_with_layout, in SendToHost()
4410 void Outfeed(const XlaOp operand, const Shape& shape_with_layout, in Outfeed()
4672 const std::optional<Shape>& shape_with_layout, in AllReduce()
4945 const Shape& shape_with_layout, const ChannelHandle& handle) { in SendToHost()
4961 const Shape& shape_with_layout, in OutfeedWithToken()
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto533 ShapeProto shape_with_layout = 2; field
560 ShapeProto shape_with_layout = 1; field
Dliteral_util.cc404 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc235 Status LayoutAssignment::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
343 LayoutAssignment* assignment, const Shape& shape_with_layout, in SetResultLayout()
394 Status LayoutAssignment::SetInstructionLayout(const Shape& shape_with_layout, in SetInstructionLayout()
990 const Shape& shape_with_layout, HloInstruction* instruction) { in CreateCopyWithNewLayout()
Dhlo_cost_analysis_test.cc826 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.cc621 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1230 xla::Shape shape_with_layout = xla::ShapeUtil::MakeTupleShape(subshapes); in ExportXlaOp() local