- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for strides (0.17 sec)
-
tensorflow/c/eager/dlpack_test.cc
dltensor_in->dtype = {kDLFloat, 32, 1}; dltensor_in->shape = shape.data(); dltensor_in->strides = strides.data(); TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx); ASSERT_NE(handle, nullptr) << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",") << "], strides=[" << absl::StrJoin(strides, ",") << "])"; auto* dlm_out =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
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) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
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)}); LOG(INFO) << "Picked test port: " << port << " for job: " << job_name << ", task: " << i; } return server_def; } namespace tensorflow {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
==============================================================================*/ #include "tensorflow/c/eager/c_api_unified_experimental.h" #include <vector> #include "absl/container/flat_hash_map.h" #include "absl/strings/str_cat.h" #include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status_helper.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_api.cc
#include "tensorflow/c/c_api.h" #include <algorithm> #include <cstring> #include <limits> #include <memory> #include <optional> #include <unordered_set> #include <utility> #include <vector> #include "absl/strings/match.h" // Required for IS_MOBILE_PLATFORM #include "tensorflow/core/platform/platform.h" // NOLINT #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
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) -
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) -
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) -
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) -
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) -
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)