- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for identity_op (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
DatasetInput GetDatasetInput(Value value) { // TODO(haoliang): add an interface for DatasetOp to avoid the following // enumeration. // Iteratively tracing upwards if parent op is `IdentityOp` or `IdentityNOp`. while ( llvm::isa_and_nonnull<IdentityOp, IdentityNOp>(value.getDefiningOp())) { value = value.getDefiningOp()->getOperand( mlir::cast<OpResult>(value).getResultNumber()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
tf_executor.yield %identity : tensor<i32> } %island_1_result, %island_1_control = tf_executor.island { %identity_n:2 = "tf.IdentityN"(%arg1, %island_0_result) : (tensor<i32>, tensor<i32>) -> (tensor<i32>, tensor<i32>) tf_executor.yield %identity_n#0 } tf_executor.fetch %island_0_result, %island_1_result : tensor<i32>, tensor<i32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
namespace mlir { namespace TF { namespace { // Returns the equivalent Value skipping through identity nodes. Value LookThroughIdentity(Value result) { while (isa_and_nonnull<IdentityOp, IdentityNOp>(result.getDefiningOp())) { auto op_result = result.cast<OpResult>(); result = op_result.getOwner()->getOperand(op_result.getResultNumber()); } return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)