- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for strides (0.15 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/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer( data.data(), shape.element_type(), shape.dimensions(), /*byte_strides=*/std::nullopt, xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, nullptr, c_api_client->pjrt_c_client()->client->addressable_devices()[0]); CHECK_OK(buffer.status());
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K 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/eager/c_api_unified_experimental_eager.cc
using tensorflow::dyn_cast; using tensorflow::ImmediateExecutionContext; using tensorflow::ImmediateExecutionTensorHandle; using tensorflow::string; using tensorflow::unwrap; using tensorflow::wrap; using tensorflow::strings::StrCat; TF_ExecutionContext* TF_NewEagerExecutionContext(TFE_ContextOptions* options, TF_Status* s) { TFE_Context* c_ctx = TFE_NewContext(options, s);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 25 04:40:46 GMT 2020 - 3.2K 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_remote_test.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "absl/strings/str_cat.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/c_api_internal.h" #include "tensorflow/c/eager/c_api_remote_test_util.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.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)