| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 36 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
39 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
45 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
48 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
54 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
57 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 36 const LaunchDimensions& launch_dimensions,
42 const LaunchDimensions& launch_dimensions,
52 const LaunchDimensions& launch_dimensions,
|
| D | kernel_thunk.cc | 90 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
98 launch_dimensions = launch_dimensions_; in ExecuteOnStream()
118 return ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, in ExecuteOnStream()
|
| D | ir_emitter_unnested.cc | 924 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitPadToStaticFromMlir() local
926 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitPadToStaticFromMlir()
929 ParallelLoopEmitter(body_generator, data_shape, launch_dimensions, &b_, in EmitPadToStaticFromMlir()
933 pad_to_static, launch_dimensions.launch_bound(), &b_))); in EmitPadToStaticFromMlir()
1047 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitSliceToDynamicFromMlir() local
1049 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitSliceToDynamicFromMlir()
1053 ParallelLoopEmitter(body_generator, data_shape, launch_dimensions, &b_, in EmitSliceToDynamicFromMlir()
1057 launch_dimensions.launch_bound(), &b_))); in EmitSliceToDynamicFromMlir()
1841 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitLoopFusionFromMlir() local
1844 UpdateLaunchDimensions(launch_dimensions, kernel_thunk, in EmitLoopFusionFromMlir()
[all …]
|
| D | BUILD | 102 name = "launch_dimensions",
104 "launch_dimensions.cc",
107 "launch_dimensions.h",
258 ":launch_dimensions",
325 ":launch_dimensions",
615 ":launch_dimensions",
1257 ":launch_dimensions",
1607 ":launch_dimensions",
1634 ":launch_dimensions",
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 98 const IrArray& output_array, const gpu::LaunchDimensions* launch_dimensions, in EmitDynamicUpdateSliceInPlaceImpl() argument
152 if (launch_dimensions != nullptr) { in EmitDynamicUpdateSliceInPlaceImpl()
154 *launch_dimensions, b) in EmitDynamicUpdateSliceInPlaceImpl()
194 const gpu::LaunchDimensions* launch_dimensions, llvm::IRBuilder<>* b) { in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
233 fusion_output_array, launch_dimensions, IrName(dynamic_update_slice), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
248 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b) { in EmitParallelFusedDynamicUpdateSliceInPlace() argument
250 fusion, fusion_output_array, fused_emitter, &launch_dimensions, b); in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | sort_util.h | 40 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions,
|
| D | dynamic_update_slice_util.h | 85 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b);
|
| D | BUILD | 184 "//tensorflow/compiler/xla/service/gpu:launch_dimensions",
201 "//tensorflow/compiler/xla/service/gpu:launch_dimensions",
|
| D | sort_util.cc | 297 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace() argument
392 launch_dimensions, b) in EmitSortInPlace()
|