Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/core/kernels/ | |
bias_op_gpu.h | 69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } function in class:tensorflow::BiasGradGPUProfileResult |
/external/tensorflow/tensorflow/stream_executor/ | |
blas.h | 137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } function in class:stream_executor::blas::ProfileResult 152 void set_algorithm(AlgorithmType val) { algorithm_ = val; } function in class:stream_executor::blas::AlgorithmConfig [all...] |
dnn.h | 776 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } function in class:stream_executor::dnn::ProfileResult 805 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } function in class:stream_executor::dnn::AlgorithmConfig [all...] |