- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for padded_shape_splits (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
// CHECK-DAG: [[PADDED_SHAPE_SPLITS:%.+]]:6 = "tf.Split" // CHECK-DAG: [[BLOCK_SHAPE_SPLITS:%.+]]:3 = "tf.Split" // CHECK-DAG: [[OUTER_SHAPE_0:%.+]] = "tf.Div"([[PADDED_SHAPE_SPLITS]]#1, [[BLOCK_SHAPE_SPLITS]]#0) // CHECK-DAG: [[OUTER_SHAPE_1:%.+]] = "tf.Div"([[PADDED_SHAPE_SPLITS]]#2, [[BLOCK_SHAPE_SPLITS]]#1) // CHECK-DAG: [[OUTER_SHAPE_2:%.+]] = "tf.Div"([[PADDED_SHAPE_SPLITS]]#3, [[BLOCK_SHAPE_SPLITS]]#2)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
SmallVector<Type, 4> padded_shape_splits_types( input_rank, tensorflow::GetTypeFromTFTensorShape({1}, rewriter.getIntegerType(64))); SmallVector<Value, 4> padded_shape_splits( rewriter .create<SplitOp>(loc, padded_shape_splits_types, zero_i32, padded_shape_tensor) .getOutput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0)