Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for works (0.12 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      tensorflow::mutex execution_mutex_;
      ExecutionState execution_state_ TF_GUARDED_BY(execution_mutex_) =
          ExecutionState::kIdle;
      // Tells the worker thread that there is new work.
      tensorflow::condition_variable start_execute_;
      // The worker thread notifies that work has finished.
      tensorflow::condition_variable finished_execute_;
      // Notifies a StartExecute that the previous Join has finished.
      tensorflow::condition_variable finished_join_;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients.cc

        forward_op_->outputs.push_back(retvals[i]);
      }
      // TODO(b/166669239): This is needed to support AttrBuilder::Get for string
      // attributes. Number type attrs and DataType attrs work fine without this.
      // Consider getting rid of this and making the behavior between number types
      // and string consistent.
      forward_op_->attrs.BuildNodeDef();
      std::unique_ptr<GradientFunction> gradient_fn;
    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)
  3. tensorflow/c/experimental/gradients/nn_grad_test.cc

      AbstractTensorHandle* backprop;
      TF_RETURN_IF_ERROR(ops::SparseSoftmaxCrossEntropyWithLogits(
          ctx, inputs[0], inputs[1], &loss, &backprop,
          "SparseSoftmaxCrossEntropyWithLogits"));
      // `gradient_checker` only works with model that returns only 1 tensor.
      // Although, `ops::SparseSoftmaxCrossEntropyWithLogits` returns 2 tensors, the
      // second tensor isn't needed for computing gradient so we could safely drop
      // it.
      outputs[0] = loss;
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 8.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_test.cc

      TF_Status* s = TF_NewStatus();
      const char* tags[] = {tensorflow::kSavedModelTagServe};
      TF_Graph* graph = TF_NewGraph();
      // NULL run_options and meta_graph_def should work.
      TF_Session* session = TF_LoadSessionFromSavedModel(
          opt, nullptr, saved_model_dir.c_str(), tags, 1, graph, nullptr, s);
      EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      TF_DeleteSessionOptions(opt);
    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