Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for operator_codes (0.3 sec)

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

      // check op and versioning.
      EXPECT_THAT(model_.operator_codes, SizeIs(2));
      EXPECT_THAT(GetBuiltinCode(model_.operator_codes[0].get()),
                  Eq(BuiltinOperator_ADD));
      ASSERT_THAT(model_.operator_codes[0]->version, Eq(2));
      EXPECT_THAT(GetBuiltinCode(model_.operator_codes[1].get()),
                  Eq(BuiltinOperator_RESHAPE));
      ASSERT_THAT(model_.operator_codes[1]->version, Eq(1));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc

          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;
      }
    };
    
    // Returns true if everything between the two graphs
    // are identical except for name field.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 32.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

        for (int j = 0; j < subgraph->operators()->Length(); ++j) {
          const tflite::Operator* op = subgraph->operators()->Get(j);
          const tflite::OperatorCode* op_code =
              model->operator_codes()->Get(op->opcode_index());
          auto status =
              tflite::CheckGpuDelegateCompatibility(op_code, op, subgraph, model);
          if (!status.ok()) {
            gpu_compatibile = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        A generic op for any TFLite custom operation.
    
        input: A list of inputs in the original op.
        custom_code: A string used to identify which exactly this op is, which
                     corresponds to operator_codes.custom_code in the flatbuffer.
        custom_option: a holder to save the op attributes in bytes fashion.
        output: A list of outputs in the original op.
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top