- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for opcode_index (0.25 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
for (auto msubgraph : *mutable_subgraphs) { auto operators = msubgraph->mutable_operators(); for (auto op : *operators) { auto opcode_idx = op->opcode_index(); auto opcodes = mutable_model->operator_codes(); auto opcode = (*opcodes)[opcode_idx]->builtin_code(); if (opcode == tflite::BuiltinOperator_CUSTOM) { std::vector<int32_t> inputs(op->inputs()->begin(), op->inputs()->end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc
ASSERT_THAT(GetBuiltinCode(model_.operator_codes[op1->opcode_index].get()), Eq(BuiltinOperator_QUANTIZE)); ASSERT_THAT(GetBuiltinCode(model_.operator_codes[op2->opcode_index].get()), Eq(BuiltinOperator_QUANTIZE)); ASSERT_THAT(GetBuiltinCode(model_.operator_codes[op3->opcode_index].get()), Eq(BuiltinOperator_PACK));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc
const auto op = subgraph->operators()->Get(op_idx); for (size_t i = 0; i < op->outputs()->size(); ++i) { if (op->outputs()->Get(i) == tensor_idx) { const uint32_t op_code_idx = op->opcode_index(); *op_code = GetBuiltinCode(model->operator_codes()->Get(op_code_idx)); return true; } } } return false; } };
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/lite/tests/flatbuffer2mlir/importer_test_min_max.cc
for (auto& sub_graph : model->subgraphs) { for (auto& op : sub_graph->operators) { if (tflite::GetBuiltinCode( model->operator_codes[op->opcode_index].get()) == tflite::BuiltinOperator_FULLY_CONNECTED) { // inject min/max to the input and output tensors auto& input_tensor = sub_graph->tensors[op->inputs[0]];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.h
// Returns the FlatBuffer offset for the operator on success; emits error and // returns std::nullopt on failure. std::optional<flatbuffers::Offset<tflite::Operator>> CreateFlatBufferOperator( Operation *mlir_op, uint32_t opcode_index, const std::vector<int32_t> &operands, const std::vector<int32_t> &results, const std::vector<int32_t> &intermediates, flatbuffers::FlatBufferBuilder *fbb);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 21:00:09 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_per_channel.pbtxt
# CHECK: builtin_options_type: Conv2DOptions, # CHECK: builtin_options: { # CHECK: stride_w: 1, # CHECK: stride_h: 1 # CHECK: } # CHECK: }, { # CHECK: opcode_index: 1, # CHECK: inputs: [ 4, 1 ], # CHECK: outputs: [ 5 ] # CHECK: } ], # CHECK: name: "main" # CHECK: } ], # CHECK: description: "MLIR Converted.", # CHECK: buffers: [ {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 18.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_per_channel_4bit.pbtxt
# CHECK: builtin_options_type: Conv2DOptions, # CHECK: builtin_options: { # CHECK: stride_w: 1, # CHECK: stride_h: 1 # CHECK: } # CHECK: }, { # CHECK: opcode_index: 1, # CHECK: inputs: [ 4, 1 ], # CHECK: outputs: [ 5 ] # CHECK: } ], # CHECK: name: "main" # CHECK: } ], # CHECK: description: "MLIR Converted.", # CHECK: buffers: [ {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 18.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
const std::unique_ptr<tflite::FlatBufferModel>& model_ptr) { llvm::SmallVector<Value, 4> operands; llvm::SmallVector<mlir::Type, 2> outputTypes; const tflite::OperatorCodeT& op_code = *op_codes.at(op.opcode_index); const std::string op_name = GetMlirOpName(op, op_code); OperationState op_state(loc, op_name); for (auto input_num : op.inputs) { if (input_num == -1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
struct OperatorBuilder { typedef Operator Table; ::flatbuffers::FlatBufferBuilder &fbb_; ::flatbuffers::uoffset_t start_; void add_opcode_index(uint32_t opcode_index) { fbb_.AddElement<uint32_t>(Operator::VT_OPCODE_INDEX, opcode_index, 0); } void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> inputs) { fbb_.AddOffset(Operator::VT_INPUTS, inputs); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)