Home
last modified time | relevance | path

Searched refs:SingleOpModel (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc49 int SingleOpModel::AddInput(const TensorData& t, bool is_variable) { in AddInput()
60 int SingleOpModel::AddNullInput() { in AddNullInput()
66 int SingleOpModel::AddOutput(const TensorData& t) { in AddOutput()
72 void SingleOpModel::SetBuiltinOp(BuiltinOperator type, in SetBuiltinOp()
83 void SingleOpModel::SetCustomOp( in SetCustomOp()
96 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
147 void SingleOpModel::Invoke() { CHECK(interpreter_->Invoke() == kTfLiteOk); } in Invoke()
149 int32_t SingleOpModel::GetTensorSize(int index) const { in GetTensorSize()
160 std::vector<string> SingleOpModel::ExtractVector(int index) { in ExtractVector()
Dtest_util.h141 class SingleOpModel {
143 SingleOpModel() {} in SingleOpModel() function
144 ~SingleOpModel() {} in ~SingleOpModel()
154 SingleOpModel(const SingleOpModel&) = delete;
155 SingleOpModel& operator=(const SingleOpModel&) = delete;
524 std::vector<string> SingleOpModel::ExtractVector(int index);
Dexp_test.cc26 class ExpOpModel : public SingleOpModel {
Dceil_test.cc27 class CeilOpModel : public SingleOpModel {
Dfloor_test.cc27 class FloorOpModel : public SingleOpModel {
Ddequantize_test.cc29 class DequantizeOpModel : public SingleOpModel {
Dlocal_response_norm_test.cc26 class LocalResponseNormOpModel : public SingleOpModel {
Dneg_test.cc26 class NegOpModel : public SingleOpModel {
Dzeros_like_test.cc26 class ZerosLikeOpModel : public SingleOpModel {
Drank_test.cc29 class RankOpModel : public SingleOpModel {
Dmfcc_test.cc38 class BaseMfccOpModel : public SingleOpModel {
Dexpand_dims_test.cc28 class ExpandDimsOpModel : public SingleOpModel {
Dlog_softmax_test.cc32 class LogSoftmaxOpModel : public SingleOpModel {
Dshape_test.cc30 class ShapeOpModel : public SingleOpModel {
Dadd_n_test.cc28 class BaseAddNOpModel : public SingleOpModel {
Dfloor_div_test.cc27 class FloorDivModel : public SingleOpModel {
Dlsh_projection_test.cc29 class LSHProjectionOpModel : public SingleOpModel {
Daudio_spectrogram_test.cc38 class BaseAudioSpectrogramOpModel : public SingleOpModel {
Dunique_test.cc27 class UniqueOpModel : public SingleOpModel {
Dsqueeze_test.cc27 class BaseSqueezeOpModel : public SingleOpModel {
Dfake_quant_test.cc26 class FakeQuantOpModel : public SingleOpModel {
Dspace_to_depth_test.cc27 class SpaceToDepthOpModel : public SingleOpModel {
/external/libtextclassifier/utils/tflite/
Ddist_diversification_test.cc28 class DistanceDiversificationOpModel : public tflite::SingleOpModel {
/external/tensorflow/tensorflow/lite/models/smartreply/ops/
Dnormalize_test.cc35 class NormalizeOpModel : public SingleOpModel {
Dextract_feature_test.cc35 class ExtractFeatureOpModel : public SingleOpModel {

1234