HomeSort by relevance Sort by last modified time
    Searched refs:tensor_ops_enabled (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_algorithm_picker.cc 67 if (algo.tensor_ops_enabled()) {
260 result.mutable_conv()->set_tensor_ops_enabled(alg.tensor_ops_enabled());
293 failure->set_tensor_ops_enabled(first_algorithm->tensor_ops_enabled());
374 << " tensor_ops_enabled: " << best_algo.conv().tensor_ops_enabled();
386 backend_config.set_tensor_ops_enabled(best_algo.conv().tensor_ops_enabled());
cudnn_conv_runner.cc 130 VLOG(3) << "tensor_ops_enabled: "
131 << algorithm.algorithm()->tensor_ops_enabled();
338 backend_config.algorithm(), backend_config.tensor_ops_enabled()));
  /external/tensorflow/tensorflow/core/kernels/
gpu_utils.cc 140 best_result->conv().tensor_ops_enabled()});
146 best_result_no_scratch->conv().tensor_ops_enabled()});
cudnn_rnn_ops.cc     [all...]
conv_ops_3d.cc 469 profile_algorithm.tensor_ops_enabled());
conv_ops_fused_impl.h 538 profile_algorithm.tensor_ops_enabled());
    [all...]
conv_grad_filter_ops.cc     [all...]
conv_grad_input_ops.cc     [all...]
conv_ops.cc     [all...]
  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 373 best_result->conv().tensor_ops_enabled()});
379 best_result_no_scratch->conv().tensor_ops_enabled()});
735 profile_algorithm.tensor_ops_enabled());
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h 747 bool tensor_ops_enabled() const { function in class:stream_executor::dnn::AlgorithmDesc
753 tensor_ops_enabled() == other.tensor_ops_enabled();
    [all...]
dnn.cc 26 return ::tensorflow::Hash64Combine(algo_id(), tensor_ops_enabled());
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]

Completed in 550 milliseconds