Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h121 StatusOr<Shape> ChooseCompactLayoutForShape(Shape subshape) { in ChooseCompactLayoutForShape() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.cc224 StatusOr<xla::Shape> TpuTransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc420 StatusOr<Shape> TransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in xla::TransferManager
Dhlo_rematerialization_test.cc584 static StatusOr<Shape> ChooseCompactLayoutForShape(const Shape& shape) { in ChooseCompactLayoutForShape() function in xla::__anonee2cb7d30111::CompressingRematerializationTest