Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for should (0.18 sec)

  1. tensorflow/c/c_api.cc

        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;
            if ((name == prefix) || absl::StartsWith(name, prefix_cmp)) {
              status->status = InvalidArgument(
    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/experimental/gradients/nn_grad.cc

      }
      ~ReluGradientFunction() override {
        for (auto output : forward_outputs_) {
          if (output) {
            output->Unref();
          }
        }
      }
    
     private:
      // TODO(b/174778737): Only hold needed outputs.
      vector<AbstractTensorHandle*> forward_outputs_;
    };
    
    Status BroadcastMul(AbstractContext* ctx, AbstractTensorHandle* vec,
                        AbstractTensorHandle* mat,
    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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/c/eager/immediate_execution_tensor_handle.cc

                          device_name, "\")");
    }
    
    Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      Status status;
      AbstractTensorPtr resolved(
          // TODO(allenl): Resolve should be const, and the caches that get updated
          // marked mutable.
          const_cast<ImmediateExecutionTensorHandle*>(this)->Resolve(&status));
      if (!status.ok()) {
        return status;
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/array_grad_test.cc

      TF_RETURN_IF_ERROR(
          ops::IdentityN(ctx, inputs, absl::MakeSpan(temp_outputs), "IdentityN"));
      // Although, `ops::IdentityN` returns 2 tensors, the first tensor isn't needed
      // for computing gradient so we could safely drop it.
      outputs[0] = temp_outputs[1];
      temp_outputs[0]->Unref();
      return absl::OkStatus();
    }
    
    class CppGradients
        : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> {
     protected:
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem_static.cc

      TF_FilesystemPluginInfo info;
      TF_InitPlugin(&info);
      Status status = filesystem_registration::RegisterFilesystemPluginImpl(&info);
      if (!status.ok()) {
        VLOG(0) << "Static POSIX filesystem could not be registered: " << status;
        return false;
      }
      return true;
    }
    
    // Perform the actual registration
    static bool unused = StaticallyRegisterLocalFilesystems();
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 1.5K bytes
    - Viewed (0)
Back to top