Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for Op (0.16 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
                                TF_Status* s) {
      TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/unified_api_testutil.h

                                       absl::Span<AbstractTensorHandle* const>,
                                       absl::Span<AbstractTensorHandle*>)>;
    
    // Runs `model` maybe wrapped in a function call op. This can be thought as
    // being equivalent to the following python code.
    //
    // if use_function:
    //   outputs = tf.function(model)(inputs)
    // else:
    //   outputs = model(inputs)
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/nn_grad.cc

    GradientFunction* ReluRegisterer(const ForwardOperation& op) {
      return new ReluGradientFunction(op.outputs);
    }
    
    GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer(
        const ForwardOperation& op) {
      return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs);
    }
    
    GradientFunction* BiasAddRegisterer(const ForwardOperation& op) {
      return new BiasAddGradientFunction(op.attrs);
    }
    
    }  // namespace gradients
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/BUILD

            "//tensorflow/c/eager:c_api",
            "//tensorflow/c/eager:c_api_experimental",
            "//tensorflow/c/eager:tfe_cancellation_manager_internal",
            "//tensorflow/c/eager:tfe_op_internal",
            "//tensorflow/c/eager:tfe_tensorhandle_internal",
            "//tensorflow/core:framework",
            "//tensorflow/core:lib",
            "@com_google_absl//absl/types:optional",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 01 20:19:06 GMT 2024
    - 4.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/tracing_utils.cc

    namespace tracing {
    
    Status MaybeSetOpName(AbstractOperation* op, const char* op_name) {
      if (isa<TracingOperation>(op)) {
        TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name));
      }
      if (isa<gradients::TapeOperation>(op)) {
        TF_RETURN_IF_ERROR(MaybeSetOpName(
            dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(),
            op_name));
      }
      return absl::OkStatus();
    }
    }  // namespace tracing
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/unified_api_testutil.cc

        }
    
        AbstractOperationPtr fn_op(ctx->CreateOperation());
        TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr));
        for (auto input : inputs) {
          TF_RETURN_IF_ERROR(fn_op->AddInput(input));
        }
        int retvals = outputs.size() - null_indices.size();
        std::vector<AbstractTensorHandle*> fn_outputs(retvals);
        TF_RETURN_IF_ERROR(fn_op->Execute(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test_util.h

    // Return an add op multiplying `a` by `b`.
    TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
    
    // Return a matmul op multiplying `a` by `b`.
    TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
    
    // Return an identity op.
    TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
    
    // Return a shape op fetching the shape of `a`.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Jul 17 23:43:59 GMT 2023
    - 7.7K bytes
    - Viewed (0)
  8. tensorflow/c/eager/gradients_test.cc

      }
    
      AbstractOperationPtr check_numerics_op(ctx->CreateOperation());
      ForwardOperation forward_op;
      Status s = Reset(check_numerics_op.get(), "CheckNumerics",
                       /*raw_device_name=*/nullptr, &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
      if (isa<TracingOperation>(check_numerics_op.get())) {
        s = dyn_cast<TracingOperation>(check_numerics_op.get())
                ->SetOpName("check_numerics");
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/array_grad.cc

          }
          grad_inputs[i] = grad_input;
        }
        return absl::OkStatus();
      }
      ~IdentityNGradientFunction() override {}
    };
    }  // namespace
    
    GradientFunction* IdentityNRegisterer(const ForwardOperation& op) {
      return new IdentityNGradientFunction;
    }
    
    }  // namespace gradients
    C++
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 1.6K bytes
    - Viewed (0)
  10. SECURITY.md

    inspected and debugged and it is intended to be used during the development
    phase.
    
    As part of the differences that make Eager mode easier to debug, the [shape
    inference
    functions](https://www.tensorflow.org/guide/create_op#define_the_op_interface)
    are skipped, and any checks implemented inside the shape inference code are not
    executed.
    
    The security impact of skipping those checks should be low, since the attack
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Sun Oct 01 06:06:35 GMT 2023
    - 9.6K bytes
    - Viewed (0)
Back to top