Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 for split_at (0.15 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-BatchMatMulV2.mlir

    // CHECK:           [[CM2:%.*]] = arith.constant -2 : index
    // CHECK:           [[LHSHEAD:%.*]], [[LHSTAIL:%.*]] = "shape.split_at"([[LHSSHAPE]], [[CM2]])
    // CHECK:           [[RHSHEAD:%.*]], [[RHSTAIL:%.*]] = "shape.split_at"([[RHSSHAPE]], [[CM2]])
    // CHECK:           [[BCASTHEAD:%.*]] = shape.broadcast [[LHSHEAD]], [[RHSHEAD]]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  2. docs/debugging/xl-meta/main.go

    										}
    									}
    									reconstructAbleConfig = valid >= m.data-hasParity && valid < m.data
    									startOffset = offset
    									offset++
    									continue
    								}
    								for shardIdx, ok := range shardConfig {
    									if ok != 0 {
    										shards[shardIdx] = append(shards[shardIdx], splitData[shardIdx][offset])
    									}
    Registered: Sun Jun 16 00:44:34 UTC 2024
    - Last Modified: Fri May 31 14:49:23 UTC 2024
    - 38.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

    LogicalResult UnrollSplit::matchAndRewrite(TFL::SplitOp split_op,
                                               PatternRewriter& rewriter) const {
      auto num_splits = split_op.getNumSplits();
      auto input = split_op.getValue();
      auto input_type = mlir::dyn_cast<RankedTensorType>(input.getType());
      if (input_type == nullptr || !input_type.hasStaticShape()) return failure();
    
      for (auto result : split_op.getResults()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  4. src/os/path_unix.go

    )
    
    // IsPathSeparator reports whether c is a directory separator character.
    func IsPathSeparator(c uint8) bool {
    	return PathSeparator == c
    }
    
    // splitPath returns the base name and parent directory.
    func splitPath(path string) (string, string) {
    	// if no better parent is found, the path is relative from "here"
    	dirname := "."
    
    	// Remove all but one leading slash.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue May 07 18:44:48 UTC 2024
    - 1.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

      }
    
      // Creates a split op that splits |src_input| along |split_dimension|.
      llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type);
      *split_op = builder->create<mlir::TF::SplitOp>(
          location, output_types, split_dimension_op.getOutput(), src_input);
      (*split_op)->setAttr(
          kNumSplitAttr,
          builder->getIntegerAttr(builder->getIntegerType(32), num_split));
      return mlir::success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc

    bool NotFromDequant(mlir::Value value) {
      auto dequant_op = value.getDefiningOp<DequantizeOp>();
      if (dequant_op) {
        return false;
      }
      auto split_op = value.getDefiningOp<SplitOp>();
      if (!split_op) {
        return true;
      }
      return !split_op.getValue().getDefiningOp<DequantizeOp>();
    }
    
    // Optimize TFLite operations in functions.
    class OptimizeBatchMatmulPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc

        auto split_op = rewriter.create<TF::SplitOp>(loc, output_types,
                                                     split_dimension_op.getOutput(),
                                                     reshape_op.getOutput());
    
        // Squeeze each batch, i.e. reshape
        // [1, num_rows, num_cols] -> [num_rows, num_cols]
        for (const auto& split_value : split_op.getOutput()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/ops.mlir

      %size_splits = arith.constant dense<[]> : tensor<0xi32>
      %split_dim = arith.constant dense<0> : tensor<i32>
      // expected-error @+1 {{'tfl.split_v' op attribute 'num_splits' failed to satisfy constraint: 32-bit signless integer attribute whose value is positive}}
      "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 0 : i32} : (tensor<16xf32>, tensor<0xi32>, tensor<i32>) -> ()
      func.return
    }
    
    // -----
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/experimental/tac/tests/device-transform-gpu.mlir

      %2:3 = "tfl.split_v"(%arg0, %0, %1) {num_splits = 3 : i32} : (tensor<?x13x13x85xf32>, tensor<3xi32>, tensor<i32>) -> (tensor<*xf32>, tensor<*xf32>, tensor<*xf32>)
      func.return %2#0, %2#1, %2#2 : tensor<*xf32>, tensor<*xf32>, tensor<*xf32>
    }
    
    // CHECK-LABEL: @unrollSplitVUnknownRankResults
    // CHECK-NOT: "tfl.slice"
    // CHECK: "tfl.split_v"
    
    // -----
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  10. cmd/iam-object-store.go

    	policyDBUsersListKey    = "policydb/users/"
    	policyDBSTSUsersListKey = "policydb/sts-users/"
    	policyDBGroupsListKey   = "policydb/groups/"
    )
    
    // splitPath splits a path into a top-level directory and a child item. The
    // parent directory retains the trailing slash.
    func splitPath(s string, lastIndex bool) (string, string) {
    	var i int
    	if lastIndex {
    		i = strings.LastIndex(s, "/")
    	} else {
    		i = strings.Index(s, "/")
    	}
    Registered: Sun Jun 16 00:44:34 UTC 2024
    - Last Modified: Thu Jun 13 22:26:38 UTC 2024
    - 19.5K bytes
    - Viewed (0)
Back to top