- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 29 for shale (0.16 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/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/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/c_api_experimental.cc
ShapeHandle shape_handle = c.output(i); TF_ShapeAndType& shape = output_shapes_result->items[i]; shape.num_dims = c.Rank(shape_handle); if (shape.num_dims == InferenceContext::kUnknownRank) { shape.dims = nullptr; continue; } shape.dims = new int64_t[shape.num_dims]; for (size_t j = 0; j < shape.num_dims; ++j) { shape.dims[j] = c.Value(c.Dim(shape_handle, j));
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/parallel_device/parallel_device_lib_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs; const std::vector<int64_t>* shape; Status s = handles[0]->Shape(&shape); ASSERT_TRUE(s.ok()); EXPECT_EQ(0, shape->size()); } TEST(PARALLEL_DEVICE_LIB, TestCancelOnError) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.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/dlpack.cc
std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; std::vector<int64_t>* stride_arr = &tf_dlm_tensor_ctx->strides; shape_arr->resize(ndim); stride_arr->resize(ndim, 1); for (int i = 0; i < ndim; i++) { (*shape_arr)[i] = tensor->dim_size(i); } for (int i = ndim - 2; i >= 0; --i) { (*stride_arr)[i] = (*shape_arr)[i + 1] * (*stride_arr)[i + 1]; }
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/c_test_util.cc
bool found_dtype = false; bool found_shape = false; for (const auto& attr : node_def.attr()) { if (attr.first == "dtype") { if (attr.second.type() == tensorflow::DT_INT32) { found_dtype = true; } else { return false; } } else if (attr.first == "shape") { found_shape = true; } } return found_dtype && found_shape; }
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/eager/gradients.cc
gtl::ArraySlice<AbstractTensorHandle*> output_gradients, absl::Span<AbstractTensorHandle*> result) const override; // Builds a tensor filled with ones with the same shape and dtype as `t`. Status BuildOnesLike(const TapeTensor& t, AbstractTensorHandle** result) const override; // Looks up the ID of a Gradient.
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/checkpoint_reader.cc
if (reader_ != nullptr) { status = reader_->GetTensor(name, out_tensor); } else { tensorflow::DataType dtype; tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape)); status = v2_reader_->Lookup(name, out_tensor->get()); if (!status.ok()) out_tensor->reset(); } } if (!status.ok()) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0)