Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
xxadev
tensorflow
提交
2b8f5924
T
tensorflow
项目概览
xxadev
/
tensorflow
与 Fork 源项目一致
从无法访问的项目Fork
通知
3
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
2b8f5924
编写于
10月 10, 2018
作者:
A
A. Unique TensorFlower
提交者:
TensorFlower Gardener
10月 10, 2018
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add Range to schema.
PiperOrigin-RevId: 216638239
上级
905deeaa
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
126 addition
and
6 deletion
+126
-6
tensorflow/contrib/lite/builtin_ops.h
tensorflow/contrib/lite/builtin_ops.h
+1
-0
tensorflow/contrib/lite/core/api/flatbuffer_conversions.cc
tensorflow/contrib/lite/core/api/flatbuffer_conversions.cc
+1
-0
tensorflow/contrib/lite/nnapi_delegate.cc
tensorflow/contrib/lite/nnapi_delegate.cc
+1
-0
tensorflow/contrib/lite/schema/schema.fbs
tensorflow/contrib/lite/schema/schema.fbs
+5
-0
tensorflow/contrib/lite/schema/schema_generated.h
tensorflow/contrib/lite/schema/schema_generated.h
+118
-6
未找到文件。
tensorflow/contrib/lite/builtin_ops.h
浏览文件 @
2b8f5924
...
...
@@ -121,6 +121,7 @@ typedef enum {
kTfLiteBuiltinZerosLike
=
93
,
kTfLiteBuiltinFill
=
94
,
kTfLiteBuiltinFloorMod
=
95
,
kTfLiteBuiltinRange
=
96
,
}
TfLiteBuiltinOperator
;
#ifdef __cplusplus
...
...
tensorflow/contrib/lite/core/api/flatbuffer_conversions.cc
浏览文件 @
2b8f5924
...
...
@@ -652,6 +652,7 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type,
case
BuiltinOperator_ZEROS_LIKE
:
case
BuiltinOperator_FILL
:
case
BuiltinOperator_FLOOR_MOD
:
case
BuiltinOperator_RANGE
:
break
;
}
return
kTfLiteOk
;
...
...
tensorflow/contrib/lite/nnapi_delegate.cc
浏览文件 @
2b8f5924
...
...
@@ -680,6 +680,7 @@ TfLiteStatus AddOpsAndParams(
case
tflite
::
BuiltinOperator_ZEROS_LIKE
:
case
tflite
::
BuiltinOperator_FILL
:
case
tflite
::
BuiltinOperator_FLOOR_MOD
:
case
tflite
::
BuiltinOperator_RANGE
:
logError
(
"Op code %d is currently not delegated to NNAPI"
,
builtin
);
return
kTfLiteError
;
break
;
...
...
tensorflow/contrib/lite/schema/schema.fbs
浏览文件 @
2b8f5924
...
...
@@ -177,6 +177,7 @@ enum BuiltinOperator : byte {
ZEROS_LIKE = 93,
FILL = 94,
FLOOR_MOD = 95,
RANGE = 96,
}
// Options for the builtin operators.
...
...
@@ -253,6 +254,7 @@ union BuiltinOptions {
BidirectionalSequenceRNNOptions,
UnidirectionalSequenceLSTMOptions,
FloorModOptions,
RangeOptions,
}
enum Padding : byte { SAME, VALID }
...
...
@@ -623,6 +625,9 @@ table FillOptions {
table FloorModOptions {
}
table RangeOptions {
}
// An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
// builtin, or a string if the operator is custom.
table OperatorCode {
...
...
tensorflow/contrib/lite/schema/schema_generated.h
浏览文件 @
2b8f5924
...
...
@@ -244,6 +244,9 @@ struct FillOptionsT;
struct
FloorModOptions
;
struct
FloorModOptionsT
;
struct
RangeOptions
;
struct
RangeOptionsT
;
struct
OperatorCode
;
struct
OperatorCodeT
;
...
...
@@ -405,11 +408,12 @@ enum BuiltinOperator {
BuiltinOperator_ZEROS_LIKE
=
93
,
BuiltinOperator_FILL
=
94
,
BuiltinOperator_FLOOR_MOD
=
95
,
BuiltinOperator_RANGE
=
96
,
BuiltinOperator_MIN
=
BuiltinOperator_ADD
,
BuiltinOperator_MAX
=
BuiltinOperator_
FLOOR_MOD
BuiltinOperator_MAX
=
BuiltinOperator_
RANGE
};
inline
const
BuiltinOperator
(
&
EnumValuesBuiltinOperator
())[
9
5
]
{
inline
const
BuiltinOperator
(
&
EnumValuesBuiltinOperator
())[
9
6
]
{
static
const
BuiltinOperator
values
[]
=
{
BuiltinOperator_ADD
,
BuiltinOperator_AVERAGE_POOL_2D
,
...
...
@@ -505,7 +509,8 @@ inline const BuiltinOperator (&EnumValuesBuiltinOperator())[95] {
BuiltinOperator_SQUARE
,
BuiltinOperator_ZEROS_LIKE
,
BuiltinOperator_FILL
,
BuiltinOperator_FLOOR_MOD
BuiltinOperator_FLOOR_MOD
,
BuiltinOperator_RANGE
};
return
values
;
}
...
...
@@ -608,6 +613,7 @@ inline const char * const *EnumNamesBuiltinOperator() {
"ZEROS_LIKE"
,
"FILL"
,
"FLOOR_MOD"
,
"RANGE"
,
nullptr
};
return
names
;
...
...
@@ -692,11 +698,12 @@ enum BuiltinOptions {
BuiltinOptions_BidirectionalSequenceRNNOptions
=
70
,
BuiltinOptions_UnidirectionalSequenceLSTMOptions
=
71
,
BuiltinOptions_FloorModOptions
=
72
,
BuiltinOptions_RangeOptions
=
73
,
BuiltinOptions_MIN
=
BuiltinOptions_NONE
,
BuiltinOptions_MAX
=
BuiltinOptions_
FloorMod
Options
BuiltinOptions_MAX
=
BuiltinOptions_
Range
Options
};
inline
const
BuiltinOptions
(
&
EnumValuesBuiltinOptions
())[
7
3
]
{
inline
const
BuiltinOptions
(
&
EnumValuesBuiltinOptions
())[
7
4
]
{
static
const
BuiltinOptions
values
[]
=
{
BuiltinOptions_NONE
,
BuiltinOptions_Conv2DOptions
,
...
...
@@ -770,7 +777,8 @@ inline const BuiltinOptions (&EnumValuesBuiltinOptions())[73] {
BuiltinOptions_BidirectionalSequenceLSTMOptions
,
BuiltinOptions_BidirectionalSequenceRNNOptions
,
BuiltinOptions_UnidirectionalSequenceLSTMOptions
,
BuiltinOptions_FloorModOptions
BuiltinOptions_FloorModOptions
,
BuiltinOptions_RangeOptions
};
return
values
;
}
...
...
@@ -850,6 +858,7 @@ inline const char * const *EnumNamesBuiltinOptions() {
"BidirectionalSequenceRNNOptions"
,
"UnidirectionalSequenceLSTMOptions"
,
"FloorModOptions"
,
"RangeOptions"
,
nullptr
};
return
names
;
...
...
@@ -1152,6 +1161,10 @@ template<> struct BuiltinOptionsTraits<FloorModOptions> {
static
const
BuiltinOptions
enum_value
=
BuiltinOptions_FloorModOptions
;
};
template
<
>
struct
BuiltinOptionsTraits
<
RangeOptions
>
{
static
const
BuiltinOptions
enum_value
=
BuiltinOptions_RangeOptions
;
};
struct
BuiltinOptionsUnion
{
BuiltinOptions
type
;
void
*
value
;
...
...
@@ -1759,6 +1772,14 @@ struct BuiltinOptionsUnion {
return
type
==
BuiltinOptions_FloorModOptions
?
reinterpret_cast
<
const
FloorModOptionsT
*>
(
value
)
:
nullptr
;
}
RangeOptionsT
*
AsRangeOptions
()
{
return
type
==
BuiltinOptions_RangeOptions
?
reinterpret_cast
<
RangeOptionsT
*>
(
value
)
:
nullptr
;
}
const
RangeOptionsT
*
AsRangeOptions
()
const
{
return
type
==
BuiltinOptions_RangeOptions
?
reinterpret_cast
<
const
RangeOptionsT
*>
(
value
)
:
nullptr
;
}
};
bool
VerifyBuiltinOptions
(
flatbuffers
::
Verifier
&
verifier
,
const
void
*
obj
,
BuiltinOptions
type
);
...
...
@@ -6302,6 +6323,46 @@ inline flatbuffers::Offset<FloorModOptions> CreateFloorModOptions(
flatbuffers
::
Offset
<
FloorModOptions
>
CreateFloorModOptions
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
,
const
FloorModOptionsT
*
_o
,
const
flatbuffers
::
rehasher_function_t
*
_rehasher
=
nullptr
);
struct
RangeOptionsT
:
public
flatbuffers
::
NativeTable
{
typedef
RangeOptions
TableType
;
RangeOptionsT
()
{
}
};
struct
RangeOptions
FLATBUFFERS_FINAL_CLASS
:
private
flatbuffers
::
Table
{
typedef
RangeOptionsT
NativeTableType
;
bool
Verify
(
flatbuffers
::
Verifier
&
verifier
)
const
{
return
VerifyTableStart
(
verifier
)
&&
verifier
.
EndTable
();
}
RangeOptionsT
*
UnPack
(
const
flatbuffers
::
resolver_function_t
*
_resolver
=
nullptr
)
const
;
void
UnPackTo
(
RangeOptionsT
*
_o
,
const
flatbuffers
::
resolver_function_t
*
_resolver
=
nullptr
)
const
;
static
flatbuffers
::
Offset
<
RangeOptions
>
Pack
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
,
const
RangeOptionsT
*
_o
,
const
flatbuffers
::
rehasher_function_t
*
_rehasher
=
nullptr
);
};
struct
RangeOptionsBuilder
{
flatbuffers
::
FlatBufferBuilder
&
fbb_
;
flatbuffers
::
uoffset_t
start_
;
explicit
RangeOptionsBuilder
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
)
:
fbb_
(
_fbb
)
{
start_
=
fbb_
.
StartTable
();
}
RangeOptionsBuilder
&
operator
=
(
const
RangeOptionsBuilder
&
);
flatbuffers
::
Offset
<
RangeOptions
>
Finish
()
{
const
auto
end
=
fbb_
.
EndTable
(
start_
);
auto
o
=
flatbuffers
::
Offset
<
RangeOptions
>
(
end
);
return
o
;
}
};
inline
flatbuffers
::
Offset
<
RangeOptions
>
CreateRangeOptions
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
)
{
RangeOptionsBuilder
builder_
(
_fbb
);
return
builder_
.
Finish
();
}
flatbuffers
::
Offset
<
RangeOptions
>
CreateRangeOptions
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
,
const
RangeOptionsT
*
_o
,
const
flatbuffers
::
rehasher_function_t
*
_rehasher
=
nullptr
);
struct
OperatorCodeT
:
public
flatbuffers
::
NativeTable
{
typedef
OperatorCode
TableType
;
BuiltinOperator
builtin_code
;
...
...
@@ -6651,6 +6712,9 @@ struct Operator FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
const
FloorModOptions
*
builtin_options_as_FloorModOptions
()
const
{
return
builtin_options_type
()
==
BuiltinOptions_FloorModOptions
?
static_cast
<
const
FloorModOptions
*>
(
builtin_options
())
:
nullptr
;
}
const
RangeOptions
*
builtin_options_as_RangeOptions
()
const
{
return
builtin_options_type
()
==
BuiltinOptions_RangeOptions
?
static_cast
<
const
RangeOptions
*>
(
builtin_options
())
:
nullptr
;
}
const
flatbuffers
::
Vector
<
uint8_t
>
*
custom_options
()
const
{
return
GetPointer
<
const
flatbuffers
::
Vector
<
uint8_t
>
*>
(
VT_CUSTOM_OPTIONS
);
}
...
...
@@ -6970,6 +7034,10 @@ template<> inline const FloorModOptions *Operator::builtin_options_as<FloorModOp
return
builtin_options_as_FloorModOptions
();
}
template
<
>
inline
const
RangeOptions
*
Operator
::
builtin_options_as
<
RangeOptions
>
()
const
{
return
builtin_options_as_RangeOptions
();
}
struct
OperatorBuilder
{
flatbuffers
::
FlatBufferBuilder
&
fbb_
;
flatbuffers
::
uoffset_t
start_
;
...
...
@@ -9377,6 +9445,29 @@ inline flatbuffers::Offset<FloorModOptions> CreateFloorModOptions(flatbuffers::F
_fbb
);
}
inline
RangeOptionsT
*
RangeOptions
::
UnPack
(
const
flatbuffers
::
resolver_function_t
*
_resolver
)
const
{
auto
_o
=
new
RangeOptionsT
();
UnPackTo
(
_o
,
_resolver
);
return
_o
;
}
inline
void
RangeOptions
::
UnPackTo
(
RangeOptionsT
*
_o
,
const
flatbuffers
::
resolver_function_t
*
_resolver
)
const
{
(
void
)
_o
;
(
void
)
_resolver
;
}
inline
flatbuffers
::
Offset
<
RangeOptions
>
RangeOptions
::
Pack
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
,
const
RangeOptionsT
*
_o
,
const
flatbuffers
::
rehasher_function_t
*
_rehasher
)
{
return
CreateRangeOptions
(
_fbb
,
_o
,
_rehasher
);
}
inline
flatbuffers
::
Offset
<
RangeOptions
>
CreateRangeOptions
(
flatbuffers
::
FlatBufferBuilder
&
_fbb
,
const
RangeOptionsT
*
_o
,
const
flatbuffers
::
rehasher_function_t
*
_rehasher
)
{
(
void
)
_rehasher
;
(
void
)
_o
;
struct
_VectorArgs
{
flatbuffers
::
FlatBufferBuilder
*
__fbb
;
const
RangeOptionsT
*
__o
;
const
flatbuffers
::
rehasher_function_t
*
__rehasher
;
}
_va
=
{
&
_fbb
,
_o
,
_rehasher
};
(
void
)
_va
;
return
tflite
::
CreateRangeOptions
(
_fbb
);
}
inline
OperatorCodeT
*
OperatorCode
::
UnPack
(
const
flatbuffers
::
resolver_function_t
*
_resolver
)
const
{
auto
_o
=
new
OperatorCodeT
();
UnPackTo
(
_o
,
_resolver
);
...
...
@@ -9854,6 +9945,10 @@ inline bool VerifyBuiltinOptions(flatbuffers::Verifier &verifier, const void *ob
auto
ptr
=
reinterpret_cast
<
const
FloorModOptions
*>
(
obj
);
return
verifier
.
VerifyTable
(
ptr
);
}
case
BuiltinOptions_RangeOptions
:
{
auto
ptr
=
reinterpret_cast
<
const
RangeOptions
*>
(
obj
);
return
verifier
.
VerifyTable
(
ptr
);
}
default:
return
false
;
}
}
...
...
@@ -10160,6 +10255,10 @@ inline void *BuiltinOptionsUnion::UnPack(const void *obj, BuiltinOptions type, c
auto
ptr
=
reinterpret_cast
<
const
FloorModOptions
*>
(
obj
);
return
ptr
->
UnPack
(
resolver
);
}
case
BuiltinOptions_RangeOptions
:
{
auto
ptr
=
reinterpret_cast
<
const
RangeOptions
*>
(
obj
);
return
ptr
->
UnPack
(
resolver
);
}
default:
return
nullptr
;
}
}
...
...
@@ -10454,6 +10553,10 @@ inline flatbuffers::Offset<void> BuiltinOptionsUnion::Pack(flatbuffers::FlatBuff
auto
ptr
=
reinterpret_cast
<
const
FloorModOptionsT
*>
(
value
);
return
CreateFloorModOptions
(
_fbb
,
ptr
,
_rehasher
).
Union
();
}
case
BuiltinOptions_RangeOptions
:
{
auto
ptr
=
reinterpret_cast
<
const
RangeOptionsT
*>
(
value
);
return
CreateRangeOptions
(
_fbb
,
ptr
,
_rehasher
).
Union
();
}
default:
return
0
;
}
}
...
...
@@ -10748,6 +10851,10 @@ inline BuiltinOptionsUnion::BuiltinOptionsUnion(const BuiltinOptionsUnion &u) FL
value
=
new
FloorModOptionsT
(
*
reinterpret_cast
<
FloorModOptionsT
*>
(
u
.
value
));
break
;
}
case
BuiltinOptions_RangeOptions
:
{
value
=
new
RangeOptionsT
(
*
reinterpret_cast
<
RangeOptionsT
*>
(
u
.
value
));
break
;
}
default:
break
;
}
...
...
@@ -11115,6 +11222,11 @@ inline void BuiltinOptionsUnion::Reset() {
delete
ptr
;
break
;
}
case
BuiltinOptions_RangeOptions
:
{
auto
ptr
=
reinterpret_cast
<
RangeOptionsT
*>
(
value
);
delete
ptr
;
break
;
}
default:
break
;
}
value
=
nullptr
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录