Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for gradients (0.24 sec)

  1. tensorflow/c/eager/gradients.cc

    Status TapeVSpace::CallBackwardFunction(
        const string& op_type, GradientFunction* gradient_function,
        const std::vector<int64_t>& unneeded_gradients,
        gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
        absl::Span<AbstractTensorHandle*> result) const {
      if (gradient_function == nullptr) {
        return errors::InvalidArgument(
            "Provided null gradient_function for '", op_type, "'.\n",
    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)
  2. tensorflow/c/eager/tape.h

              grad_it->second.push_back(new_gradients);
              vspace.MarkAsResult(new_gradients);
            }
            out_gradients.push_back(new_gradients);
          }
        }
        VLOG(1) << "Calling gradient function for '" << trace.op_type << "'";
        std::vector<Gradient*> in_gradients(trace.input_tensor_id.size());
        DCHECK(build_default_zeros_grads || zero_indices.empty());
        if (any_gradient_nonzero) {
    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)
  3. tensorflow/c/experimental/gradients/math_grad.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/experimental/gradients/math_grad.h"
    
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/gradients.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    #include "tensorflow/c/experimental/ops/math_ops.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  4. tensorflow/c/eager/BUILD

            ":c_api_unified_internal",
            "//tensorflow/c/experimental/gradients/tape:tape_operation",
            "//tensorflow/core/lib/llvm_rtti",
            "//tensorflow/core/platform:errors",
        ],
    )
    
    cc_library(
        name = "gradients_internal",
        srcs = [
            "gradients.cc",
        ],
        hdrs = [
            "gradients.h",
            "gradients_internal.h",
        ],
        visibility = [
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 33.3K bytes
    - Viewed (0)
  5. tensorflow/c/BUILD

                ":logging",
                ":tf_status",
                ":tf_tensor",
                "//tensorflow/c/experimental/filesystem:modular_filesystem",
                "//tensorflow/cc:grad_ops",
                "//tensorflow/cc:gradients",
                "//tensorflow/cc:ops",
                "//tensorflow/cc:scope_internal",
                "//tensorflow/cc:while_loop",
                "//tensorflow/cc/saved_model:loader_lite",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_function.cc

      if (TF_GetCode(status) != TF_OK) return;
    
      tensorflow::GradientDef gdef;
      gdef.set_function_name(func->record->fdef().signature().name());
      gdef.set_gradient_func(grad->record->fdef().signature().name());
      status->status = g->graph.AddGradientDef(std::move(gdef));
    }
    
    int TF_GraphNumFunctions(TF_Graph* g) {
      tensorflow::mutex_lock l(g->mu);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  7. tensorflow/c/c_api_experimental.cc

        status->status = tensorflow::errors::Internal(
            "Invalid text proto for GraphDef: ", text_proto);
        return {};
      }
      const auto& fdef_lib = gdef.library();
      if (fdef_lib.gradient_size() > 0) {
        status->status = tensorflow::errors::Internal(
            "GradientDef is not supported in reading Dataset related functions: ",
            text_proto);
        return {};
      }
    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