Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 35 for tensor_list (0.15 sec)

  1. 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)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/lite/transforms/passes.td

                 "legalized by this pass, then the IR won't be changed so that "
                 "tensorlist ops can pass through (default false).">,
          Option<
            "default_to_single_batch_", "default-to-single-batch", "bool", "false",
            "When specified to true, if the tensorlist ops has unspecified batch "
            "size, this pass will assume that the batch size is one to proceed "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 22.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      OpBuilder builder(list);
      llvm::SmallVector<int64_t, 8> element_shape;
      // Infer TensorList element shape from the return type first, and then from
      // the const element shape operand. We first check the return type because
      // shape inference might have successfully inferred the element shape from
      // write operations on the TensorList.
      if (failed(GetElementShapeFromResultType(list.getType(), &element_shape))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/common/tfl_pass_config.h

      // If `emit_builtin_tflite_ops` is true, TF Lite legalization passes will be
      // added, which produces TF Lite ops.
      bool emit_builtin_tflite_ops = true;
      // If `lower_tensor_list_ops` is true, tensorlist ops will be lowered to basic
      // TF ops before legalization to TF Lite dialect.
      bool lower_tensor_list_ops = false;
      // The allowlist of functions that would be preserved after trimming.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 19:05:30 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  7. tensorflow/cc/experimental/libtf/BUILD

            "//tensorflow/core/platform:status_matchers",
            "//tensorflow/core/platform:statusor",
        ],
    )
    
    tf_cc_test(
        name = "libtf_tensor_test",
        size = "medium",
        srcs = ["tests/tensor_test.cc"],
        tags = ["no_oss"],  # TODO(b/193268458): Need to disable TFRT.
        deps = [
            ":libtf",
            "//tensorflow/c:tf_status_helper",
            "//tensorflow/c/eager:abstract_context",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

      let summary =
        "Replicate TensorList init ops for correct shape assignments in shape inference";
    
      let description = [{
        If we pass same TensorList to a while op as multiple arguments or just use
        the same TensorList at multiple places and assign different
        TensorListSetItem to elements of TensorList, the shape inference is then
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc

        llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<bool> legalize_custom_tensor_list_ops(
        "legalize-custom-tensor-list-ops",
        llvm::cl::desc("Convert \"tf.TensorList*\" ops to \"tfl.custom_op\""
                       "if they can all be supported."),
        llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<bool> serialize_stablehlo_ops(
        "serialize-stablehlo-ops",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/flatbuffer_translate.cc

        llvm::cl::location(emit_custom_ops), llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    static opt<bool, true> lower_tensor_list_ops_flag(
        "lower-tensor-list-ops",
        llvm::cl::desc("Lower the TensorList ops within the TFLite dialect"),
        llvm::cl::location(lower_tensor_list_ops), llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    static opt<bool, true> strip_debug_info_flag(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 19:15:40 UTC 2024
    - 9.4K bytes
    - Viewed (0)
Back to top