Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 28 for tensor_list (0.45 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

        Example:
    
        ```mlir
        %2 = tfr.call @my_add(%0, %1) : (tfr.tensor, f32) -> tfr.tensor_list
        ```
    
        Note that the operands of the `call` operation can only be with tfr.tensor,
        tfr.tensor_list, tfr.attr and mlir float and integer types. The results of
        the `call` operation can only be with tfr.tensor and tfr.tensor_list types.
      }];
    
      let arguments = (ins
        FlatSymbolRefAttr:$callee,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/passes/decompose.cc

        // Create the new operands. This is mapping the operands from the target
        // TF ops to the TFR function arguments. If the TFR function argument is
        // a tensor_list, a "tfr.build_list" op is used to concat the available
        // TF op operands. If the TFR function argument isn't a tensor/tensor_list,
        // a constant is created by using the attribute stored in the TF op or the
        // default value in the argument attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

          SmallVector<Value, 4> tensor_list;
          for (int i = res.index(); i < new_op->getNumResults(); i++) {
            Value new_res = new_op->getResult(i);
            auto casted =
                rewriter.create<CastOp>(loc, unconstrainted_type, new_res);
            tensor_list.push_back(casted.getOut());
          }
          auto list_op = rewriter.create<BuildListOp>(loc, res_type, tensor_list);
          new_results.push_back(list_op.getOut());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir

      // CHECK: [[tensor_list:%.*]] = "tf.TensorListReserve"([[shape:%.*]], [[reserve_size:%.*]]) {
      %0 = "tf.TensorListReserve"(%cst_1, %cst_2) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xf32>>>
      // CHECK: [[map_fn_result:%.*]] = tf_mlrt.tf_map_fn([[reserve_size]], [[tensor_list]], %arg0)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:40:22 UTC 2024
    - 68.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc

          output << " input_index=" << xla_output_description.input_index;
        if (xla_output_description.is_constant) output << " constant";
        if (xla_output_description.is_tensor_list) output << " tensor_list";
        output << '\n';
      }
    
      for (const auto& resource_update : compilation_result.resource_updates) {
        output << "// ResourceUpdate input_index=" << resource_update.input_index
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.8K bytes
    - Viewed (0)
  6. tensorflow/c/kernels_experimental.cc

      if (a.TypeId() == tensorflow::TypeIndex::Make<::tensorflow::TensorList>()) {
        TF_RETURN_IF_ERROR(ValidateVariantType<::tensorflow::TensorList>(a));
        *out = ::tensorflow::TensorList();
    
        return ::tensorflow::TensorListBinaryAdd(
            cc_ctx, *a.get<::tensorflow::TensorList>(),
            *b.get<::tensorflow::TensorList>(),
            out->get<::tensorflow::TensorList>(), cc_binary_add);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

    // Returns a `llvm::DenseMap` which maps from the index of tensorlist in the
    // result, to the index of the same tensorlist in the arguments. For `If` op's
    // branch functions, the results and arguments are not usually matched 1-1. This
    // will let us konw which tensorlist result maps to which tensorlist in the
    // arguments. Once we know this info it will help us decide the types of the
    // result tensorlist based on the operand's of the `If` op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir

    // RUN: tf-opt -tfl-legalize-tensorlist -canonicalize -split-input-file %s | FileCheck %s
    
    // CHECK-LABEL: listReserveScalarShapeI32
    func.func @listReserveScalarShapeI32(%arg0: tensor<i32>, %arg1: tensor<i32>) -> tensor<!tf_type.variant<tensor<*xi32>>> {
      %0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xi32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  9. tensorflow/cc/experimental/base/tests/tensor_test.cc

    A. Unique TensorFlower <******@****.***> 1712999357 -0700
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 09:56:08 UTC 2024
    - 6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

          });
      return has_variant_input || has_variant_output;
    }
    
    // There are 2 standard tf ops which are not TensorList ops that may take as
    // input a tensorlist. These are tf.AddN and tf.ZeroesLike. Since the runtime
    // implementation of a tensorlist are not compatible between tf and tflite
    // we cannot use tflite tensorlist kernels until these cases are handled.
    bool IsNonTensorListVariantOp(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
Back to top