Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched defs:tensor_ops_enabled (Results 1 - 3 of 3) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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)));
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);     [all...]
/external/tensorflow/tensorflow/stream_executor/
dnn.h	709 bool tensor_ops_enabled() const { return tensor_ops_enabled_; } function in class:perftools::gputools::dnn::AlgorithmDesc     [all...]

Completed in 123 milliseconds