Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 28 for shouldn (0.28 sec)

  1. tensorflow/c/c_api.cc

        string prefix_cmp;
        const char* child_scope_name;
        if (prefix == nullptr) {
          child_scope_name = "gradients";
        } else {
          prefix_cmp = string(prefix) + "/";
          // The operation should fail if the provided name prefix has already been
          // used in this graph
          for (const auto& pair : g->name_map) {
            const string& name = pair.first;
    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)
  2. tensorflow/c/experimental/gradients/array_grad.cc

                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        for (int i = 0; i < grad_outputs.size(); i++) {
          auto grad_input = grad_outputs[i];
          // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle
          // that takes care of this similar to `Tensor`?
          if (grad_input) {
            grad_input->Ref();
          }
          grad_inputs[i] = grad_input;
        }
    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)
  3. tensorflow/c/eager/c_api_cluster_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      // The device of var_handle0 is local device which is the same before and
      // after cluster update. Remove resource with valid device should succeed.
      TFE_Op* op = TFE_NewOp(ctx, "DestroyResourceOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, var_handle0, status);
      TFE_OpSetDevice(op, dev0_name, status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

          tensorflow::GetBuildXlaOpsPassFlags()->tf_xla_disable_constant_folding);
    }
    
    void TF_SetXlaConstantFoldingDisabled(unsigned char should_enable) {
      tensorflow::GetBuildXlaOpsPassFlags()->tf_xla_disable_constant_folding =
          static_cast<bool>(should_enable);
    }
    
    void TF_SetXlaMinClusterSize(int size) {
      tensorflow::MarkForCompilationPassFlags* flags =
          tensorflow::GetMarkForCompilationPassFlags();
    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)
  5. tensorflow/c/experimental/gradients/math_grad_test.cc

      }
    
      ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients(
          DivNoNanModel, DivNoNanGradModel, immediate_execution_ctx_.get(),
          {x.get(), y.get()}, UseFunction()));
    
      // `DivNoNanGradModel` should return {`0`, `0`} when the denominator is `0`.
      AbstractTensorHandlePtr z;
      {
        AbstractTensorHandle* z_raw = nullptr;
        status_ = TestScalarTensorHandle<float, TF_FLOAT>(
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

    }
    
    static void Sync(const TF_WritableFile* file, TF_Status* status) {
      // For historical reasons, this does the same as `Flush` at the moment.
      // TODO(b/144055243): This should use `fsync`/`sync`.
      Flush(file, status);
    }
    
    static void Close(const TF_WritableFile* file, TF_Status* status) {
      auto posix_file = static_cast<PosixFile*>(file->plugin_file);
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_tensor_handle.cc

                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
      // should always return OK status.
      return absl::OkStatus();
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test.cc

      char buffer[BUFFER_SIZE];
      std::strncpy(buffer, "VALID", BUFFER_SIZE);
      TFE_OpSetAttrString(op, "padding", buffer, std::strlen(buffer));
      // Overwriting value in "buffer", should be fine since TFE_Op
      // shouldn't be holding on to it.
      std::strncpy(buffer, "NHWC", BUFFER_SIZE);
      TFE_OpSetAttrString(op, "data_format", buffer, std::strlen(buffer));
    
      TFE_OpSetAttrType(op, "T", TF_FLOAT);
    
    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)
  9. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

      TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status);
      TFE_OpSetAttrString(op.get(), "container", "", 0);
      // Use the special GUID for no buffer sharing
      //
      // TODO(allenl): Should we provide a better API for this? AFAIK this is the
      // only reasonable way to make variables with no aliasing using the eager C
      // API.
      std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347";
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  10. tensorflow/c/eager/parallel_device/parallel_device.cc

        const TFE_OpAttrs* attributes, int expected_max_outputs,
        TF_Status* status) {
      absl::optional<std::vector<MaybeParallelTensorOwned>> result;
      // TODO(allenl): We should remove "TPU" from these op names at the very least,
      // or consider other ways of packing/unpacking parallel tensors.
      if (operation_name == std::string("TPUReplicatedInput")) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top