| /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 | comparison_expander.cc | 95 HloInstruction::CreateConvert(compare_shape, lhs)); in ExpandInstruction()
97 HloInstruction::CreateConvert(compare_shape, rhs)); in ExpandInstruction()
118 HloInstruction::CreateConvert(max_shape, max_signed)); in ExpandInstruction()
|
| D | hlo_constant_folding_test.cc | 50 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {}), input)); in TEST_F()
72 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input)); in TEST_F()
94 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 | operand_upcaster.cc | 69 HloInstruction::CreateConvert(upcast_shape, operand)); in ExpandInstruction()
|
| D | all_reduce_simplifier.cc | 102 HloInstruction::CreateConvert( in Run()
|
| D | bfloat16_normalization.cc | 144 HloInstruction::CreateConvert(new_subshape, leaf)); in ConvertType()
198 HloInstruction::CreateConvert(original_subshape, leaf)); in ChangeOutputTypeThenInsertConvertBack()
364 computation_->AddInstruction(HloInstruction::CreateConvert(shape, gte)); in HandleMultipleOutputs()
|
| D | while_loop_all_reduce_code_motion.cc | 424 while_parent->AddInstruction(HloInstruction::CreateConvert( in CreateSinkedAllReduces()
439 while_parent->AddInstruction(HloInstruction::CreateConvert( in CreateSinkedAllReduces()
|
| D | bfloat16_propagation.cc | 771 HloInstruction::CreateConvert(hlo_subshape, leaf)); in ResolveInconsistentFusions()
868 HloInstruction::CreateConvert(leaf->shape(), leaf)); in Run()
938 HloInstruction::CreateConvert(converted_shape, leaf)); in Run()
|
| D | hlo_ordering_test.cc | 141 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
224 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
|
| D | convolution_group_converter.cc | 362 add(HloInstruction::CreateConvert(convert_shape, new_filter)); in HandleBatchGroupCount()
406 HloInstruction::CreateConvert(convert_back_shape, reduce_window); in HandleBatchGroupCount()
|
| D | copy_insertion_test.cc | 455 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyComputation()
528 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyOneReadOnlyComputation()
1162 HloInstruction::CreateConvert(f32_scalar_shape, iter_param)); in TEST_F()
1166 HloInstruction::CreateConvert(induction_variable_shape_, iter_value)); in TEST_F()
|
| D | batchnorm_expander.cc | 123 return HloInstruction::CreateConvert( in DynamicElementCountPerFeature()
|
| D | hlo_cse_test.cc | 165 HloInstruction::CreateConvert(shape_r0, constants[i])); in TEST_F()
|
| D | bfloat16_propagation_test.cc | 947 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte0)); in TEST_F()
949 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte1)); in TEST_F()
|
| D | hlo_creation_utils.cc | 258 hlo->parent()->AddInstruction(HloInstruction::CreateConvert(shape, hlo)); in MakeConvertToHlo()
|
| D | algebraic_simplifier_test.cc | 1894 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input)); in TEST_F()
1917 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
1919 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
1943 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
1945 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
1970 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
1973 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
4441 builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
|
| D | hlo_instruction.h | 718 static std::unique_ptr<HloInstruction> CreateConvert(const Shape& shape,
|
| D | hlo_dataflow_analysis_test.cc | 2446 HloInstruction::CreateConvert(data_shape_bf16, gte1)); in TEST_F()
2459 HloInstruction::CreateConvert(data_shape, dynamic_update_slice)); in TEST_F()
|
| D | hlo_evaluator_test.cc | 607 b.AddInstruction(HloInstruction::CreateConvert(expected.shape(), constant)); in TEST_P()
627 b.AddInstruction(HloInstruction::CreateConvert(expected.shape(), constant)); in TEST_P()
|
| /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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 138 auto shuffle_one_hot = b->AddInstruction(HloInstruction::CreateConvert( in ShuffleWithinEachPartitionUsingOneHot()
191 auto converted_partition_id = add_hlo(HloInstruction::CreateConvert( in GetCorrectionFactor()
234 auto converted_partition_id = b->AddInstruction(HloInstruction::CreateConvert( in GetFinalFftUsingCollectivePermute()
|
| D | gather_scatter_handler.cc | 76 offset = b->AddInstruction(HloInstruction::CreateConvert( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
401 adjusted_indices = b->AddInstruction(HloInstruction::CreateConvert( in PartitionIndexParallelDimensions()
590 auto not_partition_zero = b_.AddInstruction(HloInstruction::CreateConvert( in HandleScatter()
|
| D | spmd_partitioner.cc | 1872 auto partition_id_s32 = b_.AddInstruction(HloInstruction::CreateConvert( in HandleCustomCall()
2163 offset = b_.AddInstruction(HloInstruction::CreateConvert( in HandleIota()
2550 branch_index = b_.AddInstruction(HloInstruction::CreateConvert( in HandleInfeed()
2953 branch_index = b_.AddInstruction(HloInstruction::CreateConvert( in HandleOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_rewriter.cc | 51 HloInstruction::CreateConvert(shape, hlo)); in AddConvert()
112 HloInstruction::CreateConvert(shape_f32, libcall); in HandleBatchNormInference()
|