Searched refs:CreateBitcastConvert (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | comparison_expander.cc | 47 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 D | bfloat16_normalization_test.cc | 440 builder.AddInstruction(HloInstruction::CreateBitcastConvert(bf16_shape, a)); in TEST_F()
|
H A D | hlo_creation_utils.cc | 323 HloInstruction::CreateBitcastConvert(shape, hlo), metadata); in MakeBitcastConvertToHlo()
|
H A D | bfloat16_propagation_test.cc | 442 HloInstruction::CreateBitcastConvert(f32_shape, a_f)); in TEST_F()
|
H A D | hlo_instruction.h | 833 static std::unique_ptr<HloInstruction> CreateBitcastConvert(
|
H A D | hlo_instruction.cc | 1545 HloInstruction::CreateBitcastConvert(const Shape& shape, in CreateBitcastConvert() function in xla::HloInstruction 2117 clone = CreateBitcastConvert(shape, new_operands[0]); in CloneWithNewOperands()
|
H A D | hlo_parser.cc | 1424 HloInstruction::CreateBitcastConvert(*shape, operands[0])); in CreateInstruction()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | while_transformer_test.cc | 72 auto cast = builder.AddInstruction(HloInstruction::CreateBitcastConvert( in BuildBodyComputation()
|