- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 1,187 for SplitV (0.26 sec)
-
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
} // ================== splitV ======================== LogicalResult UnrollSplitV::matchAndRewrite(TFL::SplitVOp splitv_op, PatternRewriter& rewriter) const { // We need to make sure both splits & split dim are constants. auto splits = splitv_op.getSizeSplits().getDefiningOp(); mlir::DenseIntElementsAttr splits_attr; if (!splits || !matchPattern(splits, m_Constant(&splits_attr)))
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/compiler/mlir/lite/transforms/legalize_tf.cc
DECL_CONVERT_OP(BatchMatMulV2); DECL_CONVERT_OP(BatchMatMulV3); DECL_CONVERT_OP(MatMul); DECL_CONVERT_OP(MatrixDiagV2); DECL_CONVERT_OP(MatrixDiagV3); DECL_CONVERT_OP(Pack); DECL_CONVERT_OP(Split); DECL_CONVERT_OP(SplitV); DECL_CONVERT_OP(Unpack); DECL_CONVERT_OP(Conv3D); DECL_CONVERT_OP(Conv3DBackpropInputV2); #undef DECL_CONVERT_OP // Converts any IntegerAttr to an IntegerAttr of an i32 type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: "tfl.space_to_batch_nd" } func.func @split(%arg0: tensor<i32>, %arg1: tensor<1x4x3x3xf32>) -> tensor<1x4x3xf32> { %0:3 = "tf.Split"(%arg0, %arg1) : (tensor<i32>, tensor<1x4x3x3xf32>) -> (tensor<1x4x3xf32>, tensor<1x4x3xf32>, tensor<1x4x3xf32>) func.return %0#0 : tensor<1x4x3xf32> // CHECK-LABEL: split
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// tf.Split legalization //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: @split_not_match_dynamic_split_dim_input func.func @split_not_match_dynamic_split_dim_input(%input: tensor<4x4xf32>, %split_dim: tensor<i32>) -> (tensor<*xf32>, tensor<*xf32>) { // CHECK: tf.Split
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"BroadcastArgs", "BroadcastGradientArgs", "OneHot", "Concat", "ConcatV2", "ConcatOffset", "Const", "MirrorPad", "MirrorPadGrad", "Pack", "Pad", "PadV2", "Reverse", "ReverseV2", "ReverseSequence", "Slice", "Split", "SplitV", "StridedSlice", "StridedSliceGrad", "ResourceStridedSliceAssign", "Tile", "Transpose", "InvertPermutation", "Unpack", "DeviceIndex", "TensorStridedSliceUpdate", "XlaConcatND", "XlaSplitND",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
//===----------------------------------------------------------------------===// // SplitOp //===----------------------------------------------------------------------===// // Verifies the input and split dimension operands for tf.Split/tf.SplitV. // Writes the split dimension's index (adjusted with input rank) via `dim_index` // if it's a constant. template <class Op> LogicalResult VerifySplitInputAndSplitDim(Op op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
return success(); } }; // Converts the tf.SplitV op into a series of HLO slice ops when the tensor to // be split has fully static shape and the dimension to split and split sizes // are constants. // // This is similar to the conversion for tf.Split op other than that the size of // each chunk on the dimension to split is explicitly given as an op operand // and they are not necessarily the same. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
guava/src/com/google/common/collect/Streams.java
Spliterator<? extends T> splitr = stream.spliterator(); splitrsBuilder.add(splitr); characteristics &= splitr.characteristics(); estimatedSize = LongMath.saturatedAdd(estimatedSize, splitr.estimateSize()); } return StreamSupport.stream( CollectSpliterators.flatMap( splitrsBuilder.build().spliterator(), splitr -> (Spliterator<T>) splitr,
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Thu Feb 22 21:19:52 UTC 2024 - 36.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
SameOperandsAndResultsScale]> { let summary = "Splits a tensor into `num_split` tensors along one dimension."; let description = [{ Splits the `value` tensor along `split_dim` into a number of sub-tensors with same shape as the original one, except for `split_dim`. The grouping of the resultant sub-tensors is decided by `size-splits`. Same as tf.SplitV. }]; let arguments = (ins
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)