1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_ML_WEBNN_ML_CONSTANT_OPERAND_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_ML_WEBNN_ML_CONSTANT_OPERAND_H_
#include "services/webnn/public/cpp/operand_descriptor.h"
#include "third_party/blink/public/common/tokens/tokens.h"
#include "third_party/blink/renderer/modules/ml/webnn/ml_operand.h"
#include "third_party/blink/renderer/modules/modules_export.h"
#include "third_party/blink/renderer/platform/heap/visitor.h"
namespace blink {
class MLGraphBuilder;
class MLTensor;
// Represents an `MLOperand` created from the `MLGraphBuilder.constant()`
// method. See https://www.w3.org/TR/webnn/#api-mlgraphbuilder-constant.
class MODULES_EXPORT MLConstantOperand final : public MLOperand {
public:
// Creates a constant operand on `builder`. After creating an instance of this
// class, the caller may create a "pending" constant operand in the WebNN
// Service with the generated `handle_`, which identifies the weight data
// associated with this operand.
MLConstantOperand(MLGraphBuilder* builder,
webnn::OperandDescriptor descriptor);
// Similar to above but uses a tensor for weight data.
MLConstantOperand(MLGraphBuilder* builder, MLTensor* tensor);
MLConstantOperand(const MLConstantOperand&) = delete;
MLConstantOperand& operator=(const MLConstantOperand&) = delete;
~MLConstantOperand() override;
void Trace(Visitor* visitor) const override;
const WebNNPendingConstantToken& handle() const { return handle_; }
const MLTensor* tensor() const { return tensor_; }
private:
// Identifies this constant operand in the WebNN service.
const WebNNPendingConstantToken handle_;
Member<MLTensor> tensor_;
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_MODULES_ML_WEBNN_ML_CONSTANT_OPERAND_H_
|
