Searched refs:mutable_shape_do_not_use (Results 1 – 6 of 6) sorted by relevance
78 output->mutable_shape_do_not_use()->Swap(&shape); in ReshapeLiteral()
196 *literal.mutable_shape_do_not_use() = received_shape; in TransferLiteralFromOutfeed()
134 Layout* literal_layout = literal.mutable_shape_do_not_use()->mutable_layout(); in XLA_TEST_F()
494 *literal.mutable_shape_do_not_use()->mutable_layout() = in XLA_TEST_F()
599 Shape* mutable_shape_do_not_use() { return shape_.get(); } in mutable_shape_do_not_use() function
834 *output.mutable_shape_do_not_use() = in Reshape()