Searched refs:unroll_factor (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 35 int unroll_factor) in ParallelLoopEmitter() argument
38 unroll_factor_(unroll_factor) {} in ParallelLoopEmitter()
44 int unroll_factor) in ParallelLoopEmitter() argument
47 unroll_factor_(unroll_factor) {} in ParallelLoopEmitter()
53 int unroll_factor) in ParallelLoopEmitter() argument
56 unroll_factor_(unroll_factor) {} in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 37 llvm::IRBuilder<>* b, int unroll_factor = 1);
43 llvm::IRBuilder<>* b, int unroll_factor = 1);
53 llvm::IRBuilder<>* b, int unroll_factor = 1);
|
| D | partition_assignment.cc | 64 int unroll_factor) { in CalculateLaunchDimensions() argument
70 CHECK_EQ(num_elements % unroll_factor, 0); in CalculateLaunchDimensions()
71 num_elements = num_elements / unroll_factor; in CalculateLaunchDimensions()
|
| D | kernel_thunk.h | 52 int unroll_factor);
58 int unroll_factor() const { return unroll_factor_; } in unroll_factor() function
|
| D | kernel_thunk.cc | 33 int unroll_factor) in KernelThunk() argument
37 unroll_factor_(unroll_factor) {} in KernelThunk()
|
| D | partition_assignment.h | 65 int unroll_factor = 1);
|
| D | ir_emitter_unnested.cc | 585 int unroll_factor = ComputeMaxUnrollFactor(fusion); in HandleFusion() local
587 fusion, /*implements_whole_instruction=*/false, unroll_factor)); in HandleFusion()
1598 int unroll_factor) { in BuildKernelThunk() argument
1700 implements_whole_instruction ? inst : nullptr, unroll_factor); in BuildKernelThunk()
2177 int unroll_factor = thunk->unroll_factor(); in EmitTargetElementLoopInThunk() local
2185 << " for unroll_factor " << unroll_factor; in EmitTargetElementLoopInThunk()
2187 element_shape, ir_emitter_context_->device_description(), unroll_factor); in EmitTargetElementLoopInThunk()
2192 launch_dimensions, &b_, unroll_factor) in EmitTargetElementLoopInThunk()
2210 &b_, unroll_factor) in EmitTargetElementLoopInThunk()
2245 int unroll_factor = 1; in EmitTargetElementLoop() local
[all …]
|
| D | ir_emitter_unnested.h | 318 int unroll_factor = 1);
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | loop_unroller.h | 26 LoopUnroller(bool fully_unroll, int unroll_factor) in LoopUnroller() argument
27 : Pass(), fully_unroll_(fully_unroll), unroll_factor_(unroll_factor) {} in LoopUnroller()
|
| D | loop_descriptor.h | 342 size_t unroll_factor);
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_unroller.h | 26 LoopUnroller(bool fully_unroll, int unroll_factor) in LoopUnroller() argument
27 : Pass(), fully_unroll_(fully_unroll), unroll_factor_(unroll_factor) {} in LoopUnroller()
|
| D | loop_descriptor.h | 342 size_t unroll_factor);
|