- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for input_model (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
// * Options for full/partial conversion, Op exceptions list. // * Option to serialize output to TFL flatbuffer format. using llvm::cl::opt; // NOLINTNEXTLINE opt<std::string> input_model(llvm::cl::Positional, llvm::cl::desc("<input model path>"), llvm::cl::Required); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.py
quantization_options = tf.quantization.experimental.QuantizationOptions( signature_keys=['your_signature_key'], ) tf.quantization.experimental.quantize_saved_model( '/tmp/input_model', '/tmp/output_model', quantization_options=quantization_options, ) # When quantizing a model trained without QAT (Post-Training Quantization), # a representative dataset is required.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 34.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
flatbuffers::FlatBufferBuilder q_builder(/*initial_size=*/10240); const uint8_t* buffer = reinterpret_cast<const uint8_t*>(translated_result.c_str()); 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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc
QuantizeWeightsTest() {} void LoadBasicModel() { input_model_ = ReadTestModel(); model_ = input_model_->GetModel(); } void LoadSharedWeightsModel() { input_model_ = ReadSharedWeightsTestModel(); model_ = input_model_->GetModel(); } void LoadGatherTestModel() { input_model_ = ReadGatherTestModel(); model_ = input_model_->GetModel(); } void LoadCustomOpTestModel() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 32.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/representative_dataset.py
signature_keys=['serving_default'], representative_datasets=dataset_file_map, ) tf.quantization.experimental.quantize_saved_model( '/tmp/input_model', '/tmp/output_model', quantization_options=quantization_options, ) ``` """ def __init__( self, path_map: Mapping[str, os.PathLike[str]],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 14.2K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
ASSERT_NE(if_node, nullptr); const Node *input_node; TF_CHECK_OK(if_node->input_node(1, &input_node)); EXPECT_EQ(input_node->name(), "arg1"); TF_CHECK_OK(if_node->input_node(2, &input_node)); EXPECT_EQ(input_node->name(), "arg0"); const Node *ret0_node = node_name_index.at("ret0"); ASSERT_NE(ret0_node, nullptr); TF_CHECK_OK(ret0_node->input_node(0, &input_node)); EXPECT_EQ(input_node->name(), "if");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
shape_refiner->GetContext(n); for (int i = 0; i < n->num_inputs(); i++) { const Node* input_node; if (n->input_node(i, &input_node).ok()) { auto shapes_and_types = context->input_handle_shapes_and_types(i); if (shapes_and_types) { context->set_output_handle_shapes_and_types(0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
send_recv_nodes.push_back(n); } } EXPECT_EQ(num_send_from_host, 1); EXPECT_EQ(num_recv_at_host, 1); for (Node *n : send_recv_nodes) { Node *input_node; TF_CHECK_OK(n->input_node(n->num_inputs() - 1, &input_node)); EXPECT_EQ(input_node, key_placeholder); bool has_control_edge_to_sequencer = false; for (const Edge *e : n->out_edges()) { if (e->IsControlEdge() && e->dst() == sequencer) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util_test.cc
NodeBuilder while_builder(kFunctionalWhileNodeName, "While", builder.opts().op_registry()); while_builder.Input({input_node, input_node}) .Attr("cond", compilable) .Attr("body", uncompilable); builder.opts().FinalizeBuilder(&while_builder); GraphDef graph_def; TF_EXPECT_OK(builder.ToGraphDef(&graph_def)); std::unique_ptr<Graph> graph(new Graph(flib_def_.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0)