- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for LSTMOp (0.34 sec)
-
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) -
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) -
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)