- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 45 for dim_idx (0.11 sec)
-
tensorflow/compiler/mlir/lite/utils/utils.h
SmallVector<int32_t> shape; shape.reserve(output_shape.size()); bool needs_truncation = true; for (size_t dim_idx = 0; dim_idx < output_shape.size(); ++dim_idx) { int64_t dim = output_shape[dim_idx]; if (truncate && needs_truncation && dim == 1) { continue; } else if (needs_truncation && dim != 1) { needs_truncation = false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
for (int64_t dim_idx = 0; dim_idx < value_rank; ++dim_idx) { int64_t dim_size = value_type.getDimSize(dim_idx); if (dim_size == ShapedType::kDynamic) { Value dim_i = rewriter.create<arith::IndexCastOp>( loc, shape_scalar_type, rewriter.create<tensor::DimOp>(loc, op.getOperand(), dim_idx)); end_indices.push_back(dim_i); if (dim_idx != axis) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
auto value_shape = value_shape_type.getShape(); SmallVector<int32_t> return_squeeze_shape; return_squeeze_shape.reserve(value_shape.size()); for (size_t dim_idx = 0; dim_idx < value_shape.size(); ++dim_idx) { int64_t dim = value_shape[dim_idx]; if (dim == 1) { continue; } return_squeeze_shape.push_back( ShapedType::isDynamic(dim) ? -1 : static_cast<int32_t>(dim)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
} const int64_t rank = input_type.getRank(); IntegerAttr dim_int = ExtractSingleElementAsInteger(split_dim_attr); // "axis" operand could be a i64 tensor. Resolve it here. IntegerAttr dim_i32; if (failed(ConvertToI32Attr(dim_int, &dim_i32))) return failure(); int dim = dim_i32.getInt(); if (dim < 0) dim += rank; const SmallVector<Value, 4>& slice_outputs = SliceOutputs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/c/tf_tensor.h
// Return the number of dimensions that the tensor has. TF_CAPI_EXPORT extern int TF_NumDims(const TF_Tensor*); // Return the length of the tensor in the "dim_index" dimension. // REQUIRES: 0 <= dim_index < TF_NumDims(tensor) TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index); // Return the size of the underlying data in bytes. TF_CAPI_EXPORT extern size_t TF_TensorByteSize(const TF_Tensor*);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 16:40:30 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
int64_t dim_index = dim_index_attr.getValues<APInt>()[0].getSExtValue(); int64_t rank = operand_ty.getRank(); if (dim_index < 0 || dim_index >= rank) { return emitOptionalError(location, "dim_index (", dim_index, ") is out of range [0, ", rank, ")"); } shape[dim_index] = ShapedType::kDynamic; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
src/cmd/link/internal/dwtest/dwtest.go
// so as to find the right compilation unit. func (ex *Examiner) FileRef(dw *dwarf.Data, dieIdx int, fileRef int64) (string, error) { // Find the parent compilation unit DIE for the specified DIE. cuDie := ex.ParentCU(dieIdx) if cuDie == nil { return "", fmt.Errorf("no parent CU DIE for DIE with idx %d?", dieIdx) } // Construct a line reader and then use it to get the file string. lr, lrerr := dw.LineReader(cuDie)
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Feb 02 15:22:18 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/c/tf_tensor.cc
dims, num_dims); } int TF_NumDims(const TF_Tensor* t) { return t->tensor->NumDims(); } int64_t TF_Dim(const TF_Tensor* t, int dim_index) { return t->tensor->Dim(dim_index); } size_t TF_TensorByteSize(const TF_Tensor* t) { return t->tensor->ByteSize(); } void* TF_TensorData(const TF_Tensor* t) { return t->tensor->Data(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 21:57:32 UTC 2024 - 11.5K bytes - Viewed (0) -
cmd/endpoint.go
if err = checkCrossDeviceMounts(eps); err != nil { return nil, setupType, config.ErrInvalidErasureEndpoints(nil).Msg(err.Error()) } for diskIdx := range eps { eps[diskIdx].SetPoolIndex(poolIdx) eps[diskIdx].SetSetIndex(setIdx) eps[diskIdx].SetDiskIndex(diskIdx) } endpoints = append(endpoints, eps...) } if len(endpoints) == 0 {
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Fri May 24 23:05:23 UTC 2024 - 34.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
tensor_tile_rank--; } std::map<int, int> dimension_to_splits_map; for (int dim_index = 0; dim_index < tensor_tile_rank; ++dim_index) { if (sharding.tile_assignment_dimensions(dim_index) > 1) { dimension_to_splits_map.emplace( dim_index, sharding.tile_assignment_dimensions(dim_index)); } } if (dimension_to_splits_map.empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)