Method GenerateCode

tensorflow/lite/delegates/gpu/gl/kernels/mul.cc:54–87

  }

  Status GenerateCode(const GenerationContext& ctx,
                      GeneratedCode* generated_code) const final {
    if (!IsSupported(ctx)) {
      return InvalidArgumentError(
          "This case is not supported by apply mask operation");
    }
    const auto inputs = ctx.graph->FindInputs(ctx.node->id);
    const auto& shape0 = inputs[0]->tensor.shape;
    const auto& shape1 = inputs[1]->tensor.shape;

    std::string source = "value_0 = $input_data_0[gid.x, gid.y, gid.z]$ * ";
    if (shape1.c == 1) {
      // [H, W, C] x [H, W, 0][0]
      absl::StrAppend(&source, "$input_data_1[gid.x, gid.y, 0]$.x;");
    } else if (shape0.h == shape1.h && shape0.w == shape1.w) {
      // [H, W, C] x [H, W, C]
      absl::StrAppend(&source, "$input_data_1[gid.x, gid.y, gid.z]$;");
    } else {
      // [H, W, C] x [0, 0, C]
      absl::StrAppend(&source, "$input_data_1[0, 0, gid.z]$;");
    }

    *generated_code = {
        /*parameters=*/{},
        /*objects=*/{},
        /*shared_variables=*/{},
        /*workload=*/uint3(),
        /*workgroup=*/uint3(),
        /*source_code=*/std::move(source),
        /*input=*/IOStructure::ONLY_DEFINITIONS,
        /*output=*/IOStructure::AUTO,
    };
    return OkStatus();
  }
};

class MultiplyScalar : public NodeShader {