Home
last modified time | relevance | path

Searched refs:inner_op (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreplicate_invariant_op_hoisting.cc164 for (Operation& inner_op : in HoistReplicateInvariantOps()
166 if (llvm::isa<tf_device::ReturnOp>(inner_op)) continue; in HoistReplicateInvariantOps()
169 if (UsesVirtualDevice(virtual_device_list, &inner_op)) continue; in HoistReplicateInvariantOps()
171 if (IsOpReplicateInvariant(replicate_region, &inner_op)) in HoistReplicateInvariantOps()
172 inner_op.moveBefore(replicate_op); in HoistReplicateInvariantOps()
Dtpu_cluster_formation.cc143 op.walk([&](Operation* inner_op) { in CollectResourceIdsFromOp() argument
144 for (Value operand : TF::filter_resources(inner_op->getOperands())) { in CollectResourceIdsFromOp()
149 for (Value result : TF::filter_resources(inner_op->getResults())) { in CollectResourceIdsFromOp()
165 auto result = op->walk([&](Operation* inner_op) { in ShouldMoveOpAfterCluster() argument
166 for (Value operand : inner_op->getOperands()) { in ShouldMoveOpAfterCluster()
182 if (is_replicate && inner_op != op) return WalkResult::advance(); in ShouldMoveOpAfterCluster()
185 for (Value operand : TF::filter_resources(inner_op->getOperands())) { in ShouldMoveOpAfterCluster()
274 cluster_op->walk([&](Operation* inner_op) { in CreateClusterOp() argument
275 inner_op->removeAttr(kTPUReplicateAttr); in CreateClusterOp()
276 inner_op->removeAttr(kDeviceAttr); in CreateClusterOp()
Dgraph_pruning.cc129 auto result = island.walk([this](Operation* inner_op) { in ShouldPreserveIsland() argument
130 if (ShouldPreserveOp(inner_op)) return WalkResult::interrupt(); in ShouldPreserveIsland()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Draise_custom_ops.cc61 Operation *inner_op = builder.clone(*op); in runOnFunction() local
62 builder.create<YieldOp>(op->getLoc(), inner_op->getResults()); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc577 Operation& inner_op = island.GetBody().front(); in Convert() local
578 auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef()); in Convert()
588 if (IsLegacyCallInstruction(&inner_op)) { in Convert()
590 inner_op.getAttrOfType<mlir::SymbolRefAttr>("f") in Convert()
594 TF_RETURN_IF_ERROR(exporter.AddInstructionNode(&inner_op)); in Convert()
Dimport_model.cc1563 mlir::Operation* inner_op = &island_op.GetBody().front(); in ConvertFunctionArgAndRets() local
1567 if (inner_op->getNumOperands() != 1) in ConvertFunctionArgAndRets()
1569 inst_to_return.push_back(inner_op->getOperand(0)); in ConvertFunctionArgAndRets()
1789 mlir::Operation* inner_op = island_builder.createOperation(result); in CreateOperation() local
1806 inner_op->setAttr(attr_name, attr_value); in CreateOperation()
1809 if (inner_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>() || in CreateOperation()
1810 inner_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { in CreateOperation()
1818 if (inner_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>()) { in CreateOperation()
1826 if (inner_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { in CreateOperation()
1835 mlir::OperationName name = inner_op->getName(); in CreateOperation()
[all …]
/external/python/cpython3/Lib/test/
Dtest_re.py1246 inner_op = inner_reps + inner_mod
1247 self.checkPatternError(r'x%s%s' % (inner_op, outer_op),
1248 'multiple repeat', 1 + len(inner_op))
1810 for inner_op in ('{0,}', '*', '?'):
1811 r = re.compile("^((x|y)%s)%s" % (inner_op, outer_op))
/external/python/cpython2/Lib/test/
Dtest_re.py1108 for inner_op in ('{0,}', '*', '?'):
1109 r = re.compile("^((x|y)%s)%s" % (inner_op, outer_op))