Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 74 for Vector2 (0.19 sec)

  1. tensorflow/c/c_api.cc

        // Input tensors
        const std::vector<std::pair<string, Tensor>>& input_pairs,
        // Output tensors
        const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs,
        // Target nodes
        const std::vector<string>& target_oper_names, TF_Buffer* run_metadata,
        TF_Status* status) {
      const int noutputs = output_tensor_names.size();
      std::vector<Tensor> outputs(noutputs);
      Status result;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/nn_grad.cc

    #include "tensorflow/core/platform/errors.h"
    
    using std::vector;
    using tensorflow::ops::BiasAddGrad;
    using tensorflow::ops::Mul;
    using tensorflow::ops::ReluGrad;
    
    namespace tensorflow {
    namespace gradients {
    namespace {
    
    class ReluGradientFunction : public GradientFunction {
     public:
      explicit ReluGradientFunction(vector<AbstractTensorHandle*> f_outputs)
          : forward_outputs_(f_outputs) {
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/dlpack_test.cc

    #include "tensorflow/c/eager/dlpack.h"
    
    #include <vector>
    
    #include "absl/strings/str_join.h"
    #include "include/dlpack/dlpack.h"  // from @dlpack
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace {
    
    void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx,
                              std::vector<int64_t> shape,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/abstract_operation.h

      return SetAttrShape(attr_name, shape.dim_sizes().data(), shape.dims());
    }
    
    inline Status AbstractOperation::SetAttrStringList(
        const char* attr_name, absl::Span<string const> values) {
      std::vector<const char*> raw_strs;
      std::vector<size_t> lengths;
      raw_strs.reserve(values.size());
      lengths.reserve(values.size());
      for (const auto& s : values) {
        raw_strs.emplace_back(s.data());
        lengths.emplace_back(s.size());
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/eager/c_api_unified_experimental.h"
    
    #include <vector>
    
    #include "absl/container/flat_hash_map.h"
    #include "absl/strings/str_cat.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      Status Shape(const std::vector<int64_t>** shape) const;
      TF_DataType dtype() const { return dtype_; }
    
      // Sets its output argument to a summary of the values of this tensor on every
      // component device.
      Status SummarizeValue(std::string& summary);
    
      std::vector<TensorHandlePtr> release_tensors() { return std::move(tensors_); }
    
      std::vector<TFE_TensorHandle*> tensors() const {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      }
    }
    
    absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
    ParallelDevice::Join(
        const std::vector<PartialTensorShape>& expected_output_shapes,
        TF_Status* status) const {
      absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> result;
      // Compute per-device per-output tensors
      std::vector<std::vector<TensorHandlePtr>> per_device_output_tensors;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/parallel_device/parallel_device.cc

      device->delete_device = &DeleteParallelDevice;
      device->execute = &ParallelDeviceExecute;
      std::vector<std::string> underlying_devices_vector;
      underlying_devices_vector.reserve(num_underlying_devices);
      for (int device_index = 0; device_index < num_underlying_devices;
           ++device_index) {
        underlying_devices_vector.push_back(underlying_devices[device_index]);
      }
      std::unique_ptr<ParallelDevice> parallel_device(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/gradient_checker.cc

      int num_elems = TF_TensorElementCount(theta_tensor);
      vector<float> theta_data(num_elems);
      memcpy(theta_data.data(), TF_TensorData(theta_tensor),
             TF_TensorByteSize(theta_tensor));
    
      // Initialize space for the numerical gradient.
      vector<float> dtheta_approx(num_elems);
    
      // Get theta shape and store in theta_dims.
      int num_dims = TF_NumDims(theta_tensor);
      vector<int64_t> theta_dims(num_dims);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_function.cc

      // Process inputs.
      std::vector<tensorflow::OutputTensor> input_tensors;
      std::unordered_map<const Node*, std::vector<int>> input_nodes;
      status->status = tensorflow::ProcessInputs(fn_body, fn_name, ninputs, inputs,
                                                 &input_tensors, &input_nodes);
      if (TF_GetCode(status) != TF_OK) return nullptr;
    
      // Process outputs.
      std::vector<tensorflow::OutputTensor> output_tensors;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
Back to top