Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for LSTMOp (0.15 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    //===----------------------------------------------------------------------===//
    // LSTMOp
    //===----------------------------------------------------------------------===//
    
    mlir::LogicalResult LSTMOp::verify() {
      LSTMOp op = *this;
      auto operands = op.GetStatefulOperands();
      if (operands.size() != 2 || operands[0] != 18 || operands[1] != 19) {
        return op.emitOpError("LSTMOp expected to have two stateful operands");
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        unrolling the input along the time or batch dimensions, and
        implements the following operation for
        each element in the sequence s = 1...sequence_length:
          outputs[s] = state = activation(LSTMOp(inputs[s]))
    
        where LSTMOp is LSTM TF Lite Op and the “activation” is the function passed
        as the “fused_activation_function” argument (if not “NONE”).
      }];
    
      let arguments = (
        ins TFL_FpTensor:$input,
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          continue;
        }
        std::vector<int32_t> intermediates;
        // Build intermediate tensors for tfl.lstm and insert these tensors into
        // flatbuffer.
        if (llvm::isa<mlir::TFL::LSTMOp, mlir::TFL::UnidirectionalSequenceLSTMOp>(
                inst)) {
          std::vector<std::string> intermediate_names = {
              "input_to_input_intermediate", "input_to_forget_intermediate",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top