Searched refs:OutputShape (Results 1 – 13 of 13) sorted by relevance
45 .OutputShape(output_shape)66 .OutputShape(output_shape)87 .OutputShape(output_shape)108 .OutputShape(output_shape)128 .OutputShape(output_shape)148 .OutputShape(output_shape)166 .OutputShape(shape)184 .OutputShape(shape)196 .OutputShape(std::vector<int32_t>())219 .OutputShape(output_shape) in TEST()
111 reinterpret_cast<const uint8_t*>(OutputShape().data()), in CreateTfLiteModel()112 OutputShape().size() * sizeof(int32_t)))); in CreateTfLiteModel()121 builder.CreateVector<int32_t>(OutputShape().data(), in CreateTfLiteModel()122 OutputShape().size()), in CreateTfLiteModel()148 builder, builder.CreateVector<int32_t>(OutputShape().data(), in CreateTfLiteModel()149 OutputShape().size())) in CreateTfLiteModel()
45 inline ReshapeTester& OutputShape(const std::vector<int32_t>& output_shape) { in OutputShape() function55 inline const std::vector<int32_t>& OutputShape() const { in OutputShape() function
37 std::vector<int32_t> PadTester::OutputShape() const { in OutputShape() function in tflite::xnnpack::PadTester113 for (size_t i = 0; i < ComputeSize(OutputShape()); i++) { in Test()137 const std::vector<int32_t> output_shape = OutputShape(); in CreateTfLiteModel()
95 for (size_t i = 0; i < ComputeSize(OutputShape()); i++) { in Test()196 builder.CreateVector<int32_t>(OutputShape().data(), OutputShape().size()), in CreateTfLiteModel()
38 std::vector<int32_t> FullyConnectedTester::OutputShape() const { in OutputShape() function in tflite::xnnpack::FullyConnectedTester108 for (size_t i = 0; i < ComputeSize(OutputShape()); i++) { in Test()213 const std::vector<int32_t> output_shape = OutputShape(); in CreateTfLiteModel()
54 inline const std::vector<int32_t>& OutputShape() const { in OutputShape() function
70 std::vector<int32_t> OutputShape() const;
65 std::vector<int32_t> OutputShape() const;
38 std::vector<int32_t> BinaryElementwiseTester::OutputShape() const { in OutputShape() function in tflite::xnnpack::BinaryElementwiseTester163 for (size_t i = 0; i < ComputeSize(OutputShape()); i++) { in Test()256 const std::vector<int32_t> output_shape = OutputShape(); in CreateTfLiteModel()
61 std::vector<int32_t> OutputShape() const;
66 inline std::vector<int32_t> OutputShape() const { in OutputShape() function
117 const std::vector<int32_t> output_shape = OutputShape(); in CreateTfLiteModel()