Home
last modified time | relevance | path

Searched refs:ChooseCompactLayoutForShape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_transfer_manager.h70 StatusOr<xla::Shape> ChooseCompactLayoutForShape(
Dtpu_transfer_manager.cc226 StatusOr<xla::Shape> TpuTransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h241 virtual StatusOr<Shape> ChooseCompactLayoutForShape(
Dhlo_rematerialization_test.cc584 static StatusOr<Shape> ChooseCompactLayoutForShape(const Shape& shape) { in ChooseCompactLayoutForShape() function in xla::__anonfdd408820111::CompressingRematerializationTest
612 ChooseCompactLayoutForShape, in RunHloRematerialization()
Dtransfer_manager.cc416 StatusOr<Shape> TransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in xla::TransferManager
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc189 ->ChooseCompactLayoutForShape(shape); in DeserializeExecutable()
Dpjrt_stream_executor_client.cc715 transfer_manager->ChooseCompactLayoutForShape(shape)); in BufferFromHostBuffer()
926 transfer_manager->ChooseCompactLayoutForShape(shape)); in CreateUninitializedBuffer()
951 transfer_manager->ChooseCompactLayoutForShape(literal.shape())); in BufferFromHostLiteral()
2466 ->ChooseCompactLayoutForShape(shape); in Compile()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h147 StatusOr<Shape> ChooseCompactLayoutForShape(Shape subshape) { in ChooseCompactLayoutForShape() function