Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 774 for dtype (0.02 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      auto outputs =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. 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)
  5. tensorflow/c/eager/c_api_test_util.cc

      TF_Status* status = TF_NewStatus();
      // Create the variable handle.
      TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op, "shape", {}, 0, status);
      TFE_OpSetAttrString(op, "container", "localhost", 0);
      TFE_OpSetAttrString(op, "shared_name", "", 0);
      if (!device_name.empty()) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  6. 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)
  7. tensorflow/c/eager/immediate_execution_context.h

      virtual AbstractTensorInterface* CreateBoolScalar(bool value) = 0;
    
      // Tensor creation functions
      virtual AbstractTensorInterface* CreateTensor(
          DataType dtype, absl::Span<const int64_t> dim_sizes) = 0;
    
      typedef void (*MemoryReleaser)(void* data, size_t len, void* arg);
    
      // Create a tensor instance from the given data buffer and description.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental_graph.cc

            operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str()));
        TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype));
        if (!shape.unknown_rank()) {
          TF_RETURN_IF_ERROR(operation->SetAttrShape(
              "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()),
              shape.dims()));
        }
        int num_outputs = 1;
        std::vector<AbstractTensorHandle*> outputs(num_outputs);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api.cc

                                   tensorflow::CustomDevice* device,
                                   tensorflow::DataType dtype, void* data,
                                   TFE_CustomDeviceTensorHandleMethods methods)
          : tensorflow::CustomDeviceTensorHandle(context, device, dtype),
            data_(data),
            methods_(methods) {}
    
      ~CAPICustomDeviceTensorHandle() override { methods_.deallocator(data_); }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_experimental.cc

      }
    
      tensorflow::AbstractTensorInterface* t =
          tensorflow::unwrap(ctx)->CreateTensor(
              static_cast<tensorflow::DataType>(dtype), dimvec);
    
      if (t == nullptr) {
        status->status =
            tensorflow::errors::InvalidArgument("Unsupported dtype: ", dtype);
        return nullptr;
      }
    
      return new TF_Tensor{t};
    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 35.9K bytes
    - Viewed (0)
Back to top