- Sort Score
- Num 10 results
- Language All
Results 1 - 5 of 5 for dim_sizes (0.05 seconds)
-
tensorflow/c/eager/immediate_execution_context.h
virtual AbstractTensorInterface* CreateBoolScalar(bool value) = 0; // Tensor creation functions virtual AbstractTensorInterface* CreateTensor( DataType dtype, absl::Span<const int64_t> dim_sizes) = 0; typedef void (*MemoryReleaser)(void* data, size_t len, void* arg); // Create a tensor instance from the given data buffer and description.
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 12 05:11:17 GMT 2024 - 12.3K bytes - Click Count (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); if (!shape.unknown_rank()) { TF_RETURN_IF_ERROR(operation->SetAttrShape( "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()), shape.dims())); } int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs); TF_RETURN_IF_ERROR(operation->Execute(
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat May 31 07:13:41 GMT 2025 - 15.7K bytes - Click Count (0) -
tensorflow/c/eager/dlpack.cc
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]; } dlm_tensor->dl_tensor.shape = shape_arr->data();
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Mar 13 23:41:52 GMT 2025 - 13K bytes - Click Count (0) -
tensorflow/c/c_api_experimental.cc
status->status = InvalidArgument("Expected rank is ", num_dims, " but actual rank is ", rank); return; } for (int i = 0; i < num_dims; i++) { dims[i] = shape.dim_size(i); } } int TF_CheckpointReaderGetVariableNumDims(TF_CheckpointReader* reader, const char* name) { const auto& m = reader->GetVariableToShapeMap();
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 04 05:55:32 GMT 2025 - 29.4K bytes - Click Count (0) -
tensorflow/c/eager/c_api.cc
const auto& tensor_shape = default_value.shape(); if (tensor_shape.unknown_rank()) { TFE_OpSetAttrShape(op, attr_name, nullptr, -1, status); } else { const auto num_dims = tensor_shape.dim_size(); std::unique_ptr<int64_t[]> dims(new int64_t[num_dims]); for (int i = 0; i < num_dims; ++i) { dims[i] = tensor_shape.dim(i).size(); }
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Fri Nov 07 05:55:21 GMT 2025 - 43.9K bytes - Click Count (0)