| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | identity_n_spmd_expander.cc | 63 auto identity_op = builder.create<mlir::TF::IdentityNOp>( in ExpandOp() local
|
| D | qr_spmd_expander.cc | 80 auto identity_op = builder.create<mlir::TF::IdentityNOp>( in ExpandOp() local
|
| D | nullary_spmd_expander.cc | 103 auto identity_op = builder.create<mlir::TF::IdentityNOp>( in ExpandOp() local
|
| D | replicated_spmd_expander.cc | 85 auto identity_op = builder.create<mlir::TF::IdentityNOp>( in ReplicatedRelayoutOperandsAndOutputs() local
|
| D | dataparallel_spmd_expander.cc | 268 auto identity_op = builder.create<mlir::TF::IdentityNOp>( in RelayoutOperandsAndOutputs() local
|
| D | softmax_spmd_expander.cc | 494 mlir::TF::IdentityNOp identity_op = in MaybeRelayoutOutputs() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | restore_shape_inference.cc | 68 } else if (auto identity_op = in BackwardShapeInferenceToRestoreOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | readonly_references_to_resources.cc | 82 auto identity_op = op->getUsers().begin(); in GetNodeNameFromClassAttrOrSharedNameAttr() local
|
| D | executor_tpuv1_island_coarsening.cc | 591 if (auto identity_op = llvm::dyn_cast_or_null<TF::IdentityOp>(op)) { in EraseIdentityWithNoReplicationInfo() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | fuse_binary_into_following_affine_test.cc | 136 auto identity_op = new TensorFlowIdentityOperator; in TEST_F() local
|
| /external/ComputeLibrary/src/core/CL/cl_kernels/ |
| D | activation_float_helpers.h | 76 #define identity_op(DATA_TYPE, VEC_SIZE, x, A_VAL, B_VAL) (x) macro
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 175 TFOp identity_op = builder.create(identity_op_state); in GetControlAnchorForSwitchResult() local
252 Operation *identity_op = builder.create(state); in ReplaceOpWithIdentity() local
1978 Operation *identity_op = ReplaceReductionWithIdentity(rewriter, op); in matchAndRewrite() local
2034 Operation *identity_op = ReplaceReductionWithIdentity(rewriter, op); in matchAndRewrite() local
2099 Operation *identity_op = builder.create(state); in ReplaceReductionWithIdentity() local
2145 Operation *identity_op = rewriter.create(state); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 193 auto identity_op = [builder](xla::XlaOp op, in BuildComputation() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 2809 NodeDef identity_op; in AddFusedBatchNormExNode() local
2879 NodeDef identity_op; in AddFusedBatchNormGradExNode() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 2096 for (auto identity_op : identity_ops) { in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 2053 auto identity_op = CreateOpAndInfer<tosa::IdentityOp>( in matchAndRewrite() local
|
| D | legalize_common.cc | 2668 auto identity_op = CreateOpAndInfer<tosa::IdentityOp>( in convertReduceOpCommon() local
|
| D | legalize_tfl.cc | 2955 auto identity_op = CreateOpAndInfer<tosa::IdentityOp>( in matchAndRewrite() local
|