- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for num_partitions (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
LogicalResult TPUPartitionedInputV2Op::verify() { TPUPartitionedInputV2Op op = *this; int num_partitions = 1; const mlir::ArrayAttr partition_dims = op.getPartitionDims(); for (const mlir::Attribute &dim : partition_dims) { num_partitions *= dim.cast<IntegerAttr>().getInt(); } const bool is_packed = op.getIsPacked(); const bool replicated = partition_dims.empty();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)