Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for We (0.18 sec)

  1. tensorflow/c/c_api_experimental.cc

      gpu_options->set_allow_growth(gpu_memory_allow_growth);
    
      (*config.mutable_device_count())["CPU"] = num_cpu_devices;
    
      // TODO(b/113217601): This is needed for EagerContext::runner_ to use a
      // threadpool, so that we avoid the possibility of running the runner_ in the
      // threadpool of GPU event mgr, as that can trigger more callbacks to be
      // scheduled on that same threadpool, causing a deadlock in cases where the
    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)
  2. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

          TF_SetStatusFromIOError(status, errno, dst);
          return;
        }
      } else if (S_ISDIR(st.st_mode)) {
        TF_SetStatus(status, TF_FAILED_PRECONDITION, "target path is a directory");
        return;
      }
    
      // We cannot rename directories yet, so prevent this.
      if (stat(src, &st) != 0) {
        TF_SetStatusFromIOError(status, errno, src);
        return;
      } else if (S_ISDIR(st.st_mode)) {
    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)
  3. tensorflow/c/c_api_function.cc

          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
            // nodes must have a single output.
            if (node->num_outputs() != 1) {
              return InvalidArgument(
    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

        }
        if (op_type_.empty()) {
          return errors::FailedPrecondition(
              "GraphOperation::Reset must be called before calling SetOpName.");
        }
        // TODO(b/145674566): We use Graph::NewName to get a unique name here but
        // this may not be consistent with python's naming policy.
        mutex_lock l(g_->mu);
        op_.reset(new TF_OperationDescription(g_, op_type_.c_str(),
    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_experimental.cc

            tensorflow::unwrap(handles[i]);
        if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) {
          // One of the inputs we're trying to pack is on a custom device. We'll let
          // the first custom device we see handle all of the packing.
          auto* custom_device_handle =
              tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>(
                  unwrapped_handle);
    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)
  6. tensorflow/c/eager/c_api.cc

        return nullptr;
      }
      std::vector<int64_t> dimvec(num_dims);
      for (int i = 0; i < num_dims; ++i) {
        dimvec[i] = static_cast<int64_t>(dims[i]);
      }
    
      // TODO(apassos) do we need to wrap the deallocator here to make sure to sync
      // the device?
      TF_ManagedBuffer* buf =
          new TF_ManagedBuffer(data, len, deallocator, deallocator_arg,
                               /*owns_memory=*/false);
    
    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)
Back to top