Searched refs:SplitAttributes (Results 1 – 8 of 8) sorted by relevance
31 const SplitAttributes& attr, const std::vector<int>& channels);45 SplitAttributes attr_;49 const SplitAttributes& attr,
40 SplitAttributes attr; in SplitChannelsTest()84 SplitAttributes attr; in SplitChannelsX4Test()128 SplitAttributes attr; in SplitWidthTest()171 SplitAttributes attr; in SplitHeightTest()213 SplitAttributes attr; in SplitBatchTest()255 SplitAttributes attr; in SplitDepthTest()
25 const SplitAttributes& attr, const std::vector<int>& channels) in Split()212 const SplitAttributes& attr, in CreateSplit()
98 void SelectSplit(const SplitAttributes& attr, const GpuInfo& gpu_info,
174 void SelectSplit(const SplitAttributes& attr, const GpuInfo& gpu_info, in SelectSplit()
670 auto attr = absl::any_cast<SplitAttributes>(node.operation.attributes); in GPUOperationFromNodePart0()
603 struct SplitAttributes { struct
692 SplitAttributes split_attr; in ResolveGroupedConvolution()2276 SplitAttributes attr; in Parse()2318 SplitAttributes attr; in Parse()2702 SplitAttributes attr; in Parse()