Searched refs:tensor_ops_enabled (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dconvolution_thunk.cc48 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 Dcudnn_convolution_algorithm_picker.cc142 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 Dconvolution_thunk.h62 bool tensor_ops_enabled, const HloInstruction* hlo);
H A Dcudnn_convolution_runner.cc82 VLOG(3) << "tensor_ops_enabled: "
83 << algorithm.algorithm().tensor_ops_enabled();
H A Dir_emitter_unnested.cc429 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 Dcuda_dnn.cc2161 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 Ddnn.h709 bool tensor_ops_enabled() const { return tensor_ops_enabled_; } function in class:perftools::gputools::dnn::AlgorithmDesc

Completed in 584 milliseconds