Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SetBiasToCellGate (0.12 sec)

  1. tensorflow/compiler/mlir/lite/utils/lstm_utils.h

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 03 00:14:05 UTC 2023
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

          &builder_, weight_transposed_, weight_slice_shape_, begin_rec2o_values,
          weight_slice_shape_, weight_slice_size_recurrent_values_,
          fused_func_op_.getLoc());
    }
    
    void ConvertLSTMCellSimpleToFusedLSTM::SetBiasToCellGate() {
      SmallVector<int64_t, 1> begin_bias2c_values = {0};
      bias2cell_ = SliceRankedTensor(&builder_, bias_, bias_slice_shape_,
                                     begin_bias2c_values, bias_slice_shape_,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
Back to top