Home
last modified time | relevance | path

Searched defs:ChooseCompactLayoutForShape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_transfer_manager.cc226 StatusOr<xla::Shape> TpuTransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in tensorflow::tpu::TpuTransferManager
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.h147 StatusOr<Shape> ChooseCompactLayoutForShape(Shape subshape) { in ChooseCompactLayoutForShape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.cc416 StatusOr<Shape> TransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in xla::TransferManager
H A Dhlo_rematerialization_test.cc584 static StatusOr<Shape> ChooseCompactLayoutForShape(const Shape& shape) { in ChooseCompactLayoutForShape() function in xla::__anon504c9e020111::CompressingRematerializationTest