Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SetAttrShapeList (0.18 sec)

  1. tensorflow/c/eager/gradients_internal.h

    absl::Status SetAttrBoolList(AbstractOperation*, const char* attr_name,
                                 const unsigned char* values, int num_values,
                                 ForwardOperation*);
    absl::Status SetAttrShapeList(AbstractOperation*, const char* attr_name,
                                  const int64_t** dims, const int* num_dims,
                                  int num_values, ForwardOperation*);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/abstract_operation.h

      virtual absl::Status SetAttrBoolList(const char* attr_name,
                                           const unsigned char* values,
                                           int num_values) = 0;
      virtual absl::Status SetAttrShapeList(const char* attr_name,
                                            const int64_t** dims,
                                            const int* num_dims,
                                            int num_values) = 0;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

      }
      forward_op_->attrs.Set(attr_name,
                             gtl::ArraySlice<const bool>(b.get(), num_values));
      return op_->SetAttrBoolList(attr_name, values, num_values);
    }
    absl::Status SetAttrShapeList(AbstractOperation* op_, const char* attr_name,
                                  const int64_t** dims, const int* num_dims,
                                  int num_values, ForwardOperation* forward_op_) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

          b[i] = values[i];
        }
        op_->node_builder.Attr(attr_name,
                               ArraySlice<const bool>(b.get(), num_values));
    
        return absl::OkStatus();
      }
      absl::Status SetAttrShapeList(const char* attr_name, const int64_t** dims,
                                    const int* num_dims, int num_values) override {
        std::vector<PartialTensorShape> shapes;
        shapes.reserve(num_values);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

                                const int64_t** dims, const int* num_dims,
                                int num_values, TF_Status* out_status) {
      out_status->status = tensorflow::unwrap(op)->SetAttrShapeList(
          attr_name, dims, num_dims, num_values);
    }
    
    void TFE_OpSetAttrFunctionList(TFE_Op* op, const char* attr_name,
                                   const TFE_Op** value, int num_values) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top