Searched defs:splitAxis (Results 1 – 8 of 8) sorted by relevance
25 unsigned int splitAxis, in CreateSplitterNetwork()81 unsigned int splitAxis = 0; in Splitter1dEndToEnd() local109 unsigned int splitAxis = 0; in Splitter2dDim0EndToEnd() local144 unsigned int splitAxis = 1; in Splitter2dDim1EndToEnd() local182 unsigned int splitAxis = 0; in Splitter3dDim0EndToEnd() local230 unsigned int splitAxis = 1; in Splitter3dDim1EndToEnd() local278 unsigned int splitAxis = 2; in Splitter3dDim2EndToEnd() local318 unsigned int splitAxis = 0; in Splitter4dDim0EndToEnd() local398 unsigned int splitAxis = 1; in Splitter4dDim1EndToEnd() local478 unsigned int splitAxis = 2; in Splitter4dDim2EndToEnd() local[all …]
25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()34 unsigned int splitAxis) in NeonSplitterWorkloadValidate()91 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in NeonSplitterWorkload() local
25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()34 unsigned int splitAxis) in ClSplitterWorkloadValidate()91 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in ClSplitterWorkload() local
214 std::set<unsigned int> splitAxis; in ComputeSplitAxis() local
66 std::set<unsigned int> splitAxis; in CreateTensors() local
253 unsigned int splitAxis = 2; in BOOST_AUTO_TEST_CASE() local457 unsigned int splitAxis = 2; in BOOST_AUTO_TEST_CASE() local
899 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
902 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local