Lines Matching defs:inputShape
23 INetworkPtr CreateSplitterNetwork(const TensorShape& inputShape, in CreateSplitterNetwork()
83 const TensorShape& inputShape = { 4 }; in Splitter1dEndToEnd() local
111 const TensorShape& inputShape = { 4, 3 }; in Splitter2dDim0EndToEnd() local
146 const TensorShape& inputShape = { 4, 3 }; in Splitter2dDim1EndToEnd() local
184 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim0EndToEnd() local
232 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim1EndToEnd() local
280 const TensorShape& inputShape = { 2, 4, 3 }; in Splitter3dDim2EndToEnd() local
320 const TensorShape& inputShape = { 4, 3, 2, 2 }; in Splitter4dDim0EndToEnd() local
400 const TensorShape& inputShape = { 2, 6, 2, 2 }; in Splitter4dDim1EndToEnd() local
480 const TensorShape& inputShape = { 2, 3, 4, 2 }; in Splitter4dDim2EndToEnd() local
559 const TensorShape& inputShape = { 2, 3, 4, 2 }; in Splitter4dDim3EndToEnd() local