Searched defs:target_machine_features (Results 1 – 12 of 12) sorted by relevance
28 const Shape& shape, const TargetMachineFeatures& target_machine_features) { in GetMinimumAlignmentForArray()45 const TargetMachineFeatures& target_machine_features) { in PotentiallyImplementedAsEigenConvolution()
37 const TargetMachineFeatures* target_machine_features) in ConvCanonicalization()
35 const TargetMachineFeatures* target_machine_features) in CpuLayoutAssignment()
213 const TargetMachineFeatures& target_machine_features) in DotOpEmitter()753 const TargetMachineFeatures& target_machine_features) { in AreGemmShapes()772 const TargetMachineFeatures& target_machine_features) { in IsAlignedGemm()784 const TargetMachineFeatures& target_machine_features) { in CanEmitTiledLlvmIrGemm()823 const TargetMachineFeatures& target_machine_features) { in GetDotImplementationStrategy()851 const TargetMachineFeatures& target_machine_features) { in EmitNonBatchDotOperation()933 const TargetMachineFeatures& target_machine_features) { in EmitBatchDotOperation()1006 const TargetMachineFeatures& target_machine_features) { in DotImplementationCanHandleTranspose()1021 const TargetMachineFeatures& target_machine_features) { in DotOperandsAndResultMustHaveRowMajorLayout()1038 const TargetMachineFeatures& target_machine_features) { in EmitDotOperation()
94 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST_F() local156 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST_F() local
43 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in TEST() local
70 const TargetMachineFeatures* target_machine_features) in ParallelTaskAssigner()
249 LLVMTargetMachineFeatures* target_machine_features) { in RunHloPassesThroughLayoutAssn()339 LLVMTargetMachineFeatures* target_machine_features) { in RunHloPassesAfterLayoutAssn()396 LLVMTargetMachineFeatures target_machine_features(target_machine); in RunHloPasses() local663 LLVMTargetMachineFeatures target_machine_features(jit->target_machine()); in RunBackend() local854 LLVMTargetMachineFeatures target_machine_features(target_machine.get()); in CompileAheadOfTime() local
98 const TargetMachineFeatures& target_machine_features) { in OperandsAndResultMustHaveRowMajorLayout()
110 const TargetMachineFeatures* target_machine_features) in ParallelTaskAssignment()
53 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in AssignLayouts() local321 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features( in RunDotOutputFusion() local
88 const TargetMachineFeatures* target_machine_features, in IrEmitter()