Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for things (0.2 sec)

  1. tensorflow/c/c_api.cc

                        const char* mutation_type) {
      // If any session has already run this node_id, mark this session as
      // unrunnable.
      for (auto it : graph->sessions) {
        mutex_lock session_lock(it.first->mu);
        if (it.first->last_num_graph_nodes > op.node.id()) {
          it.second = strings::StrCat(
              "Operation '", op.node.DebugString(), "' was changed by ",
              mutation_type,
    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/c_api_experimental.cc

          session->graph,
          tensorflow::strings::StrCat("fifo_queue_enqueue_", tensor_id).c_str());
      if (enqueue_op == nullptr) {
        status->status = tensorflow::errors::Internal(
            "Unable to find the enqueue node in the TF graph.");
        return;
      }
    
      TF_Operation* placeholder_op = TF_GraphOperationByName(
          session->graph,
          tensorflow::strings::StrCat("arg_tensor_enqueue_", tensor_id).c_str());
    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)
  3. tensorflow/c/c_api_function.cc

        for (const Node* node : fn_body->graph.op_nodes()) {
          const auto& iter = input_nodes.find(node);
          if (iter == input_nodes.end()) {
            // This node is not referenced in inputs. Add it to the body.
            body_nodes->push_back(node);
          } else {
            // This node is referenced in inputs. Currently, we place an
            // artificial restriction and require that when num_opers=-1, such
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

    limitations under the License.
    ==============================================================================*/
    
    #include <memory>
    #include <utility>
    #include <vector>
    
    #include "absl/strings/str_cat.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/abstract_context.h"
    #include "tensorflow/c/eager/c_api_internal.h"
    #include "tensorflow/c/eager/c_api_unified_experimental.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  5. tensorflow/c/eager/c_api.cc

          const ImmediateExecutionOperation* op) override {
        TF_Status status;
        // Let this custom device choose the device to pin this op on if it
        // implements the pinning function.
        if (device_.shall_pin_to_this_device != nullptr) {
          return device_.shall_pin_to_this_device(tensorflow::wrap(op), &status);
        }
        return errors::Unimplemented("No custom device pinning implementation.");
      }
    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

    #include "tensorflow/c/eager/c_api_experimental.h"
    
    #include <cstdint>
    #include <memory>
    #include <utility>
    #include <vector>
    
    #include "absl/container/flat_hash_map.h"
    #include "absl/strings/match.h"
    #include "absl/time/time.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api_internal.h"
    #include "tensorflow/c/eager/tfe_context_internal.h"
    #include "tensorflow/c/eager/tfe_op_internal.h"
    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

    TEST(CAPI, Session_Min_GPU) {
      const string gpu_device = GPUDeviceName();
      // Skip this test if no GPU is available.
      if (gpu_device.empty()) return;
    
      RunMinTest(gpu_device, /*use_XLA=*/false);
    }
    
    TEST(CAPI, Session_Min_XLA_GPU) {
      const string gpu_device = GPUDeviceName();
      // Skip this test if no GPU is available.
      if (gpu_device.empty()) return;
    
    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