Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for op_def (0.12 sec)

  1. tensorflow/c/c_api_experimental.cc

                                               TF_Status* status) {
      const tensorflow::OpDef* op_def = nullptr;
      status->status =
          tensorflow::OpRegistry::Global()->LookUpOpDef(op_name, &op_def);
      if (!status->status.ok()) return nullptr;
    
      if (input_index >= op_def->input_arg_size() || input_index < 0) {
        status->status = tensorflow::errors::InvalidArgument(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_operation.h

      // Return true if the inputs contain custom device tensor handle. It means
      // that the argument need to be handled by a custom device.
      virtual bool HasCustomDeviceInput() const = 0;
    
      virtual const tensorflow::OpDef* OpDef() const = 0;
    
      virtual absl::Status InputLength(const char* input_name, int* length) = 0;
      virtual absl::Status OutputLength(const char* output_name, int* length) = 0;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

                          TF_Buffer* output_op_def, TF_Status* status) {
      const OpDef* op_def;
      {
        mutex_lock l(graph->mu);
        status->status = graph->graph.op_registry()->LookUpOpDef(op_name, &op_def);
        if (!status->status.ok()) return;
      }
      status->status = MessageToBuffer(*op_def, output_op_def);
    }
    
    void TF_GraphVersions(TF_Graph* graph, TF_Buffer* output_version_def,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_test.cc

        ASSERT_GE(op_list.op_size(), 1);
        typedef tensorflow::protobuf::RepeatedPtrField<tensorflow::OpDef> OpDefs;
        const OpDefs& ops = op_list.op();
        bool found = std::find_if(ops.begin(), ops.end(),
                                  [](const tensorflow::OpDef& op_def) {
                                    return op_def.name() == "TestCApi";
                                  }) != ops.end();
        EXPECT_TRUE(found);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

      server_def.set_task_index(0);
      tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster();
      tensorflow::JobDef* job_def = cluster_def->add_job();
      job_def->set_name(job_name);
      for (int i = 0; i < num_tasks; i++) {
        int port = tensorflow::testing::PickUnusedPortOrDie();
        job_def->mutable_tasks()->insert(
            {i, tensorflow::strings::StrCat("localhost", ":", port)});
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Jul 10 07:18:05 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_test_util.cc

      server_def.set_task_index(0);
      tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster();
      tensorflow::JobDef* job_def = cluster_def->add_job();
      job_def->set_name(job_name);
      for (int i = 0; i < num_tasks; i++) {
        int port = tensorflow::testing::PickUnusedPortOrDie();
        job_def->mutable_tasks()->insert(
            {i, tensorflow::strings::StrCat("localhost:", port)});
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_cluster_test.cc

    namespace {
    
    using ::tensorflow::string;
    
    void ReplaceTaskInServerDef(tensorflow::ServerDef* server_def, int task_index) {
      tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0);
      int port = tensorflow::testing::PickUnusedPortOrDie();
      job_def->mutable_tasks()->at(task_index) =
          tensorflow::strings::StrCat("localhost:", port);
    }
    
    void CheckTFE_TensorHandleHasFloats(TFE_TensorHandle* handle,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  8. RELEASE.md

    *   Improve implicit broadcasting lowering.
    *   Improving stability of GCS/BigQuery clients by a faster retrying of stale
        transmissions.
    *   Remove OpKernelConstruction::op_def() as part of minimizing proto
        dependencies.
    *   VectorLaplaceDiag distribution added.
    *   Android demo no longer requires libtensorflow_demo.so to run
        (libtensorflow_inference.so still required)
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Oct 22 14:33:53 UTC 2024
    - 735.3K bytes
    - Viewed (0)
  9. .bazelrc

    compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu,tensorflow/compiler/mlir/tfrt/transforms/mlrt,tensorflow/core/runtime_fallback,tensorflow/core/runtime_fallback/conversion,tensorflow/core/runtime_fallback/kernel,tensorflow/core/runtime_fallback/opdefs,tensorflow/core/runtime_fallback/runtime,tensorflow/core/runtime_fallback/util,tensorflow/core/runtime_fallback/test,tensorflow/core/runtime_fallback/test/gpu,tensorflow/core/runtime_fallback/test/saved_model,tensorflow/core/runtime_fallback/test...
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 28 22:02:31 UTC 2024
    - 51.3K bytes
    - Viewed (0)
Back to top