Searched refs:tensor_ops_enabled (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | convolution_thunk.cc | 48 bool tensor_ops_enabled, const HloInstruction* hlo) 62 tensor_ops_enabled_(tensor_ops_enabled) {} 40 ConvolutionThunk( CudnnConvKind convolution_kind, const BufferAllocation::Slice& input_buffer, const BufferAllocation::Slice& filter_buffer, const BufferAllocation::Slice& output_buffer, const BufferAllocation::Slice& tuple_result_buffer, const BufferAllocation::Slice& scratch_buffer, const Shape& input_shape, const Shape& filter_shape, const Shape& output_shape, const Window& window, const ConvolutionDimensionNumbers& dim_nums, int64 algorithm, bool tensor_ops_enabled, const HloInstruction* hlo) argument
|
H A D | cudnn_convolution_algorithm_picker.cc | 142 if (algo.tensor_ops_enabled()) { 254 best_result.algorithm().tensor_ops_enabled(), 297 bool tensor_ops_enabled; local 300 std::tie(algorithm, tensor_ops_enabled, scratch_bytes) = *alg_scratch_and_tc; 305 << " tensor_ops_enabled: " << tensor_ops_enabled; 317 Literal::CreateR0<bool>(tensor_ops_enabled)));
|
H A D | convolution_thunk.h | 62 bool tensor_ops_enabled, const HloInstruction* hlo);
|
H A D | cudnn_convolution_runner.cc | 82 VLOG(3) << "tensor_ops_enabled: " 83 << algorithm.algorithm().tensor_ops_enabled();
|
H A D | ir_emitter_unnested.cc | 429 bool tensor_ops_enabled = tensor_ops_enabled_inst->literal().Get<bool>({}); local 445 algorithm, tensor_ops_enabled, custom_call); 458 algorithm, tensor_ops_enabled, custom_call); 471 algorithm, tensor_ops_enabled, custom_call);
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
H A D | cuda_dnn.cc | 2161 use_tensor_ops = algorithm_config.algorithm().tensor_ops_enabled(); 2211 use_tensor_ops = algorithm_config.algorithm().tensor_ops_enabled(); 2404 use_tensor_ops = algotype.tensor_ops_enabled(); 2443 use_tensor_ops = algotype.tensor_ops_enabled(); 2552 conv.set_use_tensor_op_math(algotype.tensor_ops_enabled()); 3248 conv.set_use_tensor_op_math(algotype.tensor_ops_enabled()); 3289 conv.set_use_tensor_op_math(algotype.tensor_ops_enabled()); 3328 bool use_tensor_ops = algorithm_config.algorithm().tensor_ops_enabled(); 3497 conv.set_use_tensor_op_math(algotype.tensor_ops_enabled()); 3536 conv.set_use_tensor_op_math(algotype.tensor_ops_enabled()); [all...] |
/external/tensorflow/tensorflow/stream_executor/ |
H A D | dnn.h | 709 bool tensor_ops_enabled() const { return tensor_ops_enabled_; } function in class:perftools::gputools::dnn::AlgorithmDesc
|
Completed in 584 milliseconds