Searched refs:new_literal (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/ |
D | literal_util.cc | 354 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 …]
|
D | literal.cc | 878 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/ |
D | indexed_array_analysis.cc | 818 TF_ASSIGN_OR_RETURN(Literal* const new_literal, in ComputeArrayForReshape() 821 return Construct<ConstantArray>(new_literal); in ComputeArrayForReshape()
|
D | algebraic_simplifier.cc | 1652 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/ |
D | raw_api_test.cc | 454 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()
|