Searched refs:getNonControlOperands (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/ |
| H A D | shape_inference_utils_test.cc | 147 op.getLoc(), &op, TFOp(&op).getNonControlOperands(), in TEST_F()
185 op.getLoc(), &op, TFOp(&op).getNonControlOperands(), in TEST_F()
226 op.getLoc(), &op, TFOp(&op).getNonControlOperands(), in TEST_F()
248 op.getLoc(), &op, TFOp(&op).getNonControlOperands(), in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| H A D | pass.cc | 104 TFOp(op).getNonControlOperands().size() == 1)) { in TryToCacheResultsTensorValue()
196 op->getLoc(), op, TFOp(op).getNonControlOperands(), graph_version_, in runOnOperation()
320 TFOp(return_op).getNonControlOperands().getTypes()))); in runOnOperation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | tf_op_wrapper.h | 81 OperandRange getNonControlOperands() { return splitOperands().first; } in getNonControlOperands() function
|
| H A D | tf_op_wrapper_test.cc | 80 OperandRange data = a_op.getNonControlOperands(); in TEST()
|
| H A D | ops.cc | 235 OperandRange data = tfg_op.getNonControlOperands(); in printCustomTfOp()
830 OperandRange data = tfg_op.getNonControlOperands(); in print()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/const_dedupe_hoist/ |
| H A D | pass.cc | 139 user->eraseOperands(tf_user.getNonControlOperands().size(), in PropagateEdges()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 218 OperandControlRetRange(op.getNonControlOperands())); in ReplaceOpWithConstantTensor()
314 for (auto &it : llvm::enumerate(op.getNonControlOperands())) { in ReplaceOperationWithBroadcastTo()
523 ValueRange operands = op.getNonControlOperands(); in IsFoldableUncached()
711 for (Value operand : TFOp(op).getNonControlOperands()) { in matchAndRewrite()
945 for (const auto &it : llvm::enumerate(TFOp(op).getNonControlOperands())) { in matchAndRewrite()
1318 for (Value operand : TFOp(op).getNonControlOperands()) { in matchAndRewrite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | graphdef_export.cc | 311 TFOp(return_op).getNonControlOperands())) { in ExportFunction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/ |
| H A D | pass.cc | 373 TFOp(body.getTerminator()).getNonControlOperands().getTypes()); in runOnOperation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/region_to_functional/ |
| H A D | impl.cc | 674 TFOp(ret_op).getNonControlOperands().getTypes(), in Outline()
|