Searched refs:miopen_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 491 miopenHandle_t miopen_handle = nullptr; in Init() local
493 reinterpret_cast<miopenHandle_t*>(&miopen_handle), (hipStream_t)(0)); in Init()
495 miopen_.reset(new MIOpenAccess(miopen_handle)); in Init()
499 CHECK_EQ(miopen_handle, nullptr); in Init()
912 ScopedFusionPlanBase(miopenHandle_t miopen_handle, in ScopedFusionPlanBase() argument
915 : miopen_handle_(miopen_handle), in ScopedFusionPlanBase()
1118 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1123 : ScopedFusionPlanBase(miopen_handle, miopenVerticalFusion, in ScopedFusionPlanConvolutionBiasActivation()
1125 uint64 hash = GetFusionOpHashValue(miopen_handle, input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1198 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue() argument
[all …]
|