- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 37 for num_elements (0.23 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
CompileTimeConstantOperand<TF::TensorArrayV3Op, 0>, // $size // $element_shape CompileTimeConstantOperand<TF::TensorListFromTensorOp, 1>, // $element_shape, $num_elements CompileTimeConstantOperand<TF::TensorListReserveOp, 0, 1>, // $begin, $end, $strides CompileTimeConstantOperand<TF::TensorStridedSliceUpdateOp, 1, 2, 3>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
if (src->name() != dst->name()) { Status s = CheckIfTPUInitialized(); if (!s.ok()) { done(s); return absl::OkStatus(); } } if (input->shape().num_elements() == 0) { // Zero-element tensors have no backing buffers. done(absl::OkStatus()); return absl::OkStatus(); } se::Stream* const src_compute_stream = src_xla_context->stream();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
// We can create an MLIR Attribute more efficiently in this case. TensorShape input_tensor_shape(input_tensor.tensor_shape()); if (NumberOfMaterializedElements(input_tensor) == 1 && input_tensor_shape.num_elements() > 1) { // We first convert this TensorProto to one of shape [1]. We then create an // Attribute for that proto, and finally splat the Attribute. TensorProto tensor_copy = input_tensor;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/testdata/outside_compilation.mlir
%outputs_55, %control_56 = tf_executor.island wraps "tf.TensorListStack"(%outputs_53#3, %outputs_27) {_tpu_replicate = "cluster_sample_sequence", device = "", num_elements = 1024 : i64} : (tensor<!tf_type.variant>, tensor<1xi32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 19 20:19:45 UTC 2023 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
// Construct the tensor for the given type and buffer. static Tensor MakeTensor(DataType dtype, const TensorShape& shape, se::DeviceMemoryBase buffer, Allocator* allocator) { size_t expected_size = shape.num_elements() * DataTypeSize(dtype); auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, buffer.size(), allocator); Tensor t(dtype, shape, tensor_buffer);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir
%cst = arith.constant dense<3> : tensor<1xi32> %0 = "tf.TensorListFromTensor"(%arg0, %cst) : (tensor<?x3xf32>, tensor<1xi32>) -> tensor<!tf_type.variant<tensor<3xf32>>> %1 = "tf.TensorListStack"(%0, %cst) {num_elements = 2 : i64} : (tensor<!tf_type.variant<tensor<3xf32>>>, tensor<1xi32>) -> tensor<2x3xf32> func.return %1 : tensor<2x3xf32> // CHECK: [[ELEM_SHAPE:%cst.*]] = arith.constant dense<3> : tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 39.9K bytes - Viewed (0) -
guava/src/com/google/common/collect/Queues.java
int added = 0; while (added < numElements) { // we could rely solely on #poll, but #drainTo might be more efficient when there are multiple // elements already available (e.g. LinkedBlockingQueue#drainTo locks only once) added += q.drainTo(buffer, numElements - added); if (added < numElements) { // not enough elements immediately available; will have to poll
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Mon Apr 01 16:15:01 UTC 2024 - 18K bytes - Viewed (0) -
android/guava/src/com/google/common/collect/Queues.java
int added = 0; while (added < numElements) { // we could rely solely on #poll, but #drainTo might be more efficient when there are multiple // elements already available (e.g. LinkedBlockingQueue#drainTo locks only once) added += q.drainTo(buffer, numElements - added); if (added < numElements) { // not enough elements immediately available; will have to poll
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Mon Apr 01 16:15:01 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/c/tf_tensor.cc
} #endif // LIBTPU_EXCLUDE_C_API_IMPL namespace tensorflow { void TensorInterface::Release() { if (Type() == DT_STRING && NumElements() > 0) { TF_TString* data = static_cast<TF_TString*>(Data()); if (CanMove() && data != nullptr) { for (int64_t i = 0; i < NumElements(); ++i) { TF_TString_Dealloc(&data[i]); } } } delete this; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 21:57:32 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0)