Searched defs:algo_id (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.proto | 114 int64 algo_id = 1; field
|
| D | dnn.h | 787 AlgorithmDesc(Index algo_id, bool use_tensor_ops) in AlgorithmDesc()
789 AlgorithmDesc(Index algo_id, bool use_tensor_ops, in AlgorithmDesc()
813 Index algo_id() const { return proto_.algo_id(); } in algo_id() function
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 521 int64_t algo_id; in GetProfileResultFromConvAlgoPerf() local
2953 RocmConvRunner(GpuExecutor* parent, MIOpenAccess* miopen, int64_t algo_id, in RocmConvRunner()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4216 int64_t algo_id, bool tensor_ops_enabled, in CudnnLegacyConvRunner()
5016 int64_t algo_id, bool tensor_ops_enabled, in CudnnLegacyFusedConvRunner()
|