Lines Matching defs:outputShapes
24 const std::vector<TensorShape>& outputShapes, in CreateSplitterNetwork()
84 const std::vector<TensorShape> outputShapes{{ 2 }, { 2 }}; in Splitter1dEndToEnd() local
112 const std::vector<TensorShape> outputShapes{{ 2, 3 }, { 2, 3 }}; in Splitter2dDim0EndToEnd() local
147 const std::vector<TensorShape> outputShapes{{ 4, 1 }, { 4, 1 }, { 4, 1 }}; in Splitter2dDim1EndToEnd() local
185 const std::vector<TensorShape> outputShapes{{ 1, 4, 3 }, { 1, 4, 3 }}; in Splitter3dDim0EndToEnd() local
233 const std::vector<TensorShape> outputShapes{{ 2, 2, 3 }, { 2, 2, 3 }}; in Splitter3dDim1EndToEnd() local
281 const std::vector<TensorShape> outputShapes{{ 2, 4, 1 }, { 2, 4, 1 }, { 2, 4, 1 }}; in Splitter3dDim2EndToEnd() local
321 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim0EndToEnd() local
401 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim1EndToEnd() local
481 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim2EndToEnd() local
560 const std::vector<TensorShape> outputShapes{{ 2, 3, 4, 1 }, { 2, 3, 4, 1 }}; in Splitter4dDim3EndToEnd() local