- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for Escape (0.17 sec)
-
tensorflow/c/c_api.cc
if (shapes_and_types == nullptr) return nullptr; for (const auto& p : *shapes_and_types) { auto* out_shape_and_type = handle_data.add_shape_and_type(); ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); out_shape_and_type->set_dtype(p.dtype); *out_shape_and_type->mutable_type() = p.type; } } string str_data; handle_data.SerializeToString(&str_data);
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/dlpack_test.cc
dltensor_in->device = {kDLCPU, 0}; dltensor_in->ndim = static_cast<int32_t>(shape.size()); 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, ",") << "])";
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/c_api_unified_experimental.cc
return nullptr; } tensorflow::PartialTensorShape partial_shape; if (shape.num_dims != -1) { DCHECK(shape.dim_sizes != nullptr); Status status = tensorflow::PartialTensorShape::MakePartialShape( reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims, &partial_shape); if (!status.ok()) { tsl::Set_TF_Status_from_Status(s, status); return nullptr;
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/eager/c_api_debug_test.cc
TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); ASSERT_EQ(2, TFE_TensorDebugInfoOnDeviceNumDims(debug_info)); // Shape is the same for CPU tensors. EXPECT_EQ(3, TFE_TensorDebugInfoOnDeviceDim(debug_info, 0)); EXPECT_EQ(2, TFE_TensorDebugInfoOnDeviceDim(debug_info, 1)); TFE_DeleteTensorDebugInfo(debug_info); TFE_DeleteTensorHandle(h);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 06 22:10:09 GMT 2020 - 2.3K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.cc
namespace tensorflow { std::string AbstractTensorHandle::DebugString() const { PartialTensorShape shape; Status s = Shape(&shape); std::string shape_string; if (!s.ok()) { shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string, ", dtype=", DataType_Name(DataType()),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); 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++ - 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_lib.cc
if (combined_shape.dims() < 0 || combined_shape.dims() != component_shape.dims()) { PartialTensorShape first_shape; TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&first_shape)); return errors::Unimplemented(absl::StrCat( "Computing the shape of a ParallelTensor when the components do " "not all have the same rank is not supported. One tensor had " "shape ",
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/experimental/grappler/grappler_test.cc
TF_ASSERT_OK(s); EXPECT_EQ(DT_FLOAT, in_props.dtype()); EXPECT_FALSE(in_props.shape().unknown_rank()); EXPECT_EQ(2, in_props.shape().dim_size()); EXPECT_EQ(10, in_props.shape().dim(0).size()); EXPECT_EQ(1, in_props.shape().dim(1).size()); for (int i = 0; i < in_props_buf.size(); i++) TF_DeleteBuffer(in_props_buf[i]); } }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
device->arrived_flag = arrived_flag; device->executed_flag = executed_flag; device->device_name = name; device->underlying_device = "/job:localhost/replica:0/task:0/device:CPU:0"; device->strict_scope_placement = strict_scope_placement; TFE_RegisterCustomDevice(context, custom_device, name, device, status); } TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// number of dimensions of a parallel tensor. int ParallelTensorNumDims(void* data, TF_Status* status) { const std::vector<int64_t>* shape; Status s = reinterpret_cast<ParallelTensor*>(data)->Shape(&shape); if (!s.ok()) { tsl::Set_TF_Status_from_Status(status, s); return -1; } return shape->size(); } // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing a // dimension of a parallel tensor.
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)