- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for split_sizes (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.func @testSplitVScalarInput(%input: tensor<f32>, %split_sizes: tensor<2xi32>, %split_dim: tensor<i32>) { // expected-error @+1 {{cannot split scalar input tensor}} %0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<f32>, tensor<2xi32>, tensor<i32>) -> (tensor<*xf32>, tensor<*xf32>) func.return } // ----- func.func @testSplitVNonScalarSplitDim(%input: tensor<4x4xf32>, %split_sizes: tensor<2xi32>, %split_dim: tensor<1xi32>) {
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/tf2xla/transforms/legalize_tf.cc
// and they are not necessarily the same. // // For example, given the following IR: // // %split_sizes = "tf.Const"() {value = dense<[1, -1, 3]> : tensor<3xi32>} // %split_dim = "tf.Const"() {value = dense<1> : tensor<i32>} // %0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : // (tensor<4x6xf32>, tensor<3xi32>, tensor<i32>) ->
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/tensorflow/ir/tf_ops_n_z.cc
std::optional<int64_t> dynamic_dim_index; SmallVector<int64_t, 4> split_sizes; split_sizes.reserve( split_sizes_attr.getType().cast<ShapedType>().getNumElements()); for (const auto &dim : llvm::enumerate(split_sizes_attr)) { int64_t dim_val = dim.value().getSExtValue(); split_sizes.push_back(dim_val); if (dim_val == tensorflow::kTFDynamicSize) {
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/tests/legalize-tf.mlir
func.func @splitv_dynamic(%input: tensor<?x6xf32>) -> (tensor<?x1xf32>, tensor<?x2xf32>, tensor<?x3xf32>) { %split_sizes = "tf.Const"() {value = dense<[1, 2, 3]> : tensor<3xi32>} : () -> tensor<3xi32> %split_dim = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32> // CHECK: tf.SplitV %0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<?x6xf32>, tensor<3xi32>, tensor<i32>) -> (tensor<?x1xf32>, tensor<?x2xf32>, tensor<?x3xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
src/main/java/org/codelibs/fess/thumbnail/ThumbnailManager.java
} public void setImageExtention(final String imageExtention) { this.imageExtention = imageExtention; } public void setSplitSize(final int splitSize) { this.splitSize = splitSize; } public void setThumbnailTaskQueueSize(final int thumbnailTaskQueueSize) { this.thumbnailTaskQueueSize = thumbnailTaskQueueSize; }
Registered: Wed Jun 12 13:08:18 UTC 2024 - Last Modified: Thu Feb 22 01:37:57 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
return context_op->emitOpError() << "A map_outside_compilation op's input and output shapes must " "have rank at least one and the first dimension must be known."; } int64_t split_size = in_shape[0] / num_cores_per_replica; if (in_shape[0] % num_cores_per_replica != 0) { return context_op->emitOpError() << "A map_outside_compilation op's input and output shapes must be "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0)