| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | client_library_test_base.cc | 171 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()
|
| H A D | local_client_execute_test.cc | 264 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | ops.cc | 194 absl::Span<const XlaOp> operands, const Shape& shape_with_layout, in BuildOpsSubmodule()
200 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| H A D | outfeed_receiver.cc | 428 Shape shape_with_layout = builder->GetShape(data).ValueOrDie(); in AddOutfeedToBuilder() local
|
| H A D | xla.cc | 161 Shape shape_with_layout = shape; in PYBIND11_MODULE() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | client.cc | 41 const Shape* shape_with_layout) { in Transfer()
118 const Shape* shape_with_layout, int64_t replica_id, in TransferFromOutfeed()
|
| H A D | executable_build_options.cc | 52 const Shape& shape_with_layout) { in set_result_layout()
|
| H A D | xla_builder.cc | 1854 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment.cc | 235 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()
|
| H A D | hlo_cost_analysis_test.cc | 826 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | xla.proto | 533 ShapeProto shape_with_layout = 2; field
560 ShapeProto shape_with_layout = 1; field
|
| H A D | literal_util.cc | 404 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 621 XlaOp operand, XlaOp token, const Shape& shape_with_layout, in OutfeedWithTokenInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1230 xla::Shape shape_with_layout = xla::ShapeUtil::MakeTupleShape(subshapes); in ExportXlaOp() local
|