- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 316 for shape (0.15 sec)
-
tensorflow/c/c_api_experimental.h
// if you want shape inference to consider the input tensors of the // op for shape inference. // - The types need not be set in `input_shapes` as it is not used. // - The number of `input_tensors` should be the same as the number of items // in `input_shapes`. // // The results are returned in `output_shapes` and // `output_resource_shapes_and_types`. The caller is responsible for freeing the
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K 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/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/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/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) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
std::vector<PartialTensorShape> shapes; shapes.reserve(num_values); for (int i = 0; i < num_values; ++i) { if (num_dims[i] < 0) { shapes.emplace_back(); } else { shapes.emplace_back(ArraySlice<int64_t>( reinterpret_cast<const int64_t*>(dims[i]), num_dims[i])); } } op_->node_builder.Attr(attr_name, shapes); return absl::OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
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/eager/c_api_debug.cc
std::vector<int64_t> shape; int rank = -1; *status = handle.NumDims(&rank); if (!status->ok()) { return shape; } shape.reserve(rank); for (int i = 0; i < rank; ++i) { int64_t dim; *status = handle.Dim(i, &dim); if (!status->ok()) { return shape; } shape.push_back(dim); } return shape; } } // namespace extern "C" {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 2.5K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.h
virtual tensorflow::Status TensorHandleStatus() const; // Returns tensor shape. If tensor has unknown rank, shape remains untouched. virtual tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const = 0; // Returns tensor (full) type. // While there is no immediate plan to deprecate dtype and shape in favor // of only using full type type information, this is a future possibility. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 03 00:30:36 GMT 2023 - 3K bytes - Viewed (0) -
tensorflow/c/eager/abstract_operation.h
// fastbuild. inline Status AbstractOperation::SetAttrShape(const char* attr_name, const PartialTensorShape shape) { return SetAttrShape(attr_name, shape.dim_sizes().data(), shape.dims()); } inline Status AbstractOperation::SetAttrStringList( const char* attr_name, absl::Span<string const> values) { std::vector<const char*> raw_strs;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0)