Lines Matching refs:OutputShape

55                                            const std::vector<OutputShape>& shapes) {  in validateOutputShapesFromDriver()
70 [](const OutputShape& shape) { return shape.isSufficient; })) in validateOutputShapesFromDriver()
92 [](const OutputShape& shape) { return !shape.isSufficient; })) in validateOutputShapesFromDriver()
107 const std::vector<OutputShape>& shapes) { in validateOutputShapesFromDriver()
603 static std::tuple<int, std::vector<OutputShape>, Timing> cpuFallbackFull( in cpuFallbackFull()
624 static std::tuple<int, std::vector<OutputShape>, Timing, std::shared_ptr<StepExecutor>>
643 std::tuple<int, std::vector<OutputShape>, Timing> SimpleExecutionBuilder::computeInternal( in computeInternal()
679 std::tuple<int, std::vector<OutputShape>, Timing> CompoundExecutionBuilder::computeInternal( in computeInternal()
685 std::vector<OutputShape> outputShapes = getInitialOutputShapes(); in computeInternal()
1069 [this](ErrorStatus error, const std::vector<OutputShape>& outputShapes) { in compute()
1090 std::vector<OutputShape> ExecutionBuilder::getInitialOutputShapes() const { in getInitialOutputShapes()
1091 std::vector<OutputShape> outputShapes(mOutputs.size()); in getInitialOutputShapes()
1093 [](const auto& x) -> OutputShape { in getInitialOutputShapes()
1114 static bool isZeroSizedTensor(int executionResultCode, const OutputShape& outputShape) { in isZeroSizedTensor()
1122 const std::vector<OutputShape>& outputShapes) { in updateOutputShapes()
1152 int ExecutionBuilder::finishComputation(int result, const std::vector<OutputShape>& outputShapes) { in finishComputation()
1191 bool StepExecutor::updateOutputShapes(int executionResultCode, const std::vector<OutputShape>& from, in updateOutputShapes()
1192 std::vector<OutputShape>* to, UpdateOutputShapes* update) { in updateOutputShapes()
1460 std::tuple<int, std::vector<OutputShape>, Timing> StepExecutor::compute( in compute()
1468 std::vector<OutputShape> outputShapes; in compute()
1529 std::tuple<int, std::vector<OutputShape>, Timing> StepExecutor::computeOnCpuFallback() { in computeOnCpuFallback()