Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for dim_size (0.15 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

              int64_t axis_length = combined_shape.dim_size(axis_index);
              if (axis_length != component_shape.dim_size(axis_index)) {
                axis_length = -1;
              }
              TF_RETURN_IF_ERROR(
                  combined_shape.SetDimWithStatus(axis_index, axis_length));
            }
          }
        }
        auto dim_sizes = combined_shape.dim_sizes();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  2. 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();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

      static char empty;
      int64_t nelems = 1;
      std::vector<int64_t> dims;
      dims.reserve(shape.dims());
      for (int i = 0; i < shape.dims(); ++i) {
        dims.push_back(shape.dim_size(i));
        nelems *= shape.dim_size(i);
      }
      CHECK_EQ(nelems, 0);
      return TF_NewTensor(
          dtype, reinterpret_cast<const int64_t*>(dims.data()), shape.dims(),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. 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();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

          }
    
          // Int
          if (const int i_size = default_value.list().i_size()) {
            absl::InlinedVector<int64_t, 4> i_vector;
            i_vector.reserve(i_size);
            for (int i = 0; i < i_size; ++i) {
              i_vector.push_back(default_value.list().i(i));
            }
            TFE_OpSetAttrIntList(op, attr_name, i_vector.data(), i_size);
          }
          // Float
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  6. cmd/bucket-metadata_gen.go

    	return
    Registered: Sun Nov 03 19:28:11 UTC 2024
    - Last Modified: Wed Aug 28 15:32:18 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  7. 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;
        }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/abstract_operation.h

    // fastbuild.
    inline absl::Status AbstractOperation::SetAttrShape(
        const char* attr_name, const PartialTensorShape shape) {
      return SetAttrShape(attr_name, shape.dim_sizes().data(), shape.dims());
    }
    
    inline absl::Status AbstractOperation::SetAttrStringList(
        const char* attr_name, absl::Span<string const> values) {
      std::vector<const char*> raw_strs;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  9. 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.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  10. cmd/object-api-datatypes_gen.go

    }
    
    // Msgsize returns an upper bound estimate of the number of bytes occupied by the serialized message
    func (z *BucketInfo) Msgsize() (s int) {
    	s = 1 + 5 + msgp.StringPrefixSize + len(z.Name) + 8 + msgp.TimeSize + 8 + msgp.TimeSize + 11 + msgp.BoolSize + 14 + msgp.BoolSize
    	return
    }
    
    // MarshalMsg implements msgp.Marshaler
    func (z *CompleteMultipartUpload) MarshalMsg(b []byte) (o []byte, err error) {
    Registered: Sun Nov 03 19:28:11 UTC 2024
    - Last Modified: Mon Jun 10 15:31:51 UTC 2024
    - 70.1K bytes
    - Viewed (0)
Back to top