Searched refs:mutable_shape_do_not_use (Results 1 – 6 of 6) sorted by relevance
132 Layout* literal_layout = literal.mutable_shape_do_not_use()->mutable_layout(); in XLA_TEST_F()
498 *literal.mutable_shape_do_not_use()->mutable_layout() = in XLA_TEST_F()
242 *literal.mutable_shape_do_not_use() = received_shape; in TransferLiteralFromOutfeedOnCpu()
86 output->mutable_shape_do_not_use()->Swap(&shape); in ReshapeLiteral()
224 Shape* MutableLiteralBase::mutable_shape_do_not_use() { in mutable_shape_do_not_use() function in xla::MutableLiteralBase677 ShapeUtil::GetMutableSubshape(mutable_shape_do_not_use(), shape_index); in SetDynamicSize()974 *output.mutable_shape_do_not_use() = in Reshape()
787 Shape* mutable_shape_do_not_use();