Searched refs:GpuAsmOpts (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.h | 39 GpuAsmOpts options);
49 GpuAsmOpts options);
56 int device_ordinal, const char* ptx, GpuAsmOpts compilation_options);
|
| D | gpu_asm_opts.h | 28 struct GpuAsmOpts { struct
37 explicit GpuAsmOpts(bool disable_gpuasm_optimizations = false, argument
|
| D | redzone_allocator.h | 47 GpuAsmOpts gpu_compilation_opts_,
115 GpuAsmOpts gpu_compilation_opts_;
|
| D | asm_compiler.cc | 110 int device_ordinal, const char* ptx, GpuAsmOpts compilation_options) { in CompileGpuAsmOrGetCached()
111 using PtxCacheKey = std::tuple<int, std::string, GpuAsmOpts::PtxOptionsTuple>; in CompileGpuAsmOrGetCached()
143 GpuAsmOpts options) { in CompileGpuAsm()
200 GpuAsmOpts options) { in CompileGpuAsm()
|
| D | redzone_allocator.cc | 50 GpuAsmOpts ptx_compilation_opts, in RedzoneAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 56 GpuAsmOpts opts; in TEST()
131 GpuAsmOpts opts; in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.h | 79 se::GpuAsmOpts PtxOptsFromConfig(const HloModuleConfig& hlo_module_config);
|
| D | stream_executor_util.cc | 226 se::GpuAsmOpts PtxOptsFromConfig(const HloModuleConfig& hlo_module_config) { in PtxOptsFromConfig()
231 return se::GpuAsmOpts( in PtxOptsFromConfig()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 397 se::GpuAsmOpts()); in operator ()()
413 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_ops_3d.cc | 513 se::GpuAsmOpts()); in launch()
533 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in launch()
|
| D | conv_ops_fused_impl.h | 389 se::GpuAsmOpts());
397 stream, &tf_allocator_adapter, se::GpuAsmOpts(),
|
| D | conv_grad_filter_ops.cc | 991 se::GpuAsmOpts()); in operator ()()
1007 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_ops.cc | 1009 se::GpuAsmOpts()); in operator ()()
1018 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in operator ()()
|
| D | conv_grad_ops_3d.cc | 1431 se::GpuAsmOpts()); in Compute()
1447 stream, &tf_allocator_adapter, se::GpuAsmOpts(), in Compute()
|