Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 66 for shape (0.16 sec)

  1. tensorflow/c/c_api.cc

          auto* out_shape_and_type = handle_data.add_shape_and_type();
          ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape());
          out_shape_and_type->set_dtype(p.dtype);
          *out_shape_and_type->mutable_type() = p.type;
        }
      }
      string str_data;
      handle_data.SerializeToString(&str_data);
    
      TF_Buffer* result = TF_NewBufferFromString(str_data.c_str(), str_data.size());
      return 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/eager/dlpack_test.cc

      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
      dltensor_in->shape = shape.data();
      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
          << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",")
          << "], strides=[" << absl::StrJoin(strides, ",") << "])";
    
    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)
  3. tensorflow/c/checkpoint_reader.cc

      if (reader_ != nullptr) {
        status = reader_->GetTensor(name, out_tensor);
      } else {
        tensorflow::DataType dtype;
        tensorflow::TensorShape shape;
        status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
        if (status.ok()) {
          out_tensor->reset(new Tensor(dtype, shape));
          status = v2_reader_->Lookup(name, out_tensor->get());
          if (!status.ok()) out_tensor->reset();
        }
      }
      if (!status.ok()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_debug.cc

      std::vector<int64_t> shape;
      int rank = -1;
      *status = handle.NumDims(&rank);
      if (!status->ok()) {
        return shape;
      }
      shape.reserve(rank);
      for (int i = 0; i < rank; ++i) {
        int64_t dim;
        *status = handle.Dim(i, &dim);
        if (!status->ok()) {
          return shape;
        }
        shape.push_back(dim);
      }
      return shape;
    }
    
    }  // namespace
    
    extern "C" {
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 2.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental.cc

        return nullptr;
      }
      tensorflow::PartialTensorShape partial_shape;
      if (shape.num_dims != -1) {
        DCHECK(shape.dim_sizes != nullptr);
        Status status = tensorflow::PartialTensorShape::MakePartialShape(
            reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims,
            &partial_shape);
        if (!status.ok()) {
          tsl::Set_TF_Status_from_Status(s, status);
          return nullptr;
    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/checkpoint_reader.h

    #ifndef TENSORFLOW_C_CHECKPOINT_READER_H_
    #define TENSORFLOW_C_CHECKPOINT_READER_H_
    
    #include <memory>
    #include <string>
    
    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/platform/status.h"
    #include "tensorflow/core/platform/types.h"
    #include "tensorflow/core/util/tensor_bundle/tensor_bundle.h"
    #include "tensorflow/core/util/tensor_slice_reader.h"
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 12 08:49:52 GMT 2023
    - 3.1K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_operation.h

    // fastbuild.
    inline Status AbstractOperation::SetAttrShape(const char* attr_name,
                                                  const PartialTensorShape shape) {
      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;
    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)
  8. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(
          std::equal(std::begin(partial_shape), std::end(partial_shape), values));
    }
    
    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)};
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  9. 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.
    //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 00:49:12 GMT 2023
    - 7.6K bytes
    - Viewed (0)
  10. 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 Status Dim(int dim_index, int64_t* dim) const = 0;
    
      // Returns the device which created the handle.
      virtual const char* DeviceName(Status* status) const = 0;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 10 21:56:24 GMT 2023
    - 4.3K bytes
    - Viewed (0)
Back to top