Project import generated by Copybara.

GitOrigin-RevId: c27bb068d24890a4e3b7d2d536267e08d7e4abbd
This commit is contained in:
MediaPipe Team 2020-08-31 13:04:30 -07:00 committed by jqtang
parent c0124fb83c
commit 1db91b550a
5 changed files with 24 additions and 65 deletions

View File

@ -362,13 +362,12 @@ http_archive(
) )
#Tensorflow repo should always go after the other external dependencies. #Tensorflow repo should always go after the other external dependencies.
# 2020-05-11 # 2020-08-30
_TENSORFLOW_GIT_COMMIT = "7c09d15f9fcc14343343c247ebf5b8e0afe3e4aa" _TENSORFLOW_GIT_COMMIT = "57b009e31e59bd1a7ae85ef8c0232ed86c9b71db"
_TENSORFLOW_SHA256= "673d00cbd2676ae43df1993e0d28c10b5ffbe96d9e2ab29f88a77b43c0211299" _TENSORFLOW_SHA256= "de7f5f06204e057383028c7e53f3b352cdf85b3a40981b1a770c9a415a792c0e"
http_archive( http_archive(
name = "org_tensorflow", name = "org_tensorflow",
urls = [ urls = [
"https://mirror.bazel.build/github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT,
"https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT, "https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT,
], ],
patches = [ patches = [

View File

@ -51,7 +51,7 @@ namespace {
void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) { void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) {
for (auto& node : *graph_def->mutable_node()) { for (auto& node : *graph_def->mutable_node()) {
if (node.device().empty()) { if (node.device().empty()) {
node.set_device(device_id); node.set_device(std::string(device_id));
} }
} }
} }

View File

@ -47,7 +47,7 @@ namespace {
void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) { void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) {
for (auto& node : *graph_def->mutable_node()) { for (auto& node : *graph_def->mutable_node()) {
if (node.device().empty()) { if (node.device().empty()) {
node.set_device(device_id); node.set_device(std::string(device_id));
} }
} }
} }

View File

@ -56,4 +56,11 @@ message UnpackMediaSequenceCalculatorOptions {
// the clip start and end times and outputs these for the // the clip start and end times and outputs these for the
// AudioDecoderCalculator to consume. // AudioDecoderCalculator to consume.
optional AudioDecoderOptions base_audio_decoder_options = 9; optional AudioDecoderOptions base_audio_decoder_options = 9;
optional string keypoint_names = 10 [
default =
"NOSE,LEFT_EAR,RIGHT_EAR,LEFT_SHOULDER,RIGHT_SHOULDER,LEFT_FORE_PAW,RIGHT_FORE_PAW,LEFT_HIP,RIGHT_HIP,LEFT_HIND_PAW,RIGHT_HIND_PAW,ROOT_TAIL"
];
// When the keypoint doesn't exists, output this default value.
optional float default_keypoint_location = 11 [default = -1.0];
} }

View File

@ -23,62 +23,15 @@ index b7c22ae77b..d0ba7b48b4 100644
pct_samples.points.push_back(pct); pct_samples.points.push_back(pct);
} }
} }
diff --git a/tensorflow/lite/delegates/gpu/cl/api.cc b/tensorflow/lite/delegates/gpu/cl/api.cc diff --git a/tensorflow/core/platform/test.h b/tensorflow/core/platform/test.h
index 09c82307a5..0318c1a447 100644 index ba50783765..5de5ea01f0 100644
--- a/tensorflow/lite/delegates/gpu/cl/api.cc --- a/tensorflow/core/platform/test.h
+++ b/tensorflow/lite/delegates/gpu/cl/api.cc +++ b/tensorflow/core/platform/test.h
@@ -352,10 +352,10 @@ class GlBufferHolder : public TensorTie { @@ -42,7 +42,6 @@ limitations under the License.
}; #if defined(PLATFORM_GOOGLE) || defined(PLATFORM_GOOGLE_ANDROID)
#include "testing/base/public/gmock.h"
TensorObject TensorToObj(const Tensor& tensor) { #else
- if (tensor.StorageType() == TensorStorageType::BUFFER) { -#include <gmock/gmock-generated-matchers.h>
+ if (tensor.GetStorageType() == TensorStorageType::BUFFER) { #include <gmock/gmock-matchers.h>
return OpenClBuffer{tensor.GetMemoryPtr()}; #include <gmock/gmock-more-matchers.h>
} #endif
- if (tensor.StorageType() == TensorStorageType::IMAGE_BUFFER) {
+ if (tensor.GetStorageType() == TensorStorageType::IMAGE_BUFFER) {
return OpenClBuffer{tensor.GetMemoryPtrForWriting()};
}
return OpenClTexture{tensor.GetMemoryPtr()};
@@ -516,9 +516,9 @@ TensorObjectDef TensorToDef(const Tensor& tensor) {
def.dimensions.h = tensor.Height();
def.dimensions.w = tensor.Width();
def.dimensions.c = tensor.Channels();
- def.object_def.data_layout = ToDataLayout(tensor.StorageType());
- def.object_def.data_type = tensor.DataType();
- def.object_def.object_type = ToObjectType(tensor.StorageType());
+ def.object_def.data_layout = ToDataLayout(tensor.GetStorageType());
+ def.object_def.data_type = tensor.GetDataType();
+ def.object_def.object_type = ToObjectType(tensor.GetStorageType());
def.object_def.user_provided = false;
return def;
}
diff --git a/tensorflow/lite/delegates/gpu/cl/tensor.cc b/tensorflow/lite/delegates/gpu/cl/tensor.cc
index 308e1b6920..9b613c63f7 100644
--- a/tensorflow/lite/delegates/gpu/cl/tensor.cc
+++ b/tensorflow/lite/delegates/gpu/cl/tensor.cc
@@ -29,7 +29,7 @@ namespace cl {
namespace {
absl::Status CreateImageBufferFromBuffer(const CLContext& context,
- cl_mem memory, enum DataType data_type,
+ cl_mem memory, DataType data_type,
int width, cl_mem* result) {
cl_image_format format;
cl_image_desc desc;
diff --git a/tensorflow/lite/delegates/gpu/cl/tensor.h b/tensorflow/lite/delegates/gpu/cl/tensor.h
index a27c54a74e..39d3a04a47 100644
--- a/tensorflow/lite/delegates/gpu/cl/tensor.h
+++ b/tensorflow/lite/delegates/gpu/cl/tensor.h
@@ -75,8 +75,8 @@ class Tensor {
int4 GetWHSB() const { return int4(shape_.w, shape_.h, Slices(), shape_.b); }
int4 GetWHDS() const { return int4(shape_.w, shape_.h, shape_.d, Slices()); }
- enum DataType DataType() const { return descriptor_.data_type; }
- TensorStorageType StorageType() const { return descriptor_.storage_type; }
+ DataType GetDataType() const { return descriptor_.data_type; }
+ TensorStorageType GetStorageType() const { return descriptor_.storage_type; }
// for profiling and memory statistics
uint64_t GetMemorySizeInBytes() const;