Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 24 for tensor_list (0.2 sec)

  1. 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)
  2. 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)
  3. 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)
  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/transforms/passes.h

    // Creates a pass that adds control dependencies to keep the relative execution
    // order of operations with side effects frozen.
    std::unique_ptr<OperationPass<func::FuncOp>> CreatePinOpsWithSideEffectsPass();
    
    // Legalize TensorList Ops iff all of them are supported.
    std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTensorListPass();
    
    // Reduce the type precision of some tensor types if all values within that
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 07 21:29:34 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc

                    .getDimSize(1),
                3);
    
      auto it = fused_lstm_func_.getBody().back().rbegin();
      EXPECT_EQ(it->getName().getStringRef(),
                mlir::func::ReturnOp::getOperationName());
      it++;  // tensor_cast
      it++;  // lstm
      EXPECT_EQ(it->getName().getStringRef(),
                mlir::TFL::LSTMOp::getOperationName());
      EXPECT_EQ(it->getNumOperands(), 24);
      EXPECT_EQ(it->getNumResults(), 1);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir

    // RUN: tf-opt "-tfl-lower-static-tensor-list=allow-tensorlist-pass-through default-to-single-batch" -split-input-file %s | FileCheck %s
    
    // -----
    
    // CHECK-LABEL: tensorlistConst
    func.func @tensorlistConst(%arg0 : tensor<1xi32>) -> tensor<2x3xi32> {
      // CHECK-DAG: %[[ELEMENT0:.*]] = "tf.Const"() <{value = dense<[0, 1, 2]> : tensor<3xi32>}> : () -> tensor<3xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 39.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/xla_launch_util.cc

      }
    
      for (const XlaOutputDescription& descr : compilation_result->outputs) {
        if (descr.type == DT_VARIANT) {
          return errors::Unimplemented(
              "Support for TensorList crossing the XLA/TF boundary "
              "is not implemented");
        }
      }
    
      std::vector<TensorShape> output_tensor_shapes;
      output_tensor_shapes.reserve(ctx->num_outputs());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  10. tensorflow/c/eager/tape.h

        const int64_t id = target_tensor_ids[i];
        if (output_gradients.empty() || output_gradients[i] == nullptr) {
          auto tensor_it = tensor_tape.find(id);
          if (tensor_it != tensor_tape.end() && tensor_it->second != -1) {
            auto op_it = op_tape.find(tensor_it->second);
            if (op_it == op_tape.end()) {
              return errors::Internal(
                  "Internal state of the gradient tape is invalid: "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 12:40:29 UTC 2024
    - 47.2K bytes
    - Viewed (0)
Back to top