Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 136 for tensor_b (0.29 sec)

  1. tensorflow/c/eager/tape.h

        Gradient* grad_tensor = grad[grad_index];
        if (grad_tensor != nullptr) {
          int64_t tensor_id = vspace_.TensorId(grad_tensor);
          targets.push_back(tensor_id);
          if (sources_set.find(tensor_id) != sources_set.end()) {
            sources_that_are_targets.emplace(
                tensor_id, vspace_.TapeTensorFromGradient(grad_tensor));
          }
          Gradient* in_grad = in_grads[grad_index];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 12:40:29 UTC 2024
    - 47.2K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib.h

        std::vector<TFE_TensorHandle*> result;
        result.reserve(tensors_.size());
        for (const TensorHandlePtr& tensor : tensors_) {
          result.emplace_back(tensor.get());
        }
        return result;
      }
    
     private:
      ParallelTensor(const ParallelDevice& device,
                     std::vector<TensorHandlePtr> tensors,
                     absl::Span<const int64_t> shape, const TF_DataType dtype)
          : device_(device),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 25 15:21:13 UTC 2023
    - 12.9K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.h

                                                           TF_Status* status);
    
    // On success, enqueues `tensor` into a TF-managed FifoQueue given by
    // `tensor_id`, associated with `session`. There must be a graph node named
    // "fifo_queue_enqueue_<tensor_id>", to be executed by this API call. It reads
    // from a placeholder node "arg_tensor_enqueue_<tensor_id>".
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      for (int component_index = 0; component_index < tensors_.size();
           ++component_index) {
        // TODO(allenl): Add a C API for summarizing tensors. Currently custom
        // devices limiting themselves to a C API (for ABI compatibility) would need
        // to implement summarization for component tensors themselves.
        ImmediateExecutionTensorHandle* component =
            tensorflow::unwrap(tensors_[component_index].get());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 07:47:20 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  9. 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)
  10. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td

              (TF_SubOp $max_range, $min_range),
              (TF_ConstOp (GetF32Scalar<255>)))),
          $min_range),
        /*truncate=*/ConstBoolAttrFalse),
      [(TensorOf<[TF_Qint8, TF_Quint8]> $input),
       (TF_Float32Tensor $min_range),
       (TF_Float32Tensor $max_range),
       (TensorOf<[TF_Float32, TF_Bfloat16]> $result)]>;
    
    //===----------------------------------------------------------------------===//
    // Difference op patterns.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 24.7K bytes
    - Viewed (0)
Back to top