Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 367 for tensor2 (0.18 sec)

  1. tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.cc

        for (const int itensor : operations_[ioperation].tensors) {
          if (const auto& tensor = tensors_[itensor];
              tensor.first_use() >= begin && tensor.first_use() < end &&
              tensor.last_use() > peak_loc) {
            for (const int ioperation : tensor.operations) {
              // We return the earliest dependence on any output tensor.
              if (ioperation > peak_loc && ioperation < best) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 14 20:57:44 UTC 2023
    - 13.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir

      %0 = "tf.TensorListFromTensor"(%arg0, %arg1) : (tensor<*xf32>, tensor<1xi32>) -> tensor<!tf_type.variant<tensor<*xf32>>>
      %1 = "tf.TensorListGetItem"(%0, %arg2, %arg1) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<i32>, tensor<1xi32>) -> tensor<*xf32>
      %2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<1xi32>) -> tensor<*xf32>
      func.return %1, %2 : tensor<*xf32>, tensor<*xf32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 39.9K bytes
    - Viewed (0)
  3. tensorflow/cc/framework/ops.h

      DataType data_type() const { return data_type_; }
      Status status() const { return status_; }
      const Tensor& tensor() const { return tensor_; }
    
     private:
      Status status_;
      Output output_ = Output(Operation(nullptr), 0);
      Tensor tensor_;
      const std::string node_name_ = "";
      int32 index_ = 0;
      DataType data_type_ = DT_INVALID;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td

    // Tensor types
    def TF_BoolTensor : TensorOf<[TF_Bool]>;
    
    def TF_IntTensor : TensorOf<[TF_Int]>;
    def TF_Int4Tensor : TensorOf<[TF_Int4]>;
    def TF_Int8Tensor : TensorOf<[TF_Int8]>;
    def TF_Int16Tensor : TensorOf<[TF_Int16]>;
    def TF_Int32Tensor : TensorOf<[TF_Int32]>;
    def TF_Int64Tensor : TensorOf<[TF_Int64]>;
    def TF_I32OrI64Tensor : TensorOf<[TF_I32OrI64]>;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 30.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.h

        std::vector<int> tensors;  // The tensors that are used (input or output) by
                                   // this operation. They needn't correspond to
                                   // tensors in the TF graph -- we may add fake
                                   // tensors to model memory consumed in addition
                                   // to input and output tensors. This vector is
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 14 20:57:44 UTC 2023
    - 12K bytes
    - Viewed (0)
  6. tensorflow/c/tf_tensor.cc

        auto* ret = emptyTensor->tensor;
        delete emptyTensor;
        return ret;
      }
    
      Tensor tensor;
      if (!tensor.CopyFrom(src, src.shape())) {
        return nullptr;
      }
      return new tensorflow::TensorInterface(std::move(tensor));
    }
    
    // Non-static for testing.
    TF_Tensor* TF_TensorFromTensor(const tensorflow::Tensor& src, Status* status) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 21:57:32 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

      }
    }
    
    }  // namespace
    
    bool IsQuantized(const TensorT& tensor) {
      return (tensor.quantization != nullptr) &&
             !tensor.quantization->zero_point.empty();
    }
    
    // Returns the correct type for a quantized tensor.
    // We have a special case for constants since they have a higher minimum value.
    StatusOr<QuantizedType> GetQuantizedType(const TensorT& tensor, Builder builder,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/test_schema.fbs

    // model.
    table SubGraph {
      // A list of all tensors used in this subgraph.
      tensors:[Tensor];
    
      // Indices of the tensors that are inputs into this subgraph. Note this is
      // the list of non-static tensors that feed into the subgraph for inference.
      inputs:[int];
    
      // Indices of the tensors that are outputs out of this subgraph. Note this is
      // the list of output tensors that are considered the product of the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 19 19:46:06 UTC 2021
    - 26.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/schema/schema_v3b.fbs

    // model.
    table SubGraph {
      // A list of all tensors used in this subgraph.
      tensors:[Tensor];
    
      // Indices of the tensors that are inputs into this subgraph. Note this is
      // the list of non-static tensors that feed into the subgraph for inference.
      inputs:[int];
    
      // Indices of the tensors that are outputs out of this subgraph. Note this is
      // the list of output tensors that are considered the product of the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 14:28:27 UTC 2024
    - 30K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/schema/schema.fbs

    // model.
    table SubGraph {
      // A list of all tensors used in this subgraph.
      tensors:[Tensor];
    
      // Indices of the tensors that are inputs into this subgraph. Note this is
      // the list of non-static tensors that feed into the subgraph for inference.
      inputs:[int];
    
      // Indices of the tensors that are outputs out of this subgraph. Note this is
      // the list of output tensors that are considered the product of the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 41.7K bytes
    - Viewed (0)
Back to top