Searched refs:miopen_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 611 miopenHandle_t miopen_handle = nullptr; in Init() local
613 reinterpret_cast<miopenHandle_t*>(&miopen_handle), (hipStream_t)(0)); in Init()
615 miopen_.reset(new MIOpenAccess(miopen_handle)); in Init()
619 CHECK_EQ(miopen_handle, nullptr); in Init()
1029 ScopedFusionPlanBase(miopenHandle_t miopen_handle, in ScopedFusionPlanBase() argument
1032 : miopen_handle_(miopen_handle), in ScopedFusionPlanBase()
1231 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1236 : ScopedFusionPlanBase(miopen_handle, miopenVerticalFusion, in ScopedFusionPlanConvolutionBiasActivation()
1238 uint64 hash = GetFusionOpHashValue(miopen_handle, input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1311 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue() argument
[all …]
|