Searched refs:mutable_shape_do_not_use (Results 1 – 6 of 6) sorted by relevance
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()
240 *literal.mutable_shape_do_not_use() = received_shape; in TransferLiteralFromOutfeedOnCpu()
83 output->mutable_shape_do_not_use()->Swap(&shape); in ReshapeLiteral()
606 Shape* mutable_shape_do_not_use() { return shape_.get(); } in mutable_shape_do_not_use() function
848 *output.mutable_shape_do_not_use() = in Reshape()