- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for shale (8.23 sec)
-
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/immediate_execution_tensor_handle.cc
namespace tensorflow { std::string ImmediateExecutionTensorHandle::DebugString() const { PartialTensorShape shape; std::string shape_string; if (Shape(&shape).ok()) { shape_string = shape.DebugString(); } else { shape_string = "<error computing shape>"; } std::string value_string; if (!SummarizeValue(value_string).ok()) { value_string = "<error computing value>"; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K 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/gradient_checker.cc
AbstractTensorHandlePtr sum_dims; { vector<int32_t> vals(num_dims_out); int64_t vals_shape[] = {num_dims_out}; Range(&vals, 0, num_dims_out); AbstractTensorHandle* sum_dims_raw = nullptr; TF_RETURN_IF_ERROR(TestTensorHandleWithDims<int32_t, TF_INT32>( ctx, vals.data(), vals_shape, 1, &sum_dims_raw)); sum_dims.reset(sum_dims_raw); } // Reduce sum the output on all dimensions.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
absl::Span<AbstractTensorHandle*> outputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(inputs[0]->Shape(&shape)); if (shape.dims() != 0) { return errors::InvalidArgument( "Tensor expected to have scalar shape found rank: ", shape.dims()); } return absl::OkStatus(); } TEST_P(UnifiedAPI, TestTensorShapeScalar) { if (UseFunction() && UseMlir()) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
tracing::TracingTensorHandle* handle = nullptr; for (auto input : inputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(input->Shape(&shape)); TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( input->DataType(), shape, &handle)); params->emplace_back(handle); } return absl::OkStatus(); } // Runs `model` maybe wrapped in a function.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(tensor); if (av_tensor == nullptr) { TF_ASSIGN_OR_RETURN( *tensor, MakeTensorFromPjRtBuffer(tensor->dtype(), tensor->shape(), std::move(buffer))); } else { av_tensor->SetBuffer(std::move(buffer)); } return absl::OkStatus(); } absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0)