Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for Type (0.43 sec)

  1. tensorflow/c/c_api.cc

                metadata.type = TF_ATTR_INT;
              } else if (typestr == "list(float)") {
                metadata.type = TF_ATTR_FLOAT;
              } else if (typestr == "list(bool)") {
                metadata.type = TF_ATTR_BOOL;
              } else if (typestr == "list(type)") {
                metadata.type = TF_ATTR_TYPE;
              } else if (typestr == "list(shape)") {
                metadata.type = TF_ATTR_SHAPE;
    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/c_api_unified_experimental.cc

                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

    }
    
    void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder,
                                           const char* device_type,
                                           TF_Status* status) {
      status->status = tensorflow::FindKernelDef(
          tensorflow::DeviceType(device_type), builder->BuildNodeDef(),
          /* def = */ nullptr, /* kernel_class_name = */ nullptr);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

        if (n > INT32_MAX)
          requested_read_length = INT32_MAX;
        else
          requested_read_length = n;
    
        // `pread` returns a `ssize_t` on POSIX, but due to interface being
        // cross-platform, return type of `Read` is `int64_t`.
        int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length,
                                  static_cast<off_t>(offset))};
        if (r > 0) {
          dst += r;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack.cc

      std::string device_type = parsed_name.type;
      int device_id = 0;
      if (parsed_name.has_id) {
        device_id = parsed_name.id;
      }
    
      ctx.device_id = device_id;
      if (device_type == "CPU") {
        ctx.device_type = DLDeviceType::kDLCPU;
      } else if (device_type == "GPU") {
    #if TENSORFLOW_USE_ROCM
        ctx.device_type = DLDeviceType::kDLROCM;
    #else
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
    
      // Calls the passed-in backward function.
      // op_type is the op's name provided in RecordOperation.
      Status CallBackwardFunction(
          const string& op_type, GradientFunction* gradient_function,
          const std::vector<int64_t>& unneeded_gradients,
          gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_test.cc

      TF_OperationDescription* init(string type) {
        // Construct op_name to match the name used by REGISTER_OP in the
        // ATTR_TEST_REGISTER calls above.
        string op_name = "CApiAttributesTestOp";
        if (type.find("list(") == 0) {
          op_name += "List";
          type = type.replace(0, 5, "");
          type = type.replace(type.size() - 1, 1, "");
        }
        op_name += type;
        return TF_NewOperation(
    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)
  8. tensorflow/c/eager/c_api_distributed_test.cc

          "    signature {"
          "      name: 'AddVariablesFunction'"
          "      input_arg {"
          "        name: 'var'"
          "        type: DT_RESOURCE"
          "      }"
          "      output_arg {"
          "        name: 'sum'"
          "        type: DT_FLOAT"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'read0'"
          "      op: 'ReadVariableOp'"
          "      input: 'var'"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  9. tensorflow/c/eager/abstract_tensor_handle.cc

      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
      // should always return OK status.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  10. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      CancellationManager* cancellation_manager_ TF_GUARDED_BY(execution_mutex_);
      //   Outputs:
      std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_);
      // TF_Status is an incomplete type and so can't be stack allocated. To avoid
      // unnecessary allocations each Execute call, we keep one heap-allocated
      // version for the thread.
      StatusPtr status_ TF_GUARDED_BY(execution_mutex_);
    
    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)
Back to top