Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TFE_OpSetAttrIntList (0.25 sec)

  1. 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()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  2. 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);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  3. 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;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  4. 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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  5. tensorflow/c/eager/c_api_test.cc

      TFE_OpAddInput(op, tensor_handle, status);
      TF_DeleteTensor(tensor);
      TFE_DeleteTensorHandle(tensor_handle);
    
      std::vector<int64_t> values(4, 1);
      TFE_OpSetAttrIntList(op, "ksize", values.data(), values.size());
      TFE_OpSetAttrIntList(op, "strides", values.data(), values.size());
    
      const int BUFFER_SIZE = 10;
      char buffer[BUFFER_SIZE];
      std::strncpy(buffer, "VALID", BUFFER_SIZE);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top