Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 55 for TfLite (0.09 sec)

  1. tensorflow/compiler/mlir/lite/python/converter_python_api.cc

    }
    
    tflite::TensorType FromTocoDataTypeToTflitToTensorType(int inference_type) {
      switch (inference_type) {
        case toco::IODataType::QUANTIZED_INT16:
          return tflite::TensorType_INT16;
        case toco::IODataType::QUANTIZED_UINT8:
          return tflite::TensorType_UINT8;
        case toco::IODataType::UINT8:
          return tflite::TensorType_UINT8;
        case toco::IODataType::QUANTIZED_INT8:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

          .Case("TANH", tflite::ActivationFunctionType_TANH)
          .Case("SIGN_BIT", tflite::ActivationFunctionType_SIGN_BIT);
    }
    
    static tflite::TensorType ConvertDerivedTFLiteTypeAttrForOptionWriter(
        tflite::TensorType type, flatbuffers::FlatBufferBuilder* builder) {
      if (type == tflite::TensorType_INT64) {
        return tflite::TensorType_INT64;
      } else if (type == tflite::TensorType_INT32) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/converter_gen.cc

            }
          }
        }
    
        os << "  tflite::" << tflite_option_name << "Builder b(*fbb);\n";
        for (const auto &option : options)
          os << formatv("  b.add_{0}(std::move({0}));\n", option);
        os << "  return b.Finish();\n}\n";
      }
    }
    
    // For each TFLite op, emits a builder function that packs the TFLite op into
    // the corresponding FlatBuffer object.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 15:05:28 UTC 2023
    - 23.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

      const ::tflite::Model* input_model = ::tflite::GetModel(buffer);
    
      ::tflite::optimize::BufferType quantized_type;
      switch (quant_specs.inference_type) {
        case DT_QINT8:
          quantized_type = ::tflite::optimize::BufferType::QUANTIZED_INT8;
          break;
        case DT_HALF:
          quantized_type = ::tflite::optimize::BufferType::QUANTIZED_FLOAT16;
          break;
        default:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_operator.h

    bool IsStablehloOp(const tflite::OperatorCodeT &op_code);
    
    // Returns the MLIR op name for the flatbuffer operator corresponding to
    // `op_code`.
    std::string GetMlirOpNameFromOpCode(const ::tflite::OperatorCodeT &op_code);
    
    // Returns the builtin op code for the given MLIR operation on success; emits
    // error and returns std::nullopt on failure.
    std::optional<tflite::BuiltinOperator> GetBuiltinOpCode(Operation *mlir_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 21:00:09 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

      for (auto dim : tensor.shape) {
        shape->add_dim()->set_size(int64_t{dim});
      }
      // TensorFlow Lite uses tflite::DynamicBufer to encode vector of strings.
      if (tensor.type == tflite::TensorType_STRING) {
        for (int i = 0; i < tflite::GetStringCount(buffer.data()); ++i) {
          tflite::StringRef str = tflite::GetString(buffer.data(), i);
          ret.add_string_val(str.str, str.len);
        }
        return ret;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

              llvm::dyn_cast_or_null<mlir::TF::TensorListReserveOp>(op)) {
        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(reserve.getElementDtype());
    
        return CreateListReserveOptions(context, tflite_type);
      }
      if (auto empty = llvm::dyn_cast_or_null<mlir::TF::EmptyTensorListOp>(op)) {
        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(empty.getElementDtype());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tf_tfl_passes.cc

      }
      // The below passes only make sense if Builtin TFLite ops are enabled
      // for emission.
      if (pass_config.emit_builtin_tflite_ops) {
        // Run shape inference after variables are converted to constants.
        if (pass_config.shape_inference) {
          pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass());
        }
        // Force layout supported by TFLite, this will transpose the data
        // to match 'kTFLiteDataLayout'
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 25.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/passes.td

      let options = [
        Option<"default_min_", "default-min", "double", "-1.0",
                   "Default minimum value for TFLite quantization">,
        Option<"default_max_", "default-max", "double", "1.0",
                   "Default maximum value for TFLite quantization">,
        Option<"is_signed_", "is-signed", "bool", "false",
                   "Is the corresponding integer signed">,
      ];
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 22.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc

    //
    // Input: TF Saved Model or TF MLIR files.
    // Output: StableHLO MLIR or TFLite Flatbuffer.
    //
    // Usage:
    // 1) convert the input model to StableHLO MLIR
    // odml_to_stablehlo /path/to/model -o=model.mlir
    // 2) convert the input model to TFLite Flatbuffer
    // odml_to_stablehlo /path/to/model -o=model.tflite --export-type=tflite
    //
    // TODO(pulkitb): Add more options.
    //  * Verbose - dump intermediate translations.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 14.1K bytes
    - Viewed (0)
Back to top