- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 63 for dim_idx (0.13 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) -
src/cmd/covdata/subtractintersect.go
delete(s.mm.pod.pmm, k) } } s.imm = nil } func (s *sstate) BeginCounterDataFile(cdf string, cdr *decodecounter.CounterDataReader, dirIdx int) { dbgtrace(2, "visiting counter data file %s diridx %d", cdf, dirIdx) if s.inidx != dirIdx { if s.inidx > dirIdx { // We're relying on having data files presented in // the order they appear in the inputs (e.g. first all
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed Oct 12 12:50:46 UTC 2022 - 5.5K bytes - Viewed (0) -
src/cmd/covdata/merge.go
m.mm.endPod(*pcombineflag) } func (m *mstate) BeginCounterDataFile(cdf string, cdr *decodecounter.CounterDataReader, dirIdx int) { dbgtrace(2, "visit counter data file %s dirIdx %d", cdf, dirIdx) m.mm.beginCounterDataFile(cdr) } func (m *mstate) EndCounterDataFile(cdf string, cdr *decodecounter.CounterDataReader, dirIdx int) { } func (m *mstate) VisitFuncCounterData(data decodecounter.FuncPayload) { m.mm.visitFuncCounterData(data)
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue May 23 11:36:37 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_debug.cc
TFE_TensorDebugInfo* debug_info) { return debug_info->dev_dims.size(); } TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim( TFE_TensorDebugInfo* debug_info, int dim_index) { return debug_info->dev_dims[dim_index]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 2.5K 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/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)