Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 487 for dtype (0.02 sec)

  1. tensorflow/c/checkpoint_reader.cc

      absl::Status status;
      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();
        }
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack_test.cc

      EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id);
      EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim);
      EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code);
      EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits);
      EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes);
      for (int i = 0; i < dltensor_in->ndim; ++i) {
        EXPECT_EQ(dltensor_out->shape[i], dltensor_in->shape[i]);
        if (dltensor_out->strides) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Jun 30 03:04:46 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/abstract_tensor_handle.h

      // Returns tensor (full) type.
      // While there is no immediate plan to deprecate dtype and shape in favor
      // of only using full type type information, this is a future possibility.
      //
      // Note that map_dtype_to_child_of_tensor() from core/framework/types.h
      // can be used to set a FullTypeDef based on dtype in a derived class if
      // appropriate.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 3K bytes
    - Viewed (0)
  4. tensorflow/c/c_test_util.h

        unique_tensor_ptr;
    
    TF_Tensor* BoolTensor(int32_t v);
    
    // Create a tensor with values of type TF_INT8 provided by `values`.
    TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values);
    
    // Create a tensor with values of type TF_INT32 provided by `values`.
    TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
                           const int32_t* values);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 09 01:06:53 UTC 2018
    - 6K bytes
    - Viewed (0)
  5. tensorflow/c/eager/abstract_tensor_handle.cc

        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    absl::Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.h

      void Watch(const AbstractTensorHandle*);
      // Records an operation with given inputs and outputs
      // on the tape and marks all its outputs as watched if at
      // least one input of the op is watched and has a trainable dtype.
      // op_name is optional and is used for debugging only.
      void RecordOperation(absl::Span<AbstractTensorHandle* const> inputs,
                           absl::Span<AbstractTensorHandle* const> outputs,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  7. tensorflow/c/eager/immediate_execution_tensor_handle.cc

      const char* device_name = DeviceName(&s);
      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()), ", device=\"",
                          device_name, "\")");
    }
    
    absl::Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      absl::Status status;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:24:07 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device.h

    //
    // Implicit copies onto `device_name` are allowed, replicating the value once
    // per device in `underlying_devices`. Implicit copies off of the device throw
    // an error.
    //
    // All component tensors must have the same dtype. Currently they must also have
    // the same shape, although this requirement may be relaxed in the future.
    //
    // `device_name` must not name an existing physical or custom device (see
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jun 04 21:49:16 UTC 2020
    - 2.9K bytes
    - Viewed (0)
  9. tensorflow/c/eager/custom_device_testutil.cc

      auto dtype = TFE_TensorHandleDataType(t->tensor);
      TFE_CustomDeviceTensorHandleMethods handle_methods;
      handle_methods.num_dims = &LoggedTensorNumDims;
      handle_methods.dim = &LoggedTensorDim;
      handle_methods.deallocator = &LoggedTensorDeallocator;
      return TFE_NewCustomDeviceTensorHandle(context, logging_device_name.c_str(),
                                             dtype, t.release(), handle_methods,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_unified_experimental.cc

      TF_DeleteExecutionContext(ctx);
      return wrap(func);
    }
    
    TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func,
                                               TF_DataType dtype, TF_Shape shape,
                                               TF_Status* s) {
      DCHECK_GE(shape.num_dims, -1);
      TracingTensorHandle* t;
      TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
Back to top