Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 14 of 14 for padding_values (0.35 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      if (!paddings_op) return failure();
    
      auto paddings_value = paddings_op.getValue().dyn_cast<DenseElementsAttr>();
      if (!paddings_value ||
          paddings_value.getNumElements() != permutation.size() * 2)
        return failure();
    
      SmallVector<int32_t, 8> shuffled_paddings(paddings_value.getNumElements());
      for (const auto &index_pair :
           llvm::enumerate(paddings_value.getValues<APInt>())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        padding_value                 ; otherwise.
    ```
    where `y = max(-k[1], 0)`, `x = max(k[1], 0)`.
    
    Otherwise, the output tensor has rank `r` with dimensions
    `[I, J, ..., L, num_diags, max_diag_len]` with values:
    
    ```
    diagonal[i, j, ..., l, m, n]
      = input[i, j, ..., l, n+y, n+x] ; if 0 <= n+y < M and 0 <= n+x < N,
        padding_value                 ; otherwise.
    ```
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // values will be undefined. So we mask them later. Set up the boolean
        // expression that tells us which entries, in the output shape, are out of
        // bounds and thus become the padding_value.
        Value x_in_bounds = rewriter.create<AndOp>(
            loc,
            rewriter.create<TF::GreaterEqualOp>(loc, b_false.getType(), n_plus_x,
                                                b_zero),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let arguments = (ins
         TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$input,
         TFL_TensorOf<[I32]>:$dilations,
         TFL_0DTensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$padding_value
      );
    
      let results = (outs TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$output);
    }
    
    def TFL_AddOp : TFL_Op<"add", [
    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