experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix formatting errors for zeros_like

Browse Source
This commit is contained in:
rsun 2021-02-03 12:11:34 -08:00
parent f8ed429b25
commit d57c0cbd62
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/micro/kernels/zeros_like_test.cc
View File

@ -25,8 +25,7 @@ namespace testing {
namespace { namespace {
void TestZerosLikeFloat(const int* input_dims_data, const float* input_data, void TestZerosLikeFloat(const int* input_dims_data, const float* input_data,
const float* expected_output_data, const float* expected_output_data, float* output_data) {
float* output_data) {
TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data);
TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data);
const int output_dims_count = ElementCount(*output_dims); const int output_dims_count = ElementCount(*output_dims);

4
tensorflow/lite/micro/micro_mutable_op_resolver.h
View File

@ -439,8 +439,8 @@ class MicroMutableOpResolver : public MicroOpResolver {
} }
TfLiteStatus AddZerosLike() { TfLiteStatus AddZerosLike() {
return AddBuiltin(BuiltinOperator_ZEROS_LIKE, return AddBuiltin(BuiltinOperator_ZEROS_LIKE, Register_ZEROS_LIKE(),
Register_ZEROS_LIKE(), ParseZerosLike); ParseZerosLike);
} }
unsigned int GetRegistrationLength() { return registrations_len_; } unsigned int GetRegistrationLength() { return registrations_len_; }