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.h52 const TargetMachineFeatures& target_machine_features_; variable
73 target_machine_features_(*target_machine_features) {} in ParallelTaskAssigner()
103 const TargetMachineFeatures& target_machine_features_; variable
Dconv_canonicalization.h38 : target_machine_features_(*target_machine_features) {} in ConvCanonicalization()
48 const TargetMachineFeatures& target_machine_features_;
Dparallel_task_assignment_test.cc35 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features_; member in xla::__anonc5bbb6940111::ParallelTaskAssignmentTest
38 : HloTestBase(), target_machine_features_([](int64 shape_size) { in ParallelTaskAssignmentTest()
44 &target_machine_features_) in RunParallelTaskAssigner()
Dcpu_layout_assignment.h38 target_machine_features_(*target_machine_features) {} in CpuLayoutAssignment()
44 const TargetMachineFeatures& target_machine_features_; variable
Dparallel_task_assignment.cc113 : target_machine_features_(*target_machine_features) { in ParallelTaskAssignment()
163 target_machine_features_))) { in GetTargetParallelTaskCount()
251 &target_machine_features_); in ComputeTargetParallelTasks()
Ddot_op_emitter.cc213 target_machine_features_.vector_register_num_elements( in GetMlirGemmTileSize()
216 int64_t num_registers = target_machine_features_.vector_register_count( in GetMlirGemmTileSize()
231 const TargetMachineFeatures& target_machine_features_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter
252 target_machine_features_(target_machine_features) {} in DotOpEmitter()
322 target_machine_features_.minimum_alignment_for_allocation( in EmitLinalgMatmul()
363 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemm()
460 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemv()
528 target_machine_features_)) { in Emit()
Dcpu_layout_assignment.cc116 target_machine_features_)) { in AddBackendConstraints()
Dconv_canonicalization.cc38 target_machine_features_)) { in Run()
Dir_emitter.cc108 target_machine_features_(*target_machine_features), in IrEmitter()
324 return target_machine_features_.minimum_alignment_for_allocation(buffer_size); in MinimumAlignmentForShape()
338 target_machine_features_.minimum_alignment_for_allocation(buffer_size); in AttachAlignmentMetadataForLoad()
867 hlo_module_config_, target_machine_features_); in HandleDot()
880 target_machine_features_)) { in HandleConvolution()
1480 target_machine_features_.vector_register_byte_size( in CreateShardedVectorType()
1649 target_machine_features_.vector_register_byte_size( in EmitVectorizedReduce()
1659 target_machine_features_.vectorization_factor_in_bytes(); in EmitVectorizedReduce()
2110 hlo_module_config_, target_machine_features_)); in HandleFusion()
Dir_emitter.h590 const TargetMachineFeatures& target_machine_features_; variable