- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetReplicatedBlockArguments (0.55 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// ReplicateOp) as the last operand of TPUReshardVariablesOp. BlockArgument state = unformat_replicate.GetReplicatedBlockArguments().back(); auto replicated_block_args = unformat_replicate.GetReplicatedBlockArguments().drop_back(1); auto packed_block_args = unformat_replicate.GetPackedBlockArguments(); unformat_operands.append(replicated_block_args.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
unsigned GetNumReplicatedBlockArguments(); unsigned GetNumPackedBlockArguments(); llvm::ArrayRef<BlockArgument> GetPackedBlockArguments(); llvm::ArrayRef<BlockArgument> GetReplicatedBlockArguments(); bool IsReplicatedBlockArgument(BlockArgument block_arg); bool IsPackedBlockArgument(BlockArgument block_arg); unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
} // Returns the replicated block arguments. A copy should be made if the // replicate op is being modified. llvm::ArrayRef<BlockArgument> ReplicateOp::GetReplicatedBlockArguments() { return GetBody().getArguments().drop_back(GetNumPackedBlockArguments()); } // Returns the packed block arguments. A copy should be made if the replicate op // is being modified.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0)