Project import generated by Copybara.
GitOrigin-RevId: 19a829ffd755edb43e54d20c0e7b9348512d5108
This commit is contained in:
committed by
schmidt-sebastian
parent
c6c80c3745
commit
7fb37c80e8
@@ -187,6 +187,21 @@ cc_library(
|
||||
],
|
||||
)
|
||||
|
||||
config_setting(
|
||||
name = "opencv",
|
||||
define_values = {
|
||||
"use_opencv": "true",
|
||||
},
|
||||
)
|
||||
|
||||
config_setting(
|
||||
name = "portable_opencv",
|
||||
define_values = {
|
||||
"use_portable_opencv": "true",
|
||||
"use_opencv": "false",
|
||||
},
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "location",
|
||||
srcs = ["location.cc"],
|
||||
@@ -194,6 +209,8 @@ cc_library(
|
||||
defines = select({
|
||||
"//conditions:default": [],
|
||||
"//mediapipe:android": ["MEDIAPIPE_ANDROID_OPENCV"],
|
||||
":portable_opencv": ["MEDIAPIPE_ANDROID_OPENCV"],
|
||||
":opencv": [],
|
||||
}),
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
|
||||
@@ -76,7 +76,7 @@ class Tensor {
|
||||
|
||||
public:
|
||||
// No resources are allocated here.
|
||||
enum class ElementType { kNone, kFloat16, kFloat32, kUInt8 };
|
||||
enum class ElementType { kNone, kFloat16, kFloat32, kUInt8, kInt8 };
|
||||
struct Shape {
|
||||
Shape() = default;
|
||||
Shape(std::initializer_list<int> dimensions) : dims(dimensions) {}
|
||||
@@ -217,6 +217,8 @@ class Tensor {
|
||||
return sizeof(float);
|
||||
case ElementType::kUInt8:
|
||||
return 1;
|
||||
case ElementType::kInt8:
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
int bytes() const { return shape_.num_elements() * element_size(); }
|
||||
|
||||
Reference in New Issue
Block a user