| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_conversion_folding_test.cc | 98 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add0)); in TEST_F()
100 HloInstruction::CreateConvert(f32_shape, convert0)); in TEST_F()
104 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add1)); in TEST_F()
132 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, mul0)); in TEST_F()
134 HloInstruction::CreateConvert(f32_shape, convert0)); in TEST_F()
139 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, mul1)); in TEST_F()
167 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, sub0)); in TEST_F()
169 HloInstruction::CreateConvert(f32_shape, convert0)); in TEST_F()
174 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, sub1)); in TEST_F()
197 builder.AddInstruction(HloInstruction::CreateConvert(f32_shape, b)); in TEST_F()
[all …]
|
| D | hlo_constant_folding_test.cc | 48 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {}), input)); in TEST_F()
70 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input)); in TEST_F()
92 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {2}), input)); in TEST_F()
|
| D | hlo_element_type_converter.cc | 42 HloInstruction::CreateConvert(shape, hlo)); in ToElementType()
98 HloInstruction::CreateConvert(to_ele_shape, element)); in ConvertTupleElements()
|
| D | bfloat16_normalization.cc | 85 HloInstruction::CreateConvert(hlo->shape(), hlo)); in InsertConvertAfterOutput()
113 auto convert = computation->AddInstruction(HloInstruction::CreateConvert( in InsertConvertBeforeOperand()
249 computation_->AddInstruction(HloInstruction::CreateConvert( in HandleMultipleOutputs()
|
| D | convolution_group_converter.cc | 265 add(HloInstruction::CreateConvert(convert_shape, new_filter)); in HandleBatchGroupCount()
309 HloInstruction::CreateConvert(convert_back_shape, reduce_window); in HandleBatchGroupCount()
|
| D | hlo_ordering_test.cc | 142 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
225 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
|
| D | bfloat16_propagation.cc | 653 HloInstruction::CreateConvert(hlo_subshape, leaf)); in ResolveInconsistentFusions()
748 HloInstruction::CreateConvert(leaf->shape(), leaf)); in Run()
|
| D | batchnorm_expander.cc | 131 return HloInstruction::CreateConvert( in DynamicElementCountPerFeature()
|
| D | bfloat16_propagation_test.cc | 917 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte0)); in TEST_F()
919 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte1)); in TEST_F()
|
| D | hlo_cse_test.cc | 165 HloInstruction::CreateConvert(shape_r0, constants[i])); in TEST_F()
|
| D | hlo_instruction.h | 524 static std::unique_ptr<HloInstruction> CreateConvert(const Shape& shape,
|
| D | hlo_evaluator_test.cc | 530 b.AddInstruction(HloInstruction::CreateConvert(expected.shape(), constant)); in TEST_P()
550 b.AddInstruction(HloInstruction::CreateConvert(expected.shape(), constant)); in TEST_P()
|
| D | hlo_dataflow_analysis_test.cc | 2350 HloInstruction::CreateConvert(data_shape_bf16, gte1)); in TEST_F()
2363 HloInstruction::CreateConvert(data_shape, dynamic_update_slice)); in TEST_F()
|
| D | hlo_instruction.cc | 1015 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConvert( in CreateConvert() function in xla::HloInstruction
1454 clone = CreateConvert(shape, new_operands[0]); in CloneWithNewOperands()
|
| D | algebraic_simplifier.cc | 290 return computation_->AddInstruction(HloInstruction::CreateConvert( in AsType()
3265 computation_->AddInstruction(HloInstruction::CreateConvert( in HandleReduceWindow()
|
| D | algebraic_simplifier_test.cc | 1530 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input)); in TEST_F()
3674 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
|
| D | hlo_parser.cc | 805 HloInstruction::CreateConvert(shape, operands[0])); in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 137 HloInstruction::CreateConvert(elem_shape_U8, param0)); in RunTest1D()
139 HloInstruction::CreateConvert(elem_shape_F32, param1)); in RunTest1D()
146 HloInstruction::CreateConvert(elem_shape_F32, sub_U8)); in RunTest1D()
|