Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for input_model (0.15 sec)

  1. tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.cc

      StringRef op_name(tflite_op_name.data(), tflite_op_name.size());
      return op_name.lower();
    }
    
    std::unique_ptr<tflite::ModelT> CreateMutableModelFromFile(
        const tflite::Model* input_model) {
      auto copied_model = std::make_unique<tflite::ModelT>();
      input_model->UnPackTo(copied_model.get(), nullptr);
      return copied_model;
    }
    }  // namespace
    
    // TODO(b/214314076): Support MLIR model as an input for the C++ dynamic range
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

          "tensorflow/lite/testdata/sparse_tensor.bin");
      tflite::ModelT input_model;
      input_fbm->GetModel()->UnPackTo(&input_model);
    
      // Populate input metadata
      auto model_metadata_buffer = std::make_unique<tflite::BufferT>();
      model_metadata_buffer->data =
          std::vector<uint8_t>(expected_value.begin(), expected_value.end());
      input_model.buffers.push_back(std::move(model_metadata_buffer));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/sparsity/sparsify_model.cc

    namespace mlir {
    namespace lite {
    
    absl::Status SparsifyModel(const tflite::ModelT& input_model,
                               flatbuffers::FlatBufferBuilder* builder) {
      MLIRContext context;
      StatusScopedDiagnosticHandler statusHandler(&context,
                                                  /*propagate=*/true);
    
      // Import input_model to a MLIR module
      flatbuffers::FlatBufferBuilder input_builder;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.h

                                 const tflite::Model* input_model,
                                 int64_t weights_min_num_elements,
                                 bool use_hybrid_evaluation = true);
    
    TfLiteStatus QuantizeWeights(flatbuffers::FlatBufferBuilder* builder,
                                 const tflite::Model* input_model,
                                 BufferType quant_type = BufferType::QUANTIZED_INT8,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 4.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/sparsity/sparsify_model.h

    #include "tensorflow/compiler/mlir/lite/schema/schema_generated.h"
    
    namespace mlir {
    namespace lite {
    
    // Sparsify the `input_model` and write the result to a flatbuffer `builder`.
    absl::Status SparsifyModel(const tflite::ModelT& input_model,
                               flatbuffers::FlatBufferBuilder* builder);
    }  // namespace lite
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 1.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.cc

                                  const absl::string_view inference_type,
                                  QuantizationSpecs* quant_specs) {
      const std::vector<std::string> input_nodes = absl::StrSplit(node_names, ',');
      std::vector<std::optional<double>> node_mins;
      if (!min_values.empty()) {
        std::vector<std::string> node_mins_str = absl::StrSplit(min_values, ',');
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_compile_util_test.cc

      EXPECT_EQ(identity_node->attrs().FindByString("T")->type(), DT_FLOAT);
    
      EXPECT_EQ(identity_node->num_inputs(), 1);
      const Node* identity_input_node = nullptr;
      TF_EXPECT_OK(identity_node->input_node(0, &identity_input_node));
      EXPECT_EQ(identity_input_node->name(), "_arg0");
    
      const Node* arg_node = node_name_index.at("_arg0");
      EXPECT_EQ(arg_node->op_def().name(), "_Arg");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 16 21:48:05 UTC 2023
    - 6K bytes
    - Viewed (0)
Back to top