Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 83 for shape_1 (0.08 sec)

  1. tensorflow/c/c_api_test.cc

    TEST_F(CApiAttributesTest, ShapeList) {
      const int64_t shape_1[] = {1, 3};
      const int64_t shape_2[] = {2, 4, 6};
      const int64_t* list[] = {&shape_1[0], &shape_2[0]};
      const size_t list_size = TF_ARRAYSIZE(list);
      const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)};
      const int total_ndims = 5;  // ndims[0] + ndims[1]
    
      auto desc = init("list(shape)");
      TF_SetAttrShapeList(desc, "v", list, ndims, list_size);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      ParallelTensor(const ParallelDevice& device,
                     std::vector<TensorHandlePtr> tensors,
                     absl::Span<const int64_t> shape, const TF_DataType dtype)
          : device_(device),
            tensors_(std::move(tensors)),
            shape_(std::vector<int64_t>(shape.begin(), shape.end())),
            dtype_(dtype) {}
      ParallelTensor(const ParallelDevice& device,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

          new ParallelTensor(parallel_device, std::move(components), dtype));
    }
    
    absl::Status ParallelTensor::Shape(const std::vector<int64_t>** shape) const {
      if (!shape_.has_value()) {
        TF_Status status;
        PartialTensorShape combined_shape;
        TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&combined_shape));
    
        for (const TensorHandlePtr& component : tensors_) {
          PartialTensorShape component_shape;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        std::vector<PartialTensorShape> shapes;
        shapes.reserve(num_values);
        for (int i = 0; i < num_values; ++i) {
          if (num_dims[i] < 0) {
            shapes.emplace_back();
          } else {
            shapes.emplace_back(ArraySlice<int64_t>(
                reinterpret_cast<const int64_t*>(dims[i]), num_dims[i]));
          }
        }
        op_->node_builder.Attr(attr_name, shapes);
        return absl::OkStatus();
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

      std::vector<PartialTensorShape> shapes;
      shapes.reserve(num_shapes);
      for (int i = 0; i < num_shapes; ++i) {
        if (num_dims[i] < 0) {
          shapes.emplace_back();
        } else {
          shapes.emplace_back(ArraySlice<int64_t>(
              reinterpret_cast<const int64_t*>(dims[i]), num_dims[i]));
        }
      }
      desc->node_builder.Attr(attr_name, shapes);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_internal.h

    namespace tensorflow {
    
    // Set the shapes and types of the output's handle.
    //
    // The lengths of the arrays pointed to by `shapes`, `ranks`, and `types` must
    // all be equal to `num_shapes_and_types`. If `ranks[i] != -1`, (i.e., if the
    // rank is known), then it must be equal to the length of `shapes[i]`; if
    // `ranks[i] == 1`, then `shapes[i]` may be nullptr.
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat May 13 00:49:12 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test_util.h

    TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
    
    // Return an identity op.
    TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
    
    // Return a shape op fetching the shape of `a`.
    TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
    
    // Return an allreduce op adding up input tensor `in` from `group_size` workers.
    TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (0)
  8. tensorflow/c/eager/immediate_execution_tensor_handle.h

      //
      // -1 indicates an unknown axis length; this is unreachable for most standard
      // ImmediateExecutionTensorHandles, but comes up for example when computing
      // the shape of a parallel tensor with component shapes differing across
      // devices.
      virtual absl::Status Dim(int dim_index, int64_t* dim) const = 0;
    
      // Returns the device which created the handle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  9. tensorflow/c/checkpoint_reader.h

     public:
      CheckpointReader(const string& filename, TF_Status* status);
    
      bool HasTensor(const string& name) const;
      const string DebugString() const;
    
      // Returns a map from variable names to their shapes.  Slices of a partitioned
      // tensor are combined into a single entry.
      const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
    
      // Returns a map from variable names to their data types.  Slices of a
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  10. tensorflow/c/c_api.h

    // setting a shape of [-1, 2] with an existing shape [2, -1] would set
    // a final shape of [2, 2] based on shape merging semantics.
    //
    // Returns an error into `status` if:
    //   * `output` is not in `graph`.
    //   * An invalid shape is being set (e.g., the shape being set
    //     is incompatible with the existing shape).
    TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
Back to top