| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 104 const TargetMachineFeatures& target_machine_features);
117 const TargetMachineFeatures& target_machine_features);
213 const TargetMachineFeatures& target_machine_features) in DotOpEmitter() argument
223 target_machine_features_(target_machine_features) {} in DotOpEmitter()
753 const TargetMachineFeatures& target_machine_features) { in AreGemmShapes() argument
772 const TargetMachineFeatures& target_machine_features) { in IsAlignedGemm() argument
779 dot_info.result_shape, target_machine_features); in IsAlignedGemm()
784 const TargetMachineFeatures& target_machine_features) { in CanEmitTiledLlvmIrGemm() argument
785 CHECK(IsAlignedGemm(dot_info, target_machine_features)); in CanEmitTiledLlvmIrGemm()
823 const TargetMachineFeatures& target_machine_features) { in GetDotImplementationStrategy() argument
[all …]
|
| D | ir_emission_utils.cc | 28 const Shape& shape, const TargetMachineFeatures& target_machine_features) { in GetMinimumAlignmentForArray() argument
39 return target_machine_features.minimum_alignment_for_allocation( in GetMinimumAlignmentForArray()
45 const TargetMachineFeatures& target_machine_features) { in PotentiallyImplementedAsEigenConvolution() argument
58 return GetMinimumAlignmentForArray(shape, target_machine_features) >= in PotentiallyImplementedAsEigenConvolution()
|
| D | dot_op_emitter.h | 37 const TargetMachineFeatures& target_machine_features);
43 const TargetMachineFeatures& target_machine_features);
68 const TargetMachineFeatures& target_machine_features);
|
| D | parallel_task_assignment.h | 44 const TargetMachineFeatures* target_machine_features);
70 const TargetMachineFeatures* target_machine_features) in ParallelTaskAssigner() argument
73 target_machine_features_(*target_machine_features) {} in ParallelTaskAssigner()
|
| D | conv_canonicalization_test.cc | 94 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST_F() local
98 ConvCanonicalization conv_canonicalization(&target_machine_features); in TEST_F()
156 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST_F() local
160 ConvCanonicalization conv_canonicalization(&target_machine_features); in TEST_F()
|
| D | cpu_compiler.cc | 249 LLVMTargetMachineFeatures* target_machine_features) { in RunHloPassesThroughLayoutAssn() argument
280 pipeline.AddPass<ConvCanonicalization>(target_machine_features); in RunHloPassesThroughLayoutAssn()
316 *target_machine_features) in RunHloPassesThroughLayoutAssn()
325 LayoutAssignment::InstructionCanChangeLayout, target_machine_features); in RunHloPassesThroughLayoutAssn()
339 LLVMTargetMachineFeatures* target_machine_features) { in RunHloPassesAfterLayoutAssn() argument
379 max_parallelism, ShapeSizeBytesFunction(), target_machine_features); in RunHloPassesAfterLayoutAssn()
396 LLVMTargetMachineFeatures target_machine_features(target_machine); in RunHloPasses() local
398 &target_machine_features)); in RunHloPasses()
400 &target_machine_features); in RunHloPasses()
663 LLVMTargetMachineFeatures target_machine_features(jit->target_machine()); in RunBackend() local
[all …]
|
| D | ir_emission_utils.h | 28 const TargetMachineFeatures& target_machine_features);
33 const Shape& shape, const TargetMachineFeatures& target_machine_features);
|
| D | conv_canonicalization.h | 37 const TargetMachineFeatures* target_machine_features) in ConvCanonicalization() argument
38 : target_machine_features_(*target_machine_features) {} in ConvCanonicalization()
|
| D | ir_emission_utils_test.cc | 43 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST() local
48 *conv_instr, target_machine_features)); in TEST()
|
| D | cpu_layout_assignment.h | 35 const TargetMachineFeatures* target_machine_features) in CpuLayoutAssignment() argument
38 target_machine_features_(*target_machine_features) {} in CpuLayoutAssignment()
|
| D | cpu_layout_assignment.cc | 98 const TargetMachineFeatures& target_machine_features) { in OperandsAndResultMustHaveRowMajorLayout() argument
101 target_machine_features); in OperandsAndResultMustHaveRowMajorLayout()
104 target_machine_features); in OperandsAndResultMustHaveRowMajorLayout()
|
| D | BUILD | 94 ":target_machine_features",
276 ":target_machine_features",
314 name = "target_machine_features",
316 "target_machine_features.cc",
318 hdrs = ["target_machine_features.h"],
333 ":target_machine_features",
399 ":target_machine_features",
763 ":target_machine_features",
795 ":target_machine_features",
836 ":target_machine_features",
[all …]
|
| D | cpu_compiler.h | 164 LLVMTargetMachineFeatures* target_machine_features);
169 LLVMTargetMachineFeatures* target_machine_features);
|
| D | cpu_layout_assignment_test.cc | 53 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in AssignLayouts() local
59 &target_machine_features); in AssignLayouts()
321 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in RunDotOutputFusion() local
327 &target_machine_features); in RunDotOutputFusion()
|
| D | parallel_task_assignment.cc | 110 const TargetMachineFeatures* target_machine_features) in ParallelTaskAssignment() argument
111 : target_machine_features_(*target_machine_features) { in ParallelTaskAssignment()
|
| D | dot_op_emitter_internal.h | 83 const TargetMachineFeatures& target_machine_features);
|
| D | ir_emitter.cc | 88 const TargetMachineFeatures* target_machine_features, in IrEmitter() argument
99 target_machine_features_(*target_machine_features), in IrEmitter()
|