Searched refs:DepthWiseConv3x3StrideH2 (Results 1 – 4 of 4) sorted by relevance
35 class DepthWiseConv3x3StrideH2 : public GPUOperation {37 DepthWiseConv3x3StrideH2() = default;45 DepthWiseConv3x3StrideH2(DepthWiseConv3x3StrideH2&& kernel) = default;46 DepthWiseConv3x3StrideH2& operator=(DepthWiseConv3x3StrideH2&& kernel) =48 DepthWiseConv3x3StrideH2(const DepthWiseConv3x3StrideH2&) = delete;49 DepthWiseConv3x3StrideH2& operator=(const DepthWiseConv3x3StrideH2&) = delete;52 explicit DepthWiseConv3x3StrideH2(const OperationDef& definition) in DepthWiseConv3x3StrideH2() function54 friend DepthWiseConv3x3StrideH2 CreateDepthWiseConv3x3StrideH2(71 void DepthWiseConv3x3StrideH2::UploadWeightsAndBiases( in UploadWeightsAndBiases()111 void DepthWiseConv3x3StrideH2::RearrangeWeightsAndBiasesData( in RearrangeWeightsAndBiasesData()[all …]
190 int3 DepthWiseConv3x3StrideH2::GetGridSize() const { in GetGridSize()197 void DepthWiseConv3x3StrideH2::GetPossibleKernelWorkGroups( in GetPossibleKernelWorkGroups()208 DepthWiseConv3x3StrideH2 CreateDepthWiseConv3x3StrideH2( in CreateDepthWiseConv3x3StrideH2()215 DepthWiseConv3x3StrideH2 desc(definition); in CreateDepthWiseConv3x3StrideH2()
53 DepthWiseConv3x3StrideH2 operation = in DepthWiseConv3x3StrideH2SimpleWeightsTest()57 absl::make_unique<DepthWiseConv3x3StrideH2>(std::move(operation)), in DepthWiseConv3x3StrideH2SimpleWeightsTest()
74 return absl::make_unique<DepthWiseConv3x3StrideH2>( in SelectDWConvolutionApple()