Searched refs:unroll_factor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | launch_dimensions.cc | 59 int unroll_factor, bool few_waves) { in CalculateLaunchDimensions() argument
65 CHECK_EQ(num_elements % unroll_factor, 0); in CalculateLaunchDimensions()
66 num_elements = num_elements / unroll_factor; in CalculateLaunchDimensions()
85 RoundUpToNearest(threads_per_block / unroll_factor, int64{32}); in CalculateLaunchDimensions()
|
| D | parallel_loop_emitter.cc | 37 int unroll_factor) in ParallelLoopEmitter() argument
40 unroll_factor_(unroll_factor) {} in ParallelLoopEmitter()
46 int unroll_factor) in ParallelLoopEmitter() argument
49 unroll_factor_(unroll_factor) {} in ParallelLoopEmitter()
55 int unroll_factor) in ParallelLoopEmitter() argument
58 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 | launch_dimensions.h | 70 int unroll_factor = 1,
|
| D | ir_emitter_unnested.cc | 727 int unroll_factor = 1; in EmitUsingElementalIrEmitter() local
729 unroll_factor = ComputeMaxUnrollFactor(input.op, hlo_module_config_); in EmitUsingElementalIrEmitter()
733 return EmitLoopFusionFromMlir(input, output_shape, unroll_factor); in EmitUsingElementalIrEmitter()
817 int unroll_factor = 1; in EmitPadToStaticFromMlir() local
925 input_shape, ir_emitter_context_->gpu_device_info(), unroll_factor); in EmitPadToStaticFromMlir()
930 unroll_factor) in EmitPadToStaticFromMlir()
944 int unroll_factor = 1; in EmitSliceToDynamicFromMlir() local
1048 input_shape, ir_emitter_context_->gpu_device_info(), unroll_factor); in EmitSliceToDynamicFromMlir()
1054 unroll_factor) in EmitSliceToDynamicFromMlir()
1811 int unroll_factor; in EmitLoopFusionFromMlir() local
[all …]
|
| D | ir_emitter_unnested.h | 235 KernelThunk* thunk, int unroll_factor, bool few_waves = false);
|
| /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);
|