Searched refs:GetOutputGroupSize (Results 1 – 4 of 4) sorted by relevance
41 int GetOutputGroupSize() const;
21 int WeightsDescription::GetOutputGroupSize() const { in GetOutputGroupSize() function in tflite::gpu::WeightsDescription
143 std::to_string(conv_weights_desc.GetOutputGroupSize()) + ";\n"; in GetConverterToConvWeightsCode()179 const int out_group_size = weights_desc_.GetOutputGroupSize(); in GetGridSize()
233 AlignByN(weights_shape.b, conv_weights_desc.GetOutputGroupSize() * 4); in GPUOperationFromNode()310 weights_shape.b, conv_weights_desc.GetOutputGroupSize() * 4); in GPUOperationFromNode()354 weights_desc.GetOutputGroupSize()); in GPUOperationFromNode()