Searched defs:shape_with_layout (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 51 const Shape& shape_with_layout) { in set_result_layout()
|
| D | client.cc | 42 const Shape* shape_with_layout) { in Transfer()
119 const Shape* shape_with_layout, int64 replica_id, in TransferFromOutfeed()
|
| D | xla_builder.cc | 1385 void XlaBuilder::Outfeed(const XlaOp& operand, const Shape& shape_with_layout, in Outfeed()
1438 const Shape& shape_with_layout, in OutfeedWithToken()
2407 const Shape& shape_with_layout, in SendToHost()
3071 void Outfeed(const XlaOp& operand, const Shape& shape_with_layout, in Outfeed()
3458 const Shape& shape_with_layout, const ChannelHandle& handle) { in SendToHost()
3474 const Shape& shape_with_layout, in OutfeedWithToken()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 172 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()
|
| D | local_client_execute_test.cc | 261 Shape shape_with_layout = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 333 Shape shape_with_layout = buf.shape(); in PropagateBufferConstraint() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 321 ShapeProto shape_with_layout = 2; field
348 ShapeProto shape_with_layout = 1; field
|
| D | literal_util.cc | 308 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 198 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
262 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout()
288 const Shape& shape_with_layout, const HloInstruction* instruction, in SetInstructionLayout()
773 const Shape& shape_with_layout, HloInstruction* instruction) { in CreateCopyWithNewLayout()
|
| D | hlo_cost_analysis_test.cc | 498 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 99 const Literal& argument, const absl::optional<Shape>& shape_with_layout, in FromLiteral()
620 const Shape& shape_with_layout, in CustomCall()
|
| D | xla_client.py | 1421 shape_with_layout, argument
|