Home
last modified time | relevance | path

Searched defs:algorithm_config (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/util/autotune_maps/
Dautotune_serialize_test.cc119 AlgorithmConfig algorithm_config; in TEST() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.h339 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm()
398 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
426 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithExecutionPlan()
491 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardDataWithAlgorithm()
523 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1265 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1281 const AlgorithmConfig& algorithm_config, in PrepareForConvolution()
2017 const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2593 const AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
Dstream_executor_pimpl.cc376 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc507 AlgorithmConfig algorithm_config; in launch() local
Dconv_grad_input_ops.cc385 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_grad_ops_3d.cc1513 AlgorithmConfig algorithm_config; in Compute() local
2073 AlgorithmConfig algorithm_config; in Compute() local
Dconv_grad_filter_ops.cc989 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_ops.cc1107 AlgorithmConfig algorithm_config; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1167 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1195 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1317 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
2357 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2981 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
3044 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
3105 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
3870 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
4880 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
5027 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveWithExecutionPlan()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2570 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2907 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3683 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()