Searched defs:unroll_factor (Results 1 – 11 of 11) sorted by relevance
75 const int unroll_factor = in EmitLinearBaseAndThreadIdx() local155 const int unroll_factor = launch_config_.unroll_factor; in EmitLogicalIndexAndSetExitBasicBlock() local
86 int unroll_factor = 1; member
730 int unroll_factor = 1; in EmitPadToStatic() local856 int unroll_factor = 1; in EmitSliceToDynamic() local1685 int unroll_factor; in EmitLoopFusion() local5114 constexpr int unroll_factor = 1; in EmitInputFusibleNonStridedSlices() local5204 auto unroll_factor = ComputeMaxUnrollFactor(fusion_op, hlo_module_config_); in EmitScatter() local
181 uint32_t unroll_factor = in GetSIMDUnrollingFactor() local327 uint32_t unroll_factor = 0; in GetSIMDUnrollingFactor() local
49 …_t, dim3, dim3> calc_execution_policy(const int64_t total_elements, const uint32_t unroll_factor) { in calc_execution_policy()119 const int unroll_factor = sizeof(dist_func_return_t) / sizeof(accscalar_t); in distribution_nullary_kernel() local
83 const int unroll_factor = std::is_same<scalar_t, double>::value ? 2 : 4; in _rrelu_with_noise_cuda_train() local
274 constexpr int unroll_factor = sizeof(arg0_t) >= 4 ? 2 : 4; in gpu_kernel_impl_nocast() local
568 constexpr int unroll_factor = 8; in GammaBetaBackwardCUDAKernel_32x32() local668 constexpr int unroll_factor = 8; in GammaBetaBackwardCUDAKernel() local
26 LoopUnroller(bool fully_unroll, int unroll_factor) in LoopUnroller()