Searched refs:new_literal (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 766 auto new_literal = MakeUnique<Literal>( in Reshape() local
772 Shape shape_with_layout = new_literal->shape(); in Reshape()
783 new_literal->Set<bool>(to_multi_index, in Reshape()
787 new_literal->Set<uint8>(to_multi_index, in Reshape()
791 new_literal->Set<uint32>(to_multi_index, in Reshape()
795 new_literal->Set<int32>(to_multi_index, in Reshape()
799 new_literal->Set<uint64>(to_multi_index, in Reshape()
803 new_literal->Set<int64>(to_multi_index, in Reshape()
807 new_literal->Set<float>(to_multi_index, in Reshape()
811 new_literal->Set<double>(to_multi_index, in Reshape()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 795 std::unique_ptr<Literal> new_literal = CreateFromShape(permuted_shape); in Transpose() local
796 DCHECK_GE(ShapeUtil::ByteSizeOf(new_literal->shape()), in Transpose()
798 std::memcpy(new_literal->root_piece().buffer(), root_piece().buffer(), in Transpose()
800 return new_literal; in Transpose()
|