ONE - On-device Neural Engine
|
#include <BroadcastTo.h>
Public Member Functions | |
BroadcastToChef (const tflchef::Operation *operation) | |
tflite::BuiltinOperator | code (void) const override |
tflite::BuiltinOptions | type (void) const override |
flatbuffers::Offset< void > | value (flatbuffers::FlatBufferBuilder &fbb) const override |
flatbuffers::Offset< flatbuffers::Vector< uint8_t > > | custom_value (flatbuffers::FlatBufferBuilder &fbb) const override |
BroadcastToChef (const tflchef::Operation *operation) | |
tflite::BuiltinOperator | code (void) const override |
tflite::BuiltinOptions | type (void) const override |
flatbuffers::Offset< void > | value (flatbuffers::FlatBufferBuilder &fbb) const override |
Public Member Functions inherited from OpChef | |
virtual | ~OpChef ()=default |
virtual | ~OpChef ()=default |
Definition at line 22 of file BroadcastTo.h.
|
inlineexplicit |
Definition at line 25 of file BroadcastTo.h.
|
inlineexplicit |
Definition at line 25 of file BroadcastTo.h.
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 31 of file BroadcastTo.h.
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 31 of file BroadcastTo.h.
|
overridevirtual |
REGISTER_OP("BroadcastTo") .Input("input: T") .Input("shape: Tidx") .Output("output: T") .Attr("T: type") .Attr("Tidx: {int32, int64} = DT_INT32") .SetShapeFn([](InferenceContext* c)
Reimplemented from OpChef.
Definition at line 29 of file BroadcastTo.cpp.
References check_custom_op_value(), and flatbuffers::FlatBufferBuilder::CreateVector().
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 33 of file BroadcastTo.h.
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 33 of file BroadcastTo.h.
|
overridevirtual |
Implements OpChef.
Definition at line 23 of file BroadcastTo.cpp.
|
overridevirtual |
Implements OpChef.