Add bool type scalar

Author: fujunwei

third_party/blink/renderer/modules/ml/neural_network_context.h

@@ -34,6 +34,7 @@ class NeuralNetworkContext final : public ScriptWrappable,
   static const unsigned long kTensorFloat32 = 3;
   static const unsigned long kTensorInt32 = 4;
   static const unsigned long kTensorQuant8Asymm = 5;
+  static const unsigned long kBool = 6;
   static const unsigned long kTensorQuant8SymmPerChannel = 11;
   static const unsigned long kTensorQuant8AsymmSigned = 14;
 

third_party/blink/renderer/modules/ml/v2/nn_model.cpp

@@ -169,6 +169,18 @@ void NNModel::AddScalarOperand(uint32_t index, float value) {
   SetOperandValue(index, data.View());
 }
 
+void NNModel::AddScalarOperand(uint32_t index, bool value) {
+  model_info_->operands.push_back(ml::mojom::blink::Operand::New(
+      static_cast<int32_t>(NeuralNetworkContext::kBool),
+      WTF::Vector<uint32_t>(), 0, 0));
+
+  // setOperandValue
+  int8_t convert = value ? 1 : 0;
+  NotShared<DOMArrayBufferView> data =
+      NotShared<DOMArrayBufferView>(DOMInt8Array::Create(&convert, 1));
+  SetOperandValue(index, data.View());
+}
+
 // Types prefaced with ANEURALNETWORKS_TENSOR_* must be used for tensor data
 // (i.e., tensors with at least one dimension).
 void NNModel::AddBiasOperand(uint32_t index, uint32_t output_channel) {

third_party/blink/renderer/modules/ml/v2/nn_model.h

@@ -33,6 +33,7 @@ class NNModel final : public ScriptWrappable {
   void BuildNeuralNetworkModel(Operand*);
   void AddScalarOperand(uint32_t index, int value);
   void AddScalarOperand(uint32_t index, float value);
+  void AddScalarOperand(uint32_t index, bool value);
   void AddBiasOperand(uint32_t index, uint32_t output_channel);
   void AddTensorOperand(uint32_t index,
                         Vector<uint32_t> dimensions,

third_party/blink/renderer/modules/ml/v2/ops/conv.cc

@@ -96,7 +96,7 @@ void Conv::AddLayer(NNModel* model, uint32_t& index) {
 
   // Add layout operand and set the value.
   uint32_t layout_index = index++;
-  model->AddScalarOperand(layout_index, layout_ == "nchw" ? 1 : 0);
+  model->AddScalarOperand(layout_index, layout_ == "nchw" ? true : false);
   input_indexes.push_back(layout_index);
 
   // Add conv output operand.