Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc398 Literal new_literal( in ReshapeSlice() local
404 Shape shape_with_layout = new_literal.shape(); in ReshapeSlice()
415 new_literal.Set<bool>(to_multi_index, in ReshapeSlice()
419 new_literal.Set<uint8_t>(to_multi_index, in ReshapeSlice()
423 new_literal.Set<uint32_t>(to_multi_index, in ReshapeSlice()
427 new_literal.Set<int32_t>(to_multi_index, in ReshapeSlice()
431 new_literal.Set<uint64_t>(to_multi_index, in ReshapeSlice()
435 new_literal.Set<int64_t>(to_multi_index, in ReshapeSlice()
439 new_literal.Set<float>(to_multi_index, in ReshapeSlice()
443 new_literal.Set<double>(to_multi_index, in ReshapeSlice()
[all …]
Dliteral.cc1018 Literal new_literal(permuted_shape); in Transpose() local
1020 new_literal.SetDynamicSize(inverse_permutation[i], GetDynamicSize(i)); in Transpose()
1022 DCHECK_EQ(ShapeUtil::ByteSizeOf(new_literal.shape()), in Transpose()
1024 std::memcpy(new_literal.untyped_data(), untyped_data(), size_bytes()); in Transpose()
1025 return new_literal; in Transpose()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_simplify_padding_test.cc110 Literal new_literal = instr->literal().Clone(); in SetConstantValue() local
111 new_literal.MutableEachCell<int8_t>(value_fn); in SetConstantValue()
113 instr, HloInstruction::CreateConstant(std::move(new_literal)))); in SetConstantValue()
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc823 Literal* const new_literal, in ComputeArrayForReshape()
825 return Construct<ConstantArray>(new_literal); in ComputeArrayForReshape()
Dalgebraic_simplifier.cc1866 Literal new_literal(result_shape); in HandleDivide() local
1869 TF_RETURN_IF_ERROR(InvertConstant<half>(*c, &new_literal)); in HandleDivide()
1872 TF_RETURN_IF_ERROR(InvertConstant<float>(*c, &new_literal)); in HandleDivide()
1875 TF_RETURN_IF_ERROR(InvertConstant<bfloat16>(*c, &new_literal)); in HandleDivide()
1878 TF_RETURN_IF_ERROR(InvertConstant<double>(*c, &new_literal)); in HandleDivide()
1881 TF_RETURN_IF_ERROR(InvertConstant<complex64>(*c, &new_literal)); in HandleDivide()
1884 TF_RETURN_IF_ERROR(InvertConstant<complex128>(*c, &new_literal)); in HandleDivide()
1890 simplifier_->CreateConstantWithLayoutUpdated(new_literal.Clone())); in HandleDivide()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc453 xla::LiteralProto new_literal = in TEST() local
457 new_literal.SerializeAsString()); in TEST()
472 EXPECT_TRUE(CompareLiteralProtos(response, new_literal)); in TEST()
586 xla::LiteralProto new_literal = in TEST() local
589 new_literal.SerializeAsString()); in TEST()
603 EXPECT_TRUE(CompareLiteralProtos(new_literal, new_response)); in TEST()