Home
last modified time | relevance | path

Searched refs:CreateBitcastConvert (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcomparison_expander.cc47 HloInstruction::CreateBitcastConvert(signed_shape, value)); in BitcastConvertFloatingPointToIntegral()
49 HloInstruction::CreateBitcastConvert(unsigned_shape, value)); in BitcastConvertFloatingPointToIntegral()
53 HloInstruction::CreateBitcastConvert(signed_shape, flipped_value)); in BitcastConvertFloatingPointToIntegral()
H A Dbfloat16_normalization_test.cc440 builder.AddInstruction(HloInstruction::CreateBitcastConvert(bf16_shape, a)); in TEST_F()
H A Dhlo_creation_utils.cc323 HloInstruction::CreateBitcastConvert(shape, hlo), metadata); in MakeBitcastConvertToHlo()
H A Dbfloat16_propagation_test.cc442 HloInstruction::CreateBitcastConvert(f32_shape, a_f)); in TEST_F()
H A Dhlo_instruction.h833 static std::unique_ptr<HloInstruction> CreateBitcastConvert(
H A Dhlo_instruction.cc1545 HloInstruction::CreateBitcastConvert(const Shape& shape, in CreateBitcastConvert() function in xla::HloInstruction
2117 clone = CreateBitcastConvert(shape, new_operands[0]); in CloneWithNewOperands()
H A Dhlo_parser.cc1424 HloInstruction::CreateBitcastConvert(*shape, operands[0])); in CreateInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dwhile_transformer_test.cc72 auto cast = builder.AddInstruction(HloInstruction::CreateBitcastConvert( in BuildBodyComputation()