Searched refs:compact_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 123 Shape compact_shape; member
540 int64 MemoryReducedIfCompressed(Item* item, const Shape& compact_shape) const;
930 Item* item, const Shape& compact_shape) const { in MemoryReducedIfCompressed()
946 int64 compact_shape_size = size_function_(compact_shape); in MemoryReducedIfCompressed()
1417 Shape compact_shape = in PickRematerializationCandidates() local
1420 MemoryReducedIfCompressed(item, compact_shape); in PickRematerializationCandidates()
1427 << compact_shape.ToString(true); in PickRematerializationCandidates()
1431 best_strategy.compact_shape = compact_shape; in PickRematerializationCandidates()
1683 Item* best_item, const Shape& compact_shape, in CompressInstruction() argument
1688 best_item, compact_shape)) in CompressInstruction()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 664 TF_ASSIGN_OR_RETURN(Shape compact_shape, in BufferFromHostBuffer()
682 if (shape.layout() == compact_shape.layout() && in BufferFromHostBuffer()
722 AllocateDestinationBuffer(compact_shape, device, local_device, in BufferFromHostBuffer()
846 TF_ASSIGN_OR_RETURN(Shape compact_shape, in CreateUninitializedBuffer()
851 AllocateDestinationBuffer(compact_shape, device, local_device, in CreateUninitializedBuffer()
871 Shape compact_shape, in BufferFromHostLiteral()
875 AllocateDestinationBuffer(compact_shape, device, local_device, in BufferFromHostLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 555 auto compact_shape = in CopyFrom() local
557 CHECK(ShapeUtil::DynamicShapeIsCompatible(compact_shape, bound_shape)) in CopyFrom()
558 << compact_shape.ToString() << " vs " << bound_shape.ToString(); in CopyFrom()
|