Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for ArraySlice (0.2 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.cc

      forward_op_.attrs.Set(attr_name,
                            gtl::ArraySlice<const float>(values, num_values));
      return parent_op_->SetAttrFloatList(attr_name, values, num_values);
    }
    Status TapeOperation::SetAttrIntList(const char* attr_name,
                                         const int64_t* values, int num_values) {
      forward_op_.attrs.Set(
          attr_name, gtl::ArraySlice<const int64_t>(
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        op_->node_builder.Attr(attr_name,
                               ArraySlice<const float>(values, num_values));
        return absl::OkStatus();
      }
      Status SetAttrIntList(const char* attr_name, const int64_t* values,
                            int num_values) override {
        op_->node_builder.Attr(
            attr_name, ArraySlice<const int64_t>(
                           reinterpret_cast<const int64_t*>(values), num_values));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  3. tensorflow/c/eager/gradients.cc

      AbstractTensorHandle* AggregateGradients(
          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)
  4. tensorflow/c/eager/tape.h

      virtual int64_t NumElements(Gradient* tensor) const = 0;
    
      // Consumes references to the tensors in the gradient_tensors list and returns
      // a tensor with the result.
      virtual Gradient* AggregateGradients(
          gtl::ArraySlice<Gradient*> gradient_tensors) const = 0;
    
      // Calls the passed-in backward function.
      //
      // `unneeded_gradients` contains sorted list of input indices for which a
      // gradient is not required.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  5. tensorflow/c/c_api.cc

    using tensorflow::TensorShapeProto;
    using tensorflow::VersionDef;
    using tensorflow::errors::FailedPrecondition;
    using tensorflow::errors::InvalidArgument;
    using tensorflow::errors::OutOfRange;
    using tensorflow::gtl::ArraySlice;
    using tensorflow::strings::StrCat;
    
    extern "C" {
    
    // --------------------------------------------------------------------------
    const char* TF_Version() { return TF_VERSION_STRING; }
    
    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)
  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,
    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)
Back to top