Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 24 of 24 for add_shape (0.68 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir

      %tl = "tf.TensorListFromTensor"(%arg0, %elem_shape) : (tensor<10x8x9xf32>, tensor<2xi32>) -> tensor<!tf_type.variant<tensor<8x9xf32>>>
      // CHECK: %[[IND_SHAPE:.*]] = "tf.Const"() <{value = dense<[5, 1]> : tensor<2xi32>}> : () -> tensor<2xi32>
      // CHECK: %[[IND_RESHPE:.*]] = "tf.Reshape"(%[[ARG1]], %[[IND_SHAPE]]) : (tensor<5xi32>, tensor<2xi32>) -> tensor<5x1xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 38.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

          return failure();
        }
        // It is already a 1-D constant, no change.
        auto old_shape = shape.getShapedType().getShape();
        if (old_shape.size() == 1) {
          return failure();
        }
        // Verify all the leading dimensions are length one, except the last one.
        for (auto it = ++old_shape.rbegin(); it != old_shape.rend(); ++it) {
          if (*it != 1) {
            reshape->emitError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        Type output_type = op.getResult().getType();
        const int64_t rank = operand_type.getRank();
    
        ArrayRef<int64_t> idx_shape(rank);
        TensorType idx_type =
            operand_type.cloneWith(idx_shape, rewriter.getI32Type());
    
        ArrayRef<int64_t> start_idx_i64 = op.getStartIndices();
        ArrayRef<int64_t> limit_idx_i64 = op.getLimitIndices();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        Arg<TF_NumberTensor, [{1-D.  The `values` of the first `SparseTensor`, size `[nnz]` Vector.}]>:$a_values,
        Arg<TF_Int64Tensor, [{1-D.  The `shape` of the first `SparseTensor`, size `[ndims]` Vector.}]>:$a_shape,
        Arg<TF_Int64Tensor, [{2-D.  The `indices` of the second `SparseTensor`, size `[nnz, ndims]` Matrix.}]>:$b_indices,
        Arg<TF_NumberTensor, [{1-D.  The `values` of the second `SparseTensor`, size `[nnz]` Vector.}]>:$b_values,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top