Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetBufferPointer (0.27 sec)

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

      LoadBasicModel();
      flatbuffers::FlatBufferBuilder builder;
      auto status = QuantizeWeights(&builder, model_, 0);
      EXPECT_EQ(status, kTfLiteOk);
    
      const uint8_t* buffer = builder.GetBufferPointer();
      const Model* output_model = GetModel(buffer);
      ASSERT_TRUE(output_model);
    }
    
    TEST_F(QuantizeWeightsTest, QuantizationFails) {
      LoadBasicModel();
      flatbuffers::FlatBufferBuilder builder;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 32.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

              ::tflite::optimize::QuantizerType::OLD_QUANTIZER) != kTfLiteOk) {
        return absl::InvalidArgumentError(
            "Quantize weights transformation failed.");
      }
      const uint8_t* q_buffer = q_builder.GetBufferPointer();
      *result =
          std::string(reinterpret_cast<const char*>(q_buffer), q_builder.GetSize());
    
      return absl::OkStatus();
    }
    
    absl::Status ConvertTFExecutorToStablehloFlatbuffer(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
Back to top