Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for strides (0.2 sec)

  1. tensorflow/c/eager/dlpack.cc

      // 1) nullptr indicates tensor is compact and row-majored.
      // 2) fill in the strides array as the real case for compact row-major data.
      // Here we choose option 2, since some frameworks didn't handle the strides
      // argument properly.
      dlm_tensor->dl_tensor.strides = stride_arr->data();
    
      dlm_tensor->dl_tensor.byte_offset =
          0;  // TF doesn't handle the strides and byte_offsets here
      return static_cast<void*>(dlm_tensor);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K 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/eager/c_api_cluster_test.cc

      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,
                                        const std::vector<float>& expected_values) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/eager/gradients.h"
    
    #include "absl/strings/str_cat.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/eager/gradients_internal.h"
    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)
  5. tensorflow/c/eager/custom_device_test.cc

    limitations under the License.
    ==============================================================================*/
    
    // A simple logging device to test custom device registration.
    #include <memory>
    
    #include "absl/strings/match.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/eager/c_api_test_util.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  6. tensorflow/c/c_test_util.cc

    }
    
    void Split3Helper(TF_Operation* input, TF_Graph* graph, TF_Status* s,
                      const char* name, TF_Operation** op) {
      TF_Operation* zero = ScalarConst(
          0, graph, s, ::tensorflow::strings::StrCat(name, "_const0").c_str());
      TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name);
      TF_AddInput(desc, {zero, 0});
      TF_AddInput(desc, {input, 0});
      TF_SetAttrInt(desc, "num_split", 3);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  7. tensorflow/c/c_api_function.cc

    limitations under the License.
    ==============================================================================*/
    
    #include <algorithm>
    #include <unordered_map>
    #include <unordered_set>
    #include <utility>
    
    #include "absl/strings/match.h"
    #include "tensorflow/c/c_api_internal.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    #include "tensorflow/core/framework/attr_value_util.h"
    #include "tensorflow/core/framework/function.h"
    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)
  8. tensorflow/c/eager/parallel_device/parallel_device.cc

    ==============================================================================*/
    
    #include "tensorflow/c/eager/parallel_device/parallel_device.h"
    
    #include <cstring>
    #include <memory>
    
    #include "absl/strings/str_cat.h"
    #include "absl/types/optional.h"
    #include "absl/types/variant.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.h"
    
    #include <random>
    
    #include "absl/strings/string_view.h"
    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/platform/path.h"
    #include "tensorflow/core/platform/stacktrace_handler.h"
    #include "tensorflow/core/platform/test.h"
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Aug 31 12:04:23 GMT 2020
    - 24.9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_test_util.cc

        int port = tensorflow::testing::PickUnusedPortOrDie();
        job_def->mutable_tasks()->insert(
            {i, tensorflow::strings::StrCat("localhost:", port)});
      }
      auto* config = server_def.mutable_default_session_config();
      config->mutable_experimental()->set_collective_group_leader(
          tensorflow::strings::StrCat("/job:", job_name, "/replica:0/task:", 0));
      auto* rewrite_options =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
Back to top