ml/tensor_view.cc

// Copyright 2018 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Implementations of specializations of TensorView<> for all supported tensor
// data types

#include "ml/tensor_view.h"

namespace ml {

using ::chromeos::machine_learning::mojom::FloatList;
using ::chromeos::machine_learning::mojom::Int64List;
using ::chromeos::machine_learning::mojom::ValueList;

template <>
std::vector<int64_t>& TensorView<int64_t>::GetValues() {
  return tensor_->data->get_int64_list()->value;
}

template <>
bool TensorView<int64_t>::IsValidType() const {
  return tensor_->data->which() == ValueList::Tag::kInt64List;
}

template <>
void TensorView<int64_t>::AllocateValues() {
  tensor_->data = ValueList::NewInt64List(Int64List::New());
}

template <>
std::vector<double>& TensorView<double>::GetValues() {
  return tensor_->data->get_float_list()->value;
}

template <>
bool TensorView<double>::IsValidType() const {
  return tensor_->data->which() == ValueList::Tag::kFloatList;
}

template <>
void TensorView<double>::AllocateValues() {
  tensor_->data = ValueList::NewFloatList(FloatList::New());
}

}  // namespace ml