- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for partition_dim (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
int rank = tensor_type.getRank(); if (rank <= partition_dim) { return op->emitError() << "cannot partition " << first_operand_type << " (rank = " << rank << ") along dimension " << partition_dim << "."; } llvm::SmallVector<int64_t, 4> partition_dims(is_replicated ? 0 : rank, 1); if (!is_replicated) { partition_dims[partition_dim] = num_cores_per_replica.getInt(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_partitioned_op_conversion.mlir
// CHECK-SAME: _XlaSharding = "" // CHECK-SAME: partition_dims = [] // CHECK: [[PI_1:%.*]] = "tf.TPUPartitionedInputV2"([[ARG2]], [[ARG3]]) // CHECK-SAME: _XlaSharding = "" // CHECK-SAME: partition_dims = [] // CHECK: [[RI:%.*]] = "tf.TPUReplicatedInput"([[PI_0]], [[PI_1]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 20 17:43:51 UTC 2023 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
// Abort if TPUPartitionedInputV2(s) do not have the same attributes. if (!llvm::equal(partition_dims, op_partition_dims)) { return partitioned_input->emitOpError() << "expects partition_dims = " << partition_dims << " but found " << op_partition_dims; } else if (partitioned_input.getIsPacked() != first_partitioned_input.getIsPacked()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0)