Home
last modified time | relevance | path

Searched refs:weights_format (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc141 FullyConnectedOptionsWeightsFormat weights_format = argument
168 if (weights_format != FullyConnectedOptionsWeightsFormat_DEFAULT) {
174 CreateFullyConnectedOptions(builder_, activation_func, weights_format)
550 int batches, FullyConnectedOptionsWeightsFormat weights_format) { in SimpleTestQuantizedInt16OutputCase() argument
568 /*activation_func=*/ActivationFunctionType_NONE, weights_format); in SimpleTestQuantizedInt16OutputCase()
581 switch (weights_format) { in SimpleTestQuantizedInt16OutputCase()
Dfully_connected.cc92 params->weights_format == kTfLiteFullyConnectedWeightsFormatDefault ? 1 in Prepare()
478 if (params->weights_format == in Eval()
485 } else if (params->weights_format == in Eval()
495 if (params->weights_format == kTfLiteFullyConnectedWeightsFormatDefault) { in Eval()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dshuffle_fc_weights.cc36 if (fc_op->weights_format != FullyConnectedWeightsFormat::kDefault) { in Run()
140 fc_op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in Run()
Densure_uint8_weights_safe_for_fast_int8_kernels.cc126 CHECK(fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) in Run()
Didentify_lstm.cc247 if (static_cast<FullyConnectedOperator*>(fully_connected)->weights_format != in Run()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc287 switch (fully_connected_params->weights_format()) { in ParseOpData()
289 params->weights_format = kTfLiteFullyConnectedWeightsFormatDefault; in ParseOpData()
292 params->weights_format = in ParseOpData()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h133 TfLiteFullyConnectedWeightsFormat weights_format; member
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc445 switch (op.weights_format) { in WriteOptions()
467 switch (options.weights_format()) { in ReadOptions()
469 op->weights_format = FullyConnectedWeightsFormat::kDefault; in ReadOptions()
472 op->weights_format = FullyConnectedWeightsFormat::kShuffled4x16Int8; in ReadOptions()
476 op->weights_format = FullyConnectedWeightsFormat::kDefault; in ReadOptions()
512 if (fc_op.weights_format == in GetVersion()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h827 FullyConnectedWeightsFormat weights_format; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h851 FullyConnectedWeightsFormat weights_format; member
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs413 weights_format:FullyConnectedOptionsWeightsFormat = DEFAULT;
Dschema_generated.h3533 FullyConnectedOptionsWeightsFormat weights_format;
3536 weights_format(FullyConnectedOptionsWeightsFormat_DEFAULT) {
3549 FullyConnectedOptionsWeightsFormat weights_format() const {
3569 void add_weights_format(FullyConnectedOptionsWeightsFormat weights_format) {
3570 …dElement<int8_t>(FullyConnectedOptions::VT_WEIGHTS_FORMAT, static_cast<int8_t>(weights_format), 0);
3587 … FullyConnectedOptionsWeightsFormat weights_format = FullyConnectedOptionsWeightsFormat_DEFAULT) {
3589 builder_.add_weights_format(weights_format);
9168 { auto _e = weights_format(); _o->weights_format = _e; };
9180 auto _weights_format = _o->weights_format;
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h547 FullyConnectedWeightsFormat weights_format =
Dtooling_util.cc2360 if (fc_op.weights_format == FullyConnectedWeightsFormat::kDefault) { in UndoWeightsShuffling()