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
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()
Dconv_canonicalization.h38 : target_machine_features_(*target_machine_features) {} in ConvCanonicalization()
48 const TargetMachineFeatures& target_machine_features_;
Dcpu_layout_assignment.h38 target_machine_features_(*target_machine_features) {} in CpuLayoutAssignment()
44 const TargetMachineFeatures& target_machine_features_; variable
Dparallel_task_assignment.cc111 : target_machine_features_(*target_machine_features) { in ParallelTaskAssignment()
148 target_machine_features_)) || in GetTargetParallelTaskCount()
239 &target_machine_features_); in ComputeTargetParallelTasks()
Ddot_op_emitter.cc201 const TargetMachineFeatures& target_machine_features_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter
223 target_machine_features_(target_machine_features) {} in DotOpEmitter()
246 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemm()
320 target_machine_features_.vector_register_num_elements( in EmitTiledLlvmIrGemv()
388 target_machine_features_)) { in Emit()
Dcpu_layout_assignment.cc116 target_machine_features_)) { in AddBackendConstraints()
Dconv_canonicalization.cc37 target_machine_features_)) { in Run()
Dir_emitter.cc99 target_machine_features_(*target_machine_features), in IrEmitter()
260 return target_machine_features_.minimum_alignment_for_allocation(buffer_size); in MinimumAlignmentForShape()
274 target_machine_features_.minimum_alignment_for_allocation(buffer_size); in AttachAlignmentMetadataForLoad()
947 hlo_module_config_, target_machine_features_); in HandleDot()
1103 target_machine_features_)) { in HandleConvolution()
1515 target_machine_features_.vector_register_byte_size( in CreateShardedVectorType()
1679 target_machine_features_.vectorization_factor_in_bytes(); in EmitVectorizedReduce()
2195 hlo_module_config_, target_machine_features_)); in HandleFusion()
Dir_emitter.h564 const TargetMachineFeatures& target_machine_features_; variable