- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for setIsPacked (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
operands_per_replica, replicated_input->getAttrs()); replicate_op.setIsPacked(is_packed); operands_per_core.push_back(replicate_op); } auto pi = builder.create<TF::TPUPartitionedInputV2Op>( first_partitioned_input.getLoc(), replicated_input.getType(), operands_per_core, first_partitioned_input->getAttrs()); pi.setIsPacked(false); // inputs are now ops--not resources
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0)