- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 1,929 for SplitV (0.12 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/cc/framework/fuzzing/op_fuzzing.bzl
"ReverseV2", "ScatterNdNonAliasingAdd", "Shape", "ShapeN", "Size", "Slice", "Snapshot", "SpaceToBatch", "SpaceToBatchND", "SpaceToDepth", "Split", "SplitV", "Squeeze", "StopGradient", "StridedSlice", "StridedSliceGrad", "TensorScatterAdd", "TensorScatterMax", "TensorScatterMin", "TensorScatterSub",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 07 19:14:57 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/opens2s_gnmt_mixed_precision.golden_summary
Const 81 ExpandDims 3 Fill 1 GreaterEqual 8 Identity 1 Less 1 MatMul 10 Mul 44 Range 1 Rsqrt 1 Select 19 Shape 6 Sigmoid 24 Snapshot 8 Softmax 1 Split 8 SplitV 6 Square 1 Squeeze 1 StridedSlice 1 Sum 2 Tanh 17 cluster 12 size 6 Add 1 All 1 Const 2 GreaterEqual 1 LogicalOr 1 cluster 15 size 614 Add 22 AddN 41
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
return scope.status(); } REGISTER_GRADIENT_OP("Split", SplitGrad); Status SplitVGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { if (op.num_inputs() < 3) { return errors::InvalidArgument("SplitV requires 3 arguments"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
} TEST_F(ArrayGradTest, SplitGrad) { TensorShape x_shape({5, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); // Split along the second dimension. auto split_dim = Const(scope_, 1, {}); auto y = Split(scope_, split_dim, x, /* num_split */ 2); TensorShape y_shape = TensorShape({5, 1}); RunTest({x}, {x_shape}, y.output, {y_shape, y_shape}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K 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)