Home
last modified time | relevance | path

Searched refs:new_literal (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc354 Literal new_literal( in ReshapeSlice() local
360 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice()
371 new_literal.Set<bool>(to_multi_index, in ReshapeSlice()
375 new_literal.Set<uint8>(to_multi_index, in ReshapeSlice()
379 new_literal.Set<uint32>(to_multi_index, in ReshapeSlice()
383 new_literal.Set<int32>(to_multi_index, in ReshapeSlice()
387 new_literal.Set<uint64>(to_multi_index, in ReshapeSlice()
391 new_literal.Set<int64>(to_multi_index, in ReshapeSlice()
395 new_literal.Set<float>(to_multi_index, in ReshapeSlice()
399 new_literal.Set<double>(to_multi_index, in ReshapeSlice()
[all …]
Dliteral.cc878 Literal new_literal(permuted_shape); in Transpose() local
880 new_literal.SetDynamicSize(inverse_permutation[i], GetDynamicSize(i)); in Transpose()
882 DCHECK_EQ(ShapeUtil::ByteSizeOf(new_literal.shape()), in Transpose()
884 std::memcpy(new_literal.untyped_data(), untyped_data(), size_bytes()); in Transpose()
885 return new_literal; in Transpose()
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc818 TF_ASSIGN_OR_RETURN(Literal* const new_literal, in ComputeArrayForReshape()
821 return Construct<ConstantArray>(new_literal); in ComputeArrayForReshape()
Dalgebraic_simplifier.cc1652 Literal new_literal(result_shape); in HandleDivide() local
1655 TF_RETURN_IF_ERROR(InvertConstant<half>(*c, &new_literal)); in HandleDivide()
1658 TF_RETURN_IF_ERROR(InvertConstant<float>(*c, &new_literal)); in HandleDivide()
1661 TF_RETURN_IF_ERROR(InvertConstant<bfloat16>(*c, &new_literal)); in HandleDivide()
1664 TF_RETURN_IF_ERROR(InvertConstant<double>(*c, &new_literal)); in HandleDivide()
1667 TF_RETURN_IF_ERROR(InvertConstant<complex64>(*c, &new_literal)); in HandleDivide()
1670 TF_RETURN_IF_ERROR(InvertConstant<complex128>(*c, &new_literal)); in HandleDivide()
1676 simplifier_->CreateConstantWithLayoutUpdated(new_literal.Clone())); in HandleDivide()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc454 xla::LiteralProto new_literal = in TEST() local
458 new_literal.SerializeAsString()); in TEST()
473 EXPECT_TRUE(CompareLiteralProtos(response, new_literal)); in TEST()
587 xla::LiteralProto new_literal = in TEST() local
590 new_literal.SerializeAsString()); in TEST()
604 EXPECT_TRUE(CompareLiteralProtos(new_literal, new_response)); in TEST()