Home
last modified time | relevance | path

Searched refs:target_machine_features_ (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.h53 const TargetMachineFeatures& target_machine_features_; variable
74 target_machine_features_(*target_machine_features) {} in ParallelTaskAssigner()
109 const TargetMachineFeatures& target_machine_features_; variable
Dparallel_task_assignment_test.cc36 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features_; member in xla::__anon2217d0190111::ParallelTaskAssignmentTest
39 : HloTestBase(), target_machine_features_([](int64_t shape_size) { in ParallelTaskAssignmentTest()
45 &target_machine_features_) in RunParallelTaskAssigner()
Dcpu_layout_assignment.h36 target_machine_features_(*target_machine_features) {} in LayoutAssignment()
42 const TargetMachineFeatures& target_machine_features_; variable
Dconv_canonicalization.h38 : target_machine_features_(*target_machine_features) {} in ConvCanonicalization()
50 const TargetMachineFeatures& target_machine_features_;
Dparallel_task_assignment.cc116 : target_machine_features_(*target_machine_features) { in ParallelTaskAssignment()
167 target_machine_features_))) { in GetTargetParallelTaskCount()
257 &target_machine_features_); in ComputeTargetParallelTasks()
Ddot_op_emitter.cc224 target_machine_features_.vector_register_num_elements( in GetMlirGemmTileSize()
227 int64_t num_registers = target_machine_features_.vector_register_count( in GetMlirGemmTileSize()
242 const TargetMachineFeatures& target_machine_features_; member in xla::cpu::__anon162db8010111::DotOpEmitter
263 target_machine_features_(target_machine_features) {} in DotOpEmitter()
401 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemm()
498 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemv()
566 target_machine_features_)) { in Emit()
Dcpu_layout_assignment.cc117 target_machine_features_)) { in AddBackendConstraints()
Dconv_canonicalization.cc40 target_machine_features_)) { in Run()
Dir_emitter.cc119 target_machine_features_(*target_machine_features), in IrEmitter()
351 return target_machine_features_.minimum_alignment_for_allocation(buffer_size); in MinimumAlignmentForShape()
365 target_machine_features_.minimum_alignment_for_allocation(buffer_size); in AttachAlignmentMetadataForLoad()
902 hlo_module_config_, target_machine_features_); in HandleDot()
916 target_machine_features_)) { in HandleConvolution()
1576 target_machine_features_.vector_register_byte_size( in CreateShardedVectorType()
1751 target_machine_features_.vector_register_byte_size( in EmitVectorizedReduce()
1761 target_machine_features_.vectorization_factor_in_bytes(); in EmitVectorizedReduce()
2222 hlo_module_config_, target_machine_features_)); in HandleFusion()
Dir_emitter.h651 const TargetMachineFeatures& target_machine_features_; variable