Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for enum_values (0.12 sec)

  1. tensorflow/c/eager/gradients.cc

      return op_->SetAttrTypeList(attr_name, values, num_values);
    }
    absl::Status SetAttrBoolList(AbstractOperation* op_, const char* attr_name,
                                 const unsigned char* values, int num_values,
                                 ForwardOperation* forward_op_) {
      std::unique_ptr<bool[]> b(new bool[num_values]);
      for (int i = 0; i < num_values; ++i) {
        b[i] = values[i];
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

          v.reserve(num_values);
          for (int i = 0; i < num_values; ++i) {
            v.emplace_back(static_cast<const char*>(values[i]), lengths[i]);
          }
          op_->node_builder.Attr(attr_name, v);
        }
        return absl::OkStatus();
      }
      absl::Status SetAttrFloatList(const char* attr_name, const float* values,
                                    int num_values) override {
        op_->node_builder.Attr(attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  3. tensorflow/c/c_test_util.cc

    }
    
    TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) {
      int64_t num_values = 1;
      for (int i = 0; i < num_dims; ++i) {
        num_values *= dims[i];
      }
      TF_Tensor* t =
          TF_AllocateTensor(TF_INT8, dims, num_dims, sizeof(char) * num_values);
      memcpy(TF_TensorData(t), values, sizeof(char) * num_values);
      return t;
    }
    
    TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.h

                                                       int num_values);
    TF_CAPI_EXPORT extern void TFE_OpSetAttrIntList(TFE_Op* op,
                                                    const char* attr_name,
                                                    const int64_t* values,
                                                    int num_values);
    TF_CAPI_EXPORT extern void TFE_OpSetAttrFloatList(TFE_Op* op,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

                                const float* values, int num_values) {
      auto s =
          tensorflow::unwrap(op)->SetAttrFloatList(attr_name, values, num_values);
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrIntList(TFE_Op* op, const char* attr_name,
                              const int64_t* values, int num_values) {
      auto s =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental.cc

                                   const TF_DataType* values, int num_values) {
      auto iter = builder->attr_names.insert(attr_name).first;
      builder->Set(*iter, tensorflow::gtl::ArraySlice<const tensorflow::DataType>(
                              reinterpret_cast<const tensorflow::DataType*>(values),
                              num_values));
    }
    
    void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_experimental.h

                                                         const char* attr_name,
                                                         const TF_DataType* values,
                                                         int num_values);
    
    // Checks the tensorflow::NodeDef built via the methods above to see if it can
    // run on device_type.
    TF_CAPI_EXPORT extern void TF_AttrBuilderCheckCanRunOnDevice(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
Back to top