- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TPUPartitionedInputV2 (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
// CHECK: %[[READ_VAR_1:[0-9]*]] = "tf.ReadVariableOp"(%arg1) %read1 = "tf.ReadVariableOp"(%arg1) : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> // CHECK-NOT: tf.TPUPartitionedInputV2 %partitioned_input = "tf.TPUPartitionedInputV2"(%read0, %read1) {N = 2 : i64, partition_dims = []} : (tensor<i32>, tensor<i32>) -> tensor<i32> // CHECK: %[[COMPILE_OUTPUT:[0-9]*]]:3 = "tf_device.launch"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K 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/tensorflow/ir/tf_ops_n_z.cc
tensorflow::GetTypeFromTFTensorShape({}, IntegerType::get(context, 1))); return success(); } //===----------------------------------------------------------------------===// // TPUPartitionedInputV2 //===----------------------------------------------------------------------===// // This method mimics this op's core/TF-level shape inference logic LogicalResult TPUPartitionedInputV2Op::verify() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)