Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for ConvertToTensor (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h

    Status ConvertToTensorProto(mlir::ElementsAttr attr,
                                TensorProto* output_tensor);
    
    // Converts an MLIR elements attribute to a TensorFlow tensor.
    Status ConvertToTensor(mlir::ElementsAttr attr, Tensor* output_tensor);
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc

            "Operand idx 1 is not a tf.ConstOp. The initializing tensor is not "
            "saved to checkpoint.");
        return "";
      }
    
      Tensor const_tensor{};
      if (const absl::Status status = mlir::tfg::ConvertToTensor(
              /*attr=*/const_op.getValue(), /*output_tensor=*/&const_tensor);
          !status.ok()) {
        return status;
      }
    
      if (!bundle_writer.Add(/*key=*/var_handle_op.getSharedName(), const_tensor)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:36:55 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/eval_util.cc

      }
    
      VLOG(1) << "Start to evaluate node: " << *node_def;
    
      // Adds inputs to the TF operation.
      for (const auto operand : operands) {
        Tensor tensor;
        RETURN_FAILURE_IF_ERROR(ConvertToTensor(operand, &tensor));
        TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status->status);
        RETURN_FAILURE_IF_ERROR(status);
        auto clean_tensor =
            MakeCleanup([tf_tensor] { TF_DeleteTensor(tf_tensor); });
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 13 06:02:14 UTC 2023
    - 5.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold_utils.cc

      std::vector<tensorflow::Tensor> inputs;
    
      // Adds inputs to the TF operation.
      for (const ElementsAttr& operand : operands) {
        tensorflow::Tensor tensor;
        RETURN_FAILURE_IF_ERROR(tensorflow::ConvertToTensor(operand, &tensor));
        inputs.push_back(std::move(tensor));
      }
    
      std::vector<tensorflow::TensorValue> input_values;
      for (tensorflow::Tensor& tensor : inputs) {
        input_values.emplace_back();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.3K bytes
    - Viewed (0)
Back to top