Searched refs:ChooseCompactLayoutForShape (Results 1 – 7 of 7) sorted by relevance
70 StatusOr<xla::Shape> ChooseCompactLayoutForShape(
224 StatusOr<xla::Shape> TpuTransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in tensorflow::tpu::TpuTransferManager
243 virtual StatusOr<Shape> ChooseCompactLayoutForShape(
584 static StatusOr<Shape> ChooseCompactLayoutForShape(const Shape& shape) { in ChooseCompactLayoutForShape() function in xla::__anonee2cb7d30111::CompressingRematerializationTest608 /*block_size_limit=*/1, ChooseCompactLayoutForShape, in RunHloRematerialization()
420 StatusOr<Shape> TransferManager::ChooseCompactLayoutForShape( in ChooseCompactLayoutForShape() function in xla::TransferManager
121 StatusOr<Shape> ChooseCompactLayoutForShape(Shape subshape) { in ChooseCompactLayoutForShape() function
665 transfer_manager->ChooseCompactLayoutForShape(shape)); in BufferFromHostBuffer()847 transfer_manager->ChooseCompactLayoutForShape(shape)); in CreateUninitializedBuffer()872 transfer_manager->ChooseCompactLayoutForShape(literal.shape())); in BufferFromHostLiteral()2143 ->ChooseCompactLayoutForShape(shape); in Compile()