Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for shouldn (0.27 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/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)
  4. 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)
  5. tensorflow/c/eager/c_api.cc

    #endif  // !IS_MOBILE_PLATFORM
      return tensorflow::wrap(eager_context);
    }
    
    void TFE_DeleteContext(TFE_Context* ctx) {
      if (ctx == nullptr) {
        return;
      }
    
      // ctx->RefCountIsOne() should be true here.
      tensorflow::unwrap(ctx)->Release();
    }
    
    TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) {
      TF_DeviceList* l = new TF_DeviceList;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  6. tensorflow/c/eager/c_api_experimental.cc

            tensorflow::unwrap(op_to_reset);
        op->Clear();
        status->status = op->Reset(op_or_function_name, raw_device_name);
      } else {
        TF_SetStatus(status, TF_INVALID_ARGUMENT,
                     "op_to_reset should not be nullptr");
      }
    }
    
    void TFE_ContextEnableGraphCollection(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true);
    }
    
    void TFE_ContextDisableGraphCollection(TFE_Context* ctx) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  7. tensorflow/c/c_api_test.cc

    }
    
    void NoOpDeallocator(void* data, size_t, void*) {}
    
    TEST(CAPI, MalformedTensor) {
      // See https://github.com/tensorflow/tensorflow/issues/7394
      // num_dims = 0 implies a scalar, so should be backed by at least 4 bytes of
      // data.
      TF_Tensor* t =
          TF_NewTensor(TF_FLOAT, nullptr, 0, nullptr, 0, &NoOpDeallocator, nullptr);
      ASSERT_TRUE(t == nullptr);
    }
    
    TEST(CAPI, AllocateTensor) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top