Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 30 for dim_size (0.16 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. cmd/tier-last-day-stats_gen.go

    			_ = zb0008
    			s += msgp.StringPrefixSize + len(zb0007) + 1 + 5 + msgp.ArrayHeaderSize
    			for zb0009 := range zb0008.Bins {
    				s += zb0008.Bins[zb0009].Msgsize()
    			}
    			s += 10 + msgp.TimeSize
    		}
    	}
    	return
    }
    
    // DecodeMsg implements msgp.Decodable
    func (z *lastDayTierStats) DecodeMsg(dc *msgp.Reader) (err error) {
    	var field []byte
    	_ = field
    	var zb0001 uint32
    Registered: Sun Nov 03 19:28:11 UTC 2024
    - Last Modified: Thu Mar 21 17:21:35 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  9. 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)
  10. cmd/data-scanner-metric.go

    		atomic.AddUint64(&p.operations[s], 1)
    		if s < scannerMetricLastRealtime {
    			p.latency[s].add(duration)
    		}
    	}
    }
    
    // timeSize add time and size of a scanner action.
    // Use for s < scannerMetricLastRealtime
    func (p *scannerMetrics) timeSize(s scannerMetric) func(sz int) {
    	startTime := time.Now()
    	return func(sz int) {
    		duration := time.Since(startTime)
    
    		atomic.AddUint64(&p.operations[s], 1)
    Registered: Sun Nov 03 19:28:11 UTC 2024
    - Last Modified: Mon Sep 25 05:15:31 UTC 2023
    - 9.1K bytes
    - Viewed (0)
Back to top