Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_OpSetAttrIntList (0.2 sec)

  1. tensorflow/c/eager/c_api.h

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

      TFE_OpSetAttrInt(op, "instance_key", 456);
      TFE_OpSetAttrString(op, "merge_op", "Add", 3);
      TFE_OpSetAttrString(op, "final_op", "Id", 2);
      std::vector<int64_t> subdiv_offsets;
      TFE_OpSetAttrIntList(op, "subdiv_offsets", subdiv_offsets.data(),
                           subdiv_offsets.size());
    
      return op;
    }
    
    TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

      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 =
          tensorflow::unwrap(op)->SetAttrIntList(attr_name, values, num_values);
      if (!s.ok()) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_test.cc

                          merge_op.length());
      const std::string final_op("Id");
      TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(),
                          final_op.length());
      TFE_OpSetAttrIntList(op.get(), "subdiv_offsets", nullptr, 0);
    
      TFE_OpAddInput(op.get(), input, status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
    
      TFE_TensorHandle* result_handle;
      int num_retvals = 1;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
Back to top