Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 45 for tensor_list (0.28 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        // meaningless value -1 for non resource outputs.
        out_desc.input_index =
            it != output_to_input_alias.end() ? it->getSecond() : -1;
        // MLIR-based TF-Compiler bridge doesn't support tensorlist output yet.
        // TODO(ycao): Support tensorlist-type output.
        out_desc.is_tensor_list = false;
      }
    
      // XLA computation always uses Tuple shape.
      *xla_output_shape = xla::ShapeUtil::MakeTupleShape(shapes);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  9. tensorflow/cc/experimental/base/tests/BUILD

        name = "tensor_types_test_util",
        testonly = True,
        hdrs = ["tensor_types_test_util.h"],
        deps = [
            "//tensorflow/c:tf_datatype",
        ],
    )
    
    tf_cc_test(
        name = "tensor_test",
        srcs = [
            "tensor_test.cc",
        ],
        deps = [
            ":tensor_types_test_util",
            "//tensorflow/c:tf_datatype",
            "//tensorflow/cc/experimental/base/public:status",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        }
      };
    
      while (!worklist.empty()) {
        tensorlist = worklist.top();
        worklist.pop();
    
        // TensorLists are semantically immutable. For example, TensorListSetItem
        // takes a TensorList as input and produces a TensorList as output. So to
        // traverse modifications to TensorList and verify that all elements written
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top