Searched refs:parent_op (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_functional_ops.cc | 82 FuncOp parent_op = op->getParentOfType<FuncOp>(); in matchAndRewrite() local
83 if (!llvm::hasSingleElement(parent_op)) return failure(); in matchAndRewrite()
141 UpdateFuncType(parent_op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/c/experimental/op_handler/ |
| D | internal.cc | 56 OpHandlerOperation::OpHandlerOperation(AbstractOperation* parent_op) in OpHandlerOperation() argument
57 : WrapperOperation(parent_op, kOpHandler) {} in OpHandlerOperation()
|
| D | wrapper_operation.cc | 18 WrapperOperation::WrapperOperation(AbstractOperation* parent_op, in WrapperOperation() argument
20 : AbstractOperation(kind), parent_op_(parent_op) { in WrapperOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | mark_ops_for_outside_compilation.cc | 310 while (auto* parent_op = iter_op->getParentOp()) { in UnmarkChildren() local
311 if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in UnmarkChildren()
315 iter_op = parent_op; in UnmarkChildren()
|
| D | resource_op_lifting_cleanup.cc | 72 LogicalResult CleanupAndCanonicalize(Operation *parent_op);
402 LogicalResult CleanupAndCanonicalize(Operation *parent_op) { in CleanupAndCanonicalize() argument
403 auto walk_result = parent_op->walk([](Operation *op) { in CleanupAndCanonicalize()
|
| D | tpu_extract_outside_compilation.cc | 85 Operation* parent_op = op->getParentOp(); in HasOutsideCompilationAncestorExclusive() local
86 if (!parent_op) return false; in HasOutsideCompilationAncestorExclusive()
87 return HasOutsideCompilationAncestor(parent_op); in HasOutsideCompilationAncestorExclusive()
|
| D | tpu_extract_head_tail_outside_compilation.cc | 73 if (auto* parent_op = op_block->getParentOp()) { in OpInBlock() local
74 op_block = parent_op->getBlock(); in OpInBlock()
|
| D | legalize_hlo.cc | 971 LogicalResult NormalizeIndexVector(Operation *parent_op, Value &indices, in NormalizeIndexVector() argument
981 indices = rewriter.create<mhlo::ReshapeOp>(parent_op->getLoc(), in NormalizeIndexVector()
988 parent_op, in NormalizeIndexVector()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_operation.cc | 22 TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, in TapeOperation() argument
25 parent_op_(parent_op), in TapeOperation()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 863 parent_op=op,
2010 def __init__(self, handle, dtype, shape, in_graph_mode, deleter, parent_op, argument
2022 with ops.control_dependencies([parent_op]):
2034 self._parent_op = parent_op
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1253 auto parent_op = op.getOperation()->getParentRegion()->getParentOp(); in matchAndRewrite() local
1255 parent_op)) { in matchAndRewrite()
|