- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for dim_idx (0.1 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/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) -
src/syscall/fs_js.go
return 0, err } if f.entries == nil { return 0, EINVAL } n := 0 for f.dirIdx < len(f.entries) { entry := f.entries[f.dirIdx] l := 2 + len(entry) if l > len(buf) { break } buf[0] = byte(l) buf[1] = byte(l >> 8) copy(buf[2:], entry) buf = buf[l:] n += l f.dirIdx++ } return n, nil } func setStat(st *Stat_t, jsSt js.Value) {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu May 11 18:19:17 UTC 2023 - 10.8K 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) -
src/cmd/covdata/dump.go
if d.cmd == debugDumpMode { d.cm.ResetModeAndGranularity() } } func (d *dstate) BeginCounterDataFile(cdf string, cdr *decodecounter.CounterDataReader, dirIdx int) { dbgtrace(2, "visit counter data file %s dirIdx %d", cdf, dirIdx) if d.cmd == debugDumpMode { fmt.Printf("data file %s", cdf) if cdr.Goos() != "" { fmt.Printf(" GOOS=%s", cdr.Goos()) } if cdr.Goarch() != "" {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue May 23 11:36:57 UTC 2023 - 11.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims( TFE_TensorDebugInfo* debug_info); // Returns the number of elements in dimension `dim_index`. // Tensor representation on device can be transposed from its representation // on host. The data contained in dimension `dim_index` on device // can correspond to the data contained in another dimension in on-host // representation. The dimensions are indexed using the standard TensorFlow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
CompileTimeConstantOperand<TF::XlaReduceWindowOp, 2, 3, 4, 5, 6>, // $dim_index CompileTimeConstantOperand<TF::XlaRemoveDynamicDimensionSizeOp, 1>, // $window_dimensions, $window_strides, $padding CompileTimeConstantOperand<TF::XlaSelectAndScatterOp, 1, 2, 3>, CompileTimeConstantOperand<TF::XlaSetBoundOp, 1>, // $bound // $dim_index CompileTimeConstantOperand<TF::XlaSetDynamicDimensionSizeOp, 1>
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/c/eager/c_api.cc
return num_elements; } int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return -1; } int64_t dim = -1; status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim); return dim; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)