| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Value.h | 23 class BlockArgument; variable
61 BlockArgument = 3 enumerator
261 friend BlockArgument; variable
266 class BlockArgument : public Value {
271 return value.getKind() == Kind::BlockArgument; in classof()
288 static BlockArgument create(Type type, Block *owner) { in create()
318 return value.getKind() != Kind::BlockArgument; in classof()
384 template <> struct DenseMapInfo<mlir::BlockArgument> {
385 static mlir::BlockArgument getEmptyKey() {
387 return mlir::BlockArgument(
[all …]
|
| D | Block.h | 73 using BlockArgListType = MutableArrayRef<BlockArgument>;
90 BlockArgument addArgument(Type type);
95 BlockArgument insertArgument(args_iterator it, Type type);
101 BlockArgument insertArgument(unsigned index, Type type);
114 BlockArgument getArgument(unsigned i) { return arguments[i]; } in getArgument()
327 std::vector<BlockArgument> arguments;
|
| D | Region.h | 73 using BlockArgListType = MutableArrayRef<BlockArgument>;
90 BlockArgument addArgument(Type type) { return front().addArgument(type); } in addArgument()
95 BlockArgument insertArgument(args_iterator it, Type type) { in insertArgument()
103 BlockArgument insertArgument(unsigned index, Type type) { in insertArgument()
111 BlockArgument getArgument(unsigned i) { return getArguments()[i]; } in getArgument()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Value.cpp | 20 : ownerAndKind(impl, Kind::BlockArgument) {} in Value()
35 if (BlockArgument arg = dyn_cast<BlockArgument>()) in getType()
48 if (BlockArgument arg = dyn_cast<BlockArgument>()) in setType()
83 Operation *parentOp = cast<BlockArgument>().getOwner()->getParentOp(); in getLoc()
91 return cast<BlockArgument>().getOwner()->getParent(); in getParentRegion()
98 return cast<BlockArgument>().getOwner(); in getParentBlock()
107 if (BlockArgument arg = dyn_cast<BlockArgument>()) in getUseList()
|
| D | Block.cpp | 20 unsigned BlockArgument::getArgNumber() const { in getArgNumber()
33 for (BlockArgument arg : arguments) in ~Block()
152 BlockArgument Block::addArgument(Type type) { in addArgument()
153 BlockArgument arg = BlockArgument::create(type, this); in addArgument()
167 BlockArgument Block::insertArgument(unsigned index, Type type) { in insertArgument()
168 auto arg = BlockArgument::create(type, this); in insertArgument()
198 BlockArgument Block::insertArgument(args_iterator it, Type type) { in insertArgument()
205 auto arg = BlockArgument::create(type, this); in insertArgument()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 48 mlir::BlockArgument arg0 = in MatchReductionComputation()
49 reduction_op->getOperand(0).dyn_cast<mlir::BlockArgument>(); in MatchReductionComputation()
50 mlir::BlockArgument arg1 = in MatchReductionComputation()
51 reduction_op->getOperand(1).dyn_cast<mlir::BlockArgument>(); in MatchReductionComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 123 BlockArgument resource_arg, in ValidateResourceArgument()
202 auto add_resource_argument = [&](BlockArgument arg, in PromoteResourcesToArguments()
216 for (BlockArgument& func_arg : func_args) { in PromoteResourcesToArguments()
232 for (BlockArgument& var_handle_arg : var_handle_args) { in PromoteResourcesToArguments()
247 if (auto func_arg = read_op.resource().dyn_cast<BlockArgument>()) { in PromoteResourcesToArguments()
260 if (auto func_arg = write_op.resource().dyn_cast<BlockArgument>()) { in PromoteResourcesToArguments()
|
| D | region_control_flow_to_functional.cc | 367 if (!first.isa<BlockArgument>() || !second.isa<BlockArgument>()) in ConvertWhileOp()
369 BlockArgument first_block_arg = first.cast<BlockArgument>(); in ConvertWhileOp()
370 BlockArgument second_block_arg = second.cast<BlockArgument>(); in ConvertWhileOp()
|
| D | tensor_device_copy_conversion.cc | 47 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction()
|
| D | tpu_dynamic_layout_pass.cc | 106 if (val.isa<BlockArgument>()) return true; in IsSupportedInputOp()
119 if (auto arg = alias.dyn_cast<BlockArgument>()) { in IsSupportedInputOp()
185 mlir::BlockArgument replicate_arg, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
245 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in HandleCompileAndExecutes()
|
| D | replicate_invariant_op_hoisting.cc | 81 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant()
103 if (auto block_arg = read_var_op.resource().dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant()
|
| D | resource_op_lifting_cleanup.cc | 167 for (BlockArgument arg : func.getArguments()) { in EliminateUnusedResultsForIfCase()
204 BlockArgument cond_arg = cond.getArgument(result_idx); in EliminateUnusedResultsForWhile()
205 BlockArgument body_arg = cond.getArgument(result_idx); in EliminateUnusedResultsForWhile()
261 auto block_arg = ret->getOperand(result_idx).dyn_cast<BlockArgument>(); in ForwardCommonArgToOutput()
383 .dyn_cast<BlockArgument>(); in CanonicalizeWhileRegion()
|
| D | tpu_space_to_depth_pass.cc | 135 auto block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast()
150 block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast()
453 BlockArgument block_arg, in HandleHostReplicatedInputs()
487 if (auto block_arg = input.value().dyn_cast<BlockArgument>()) { in HandleCluster()
531 if (auto block_arg = arg.dyn_cast<mlir::BlockArgument>()) { in GetBlockArgNum()
595 if (auto block_arg = conv2d_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution()
604 if (auto block_arg = pad_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution()
|
| D | cross_host_transfer.cc | 96 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction()
|
| D | stack_ops_decomposition.cc | 103 auto maybe_arg = return_val.dyn_cast<BlockArgument>(); in FindAliasedInput()
121 llvm::function_ref<void(ArrayRef<BlockArgument>)> handle_new_size_vars = in ModifyFunctionSignature()
171 auto add_size_vars_to_return = [&](ArrayRef<BlockArgument> new_args) { in HandleWhileOp()
357 [entry.getFirst().cast<BlockArgument>().getArgNumber()] = in HandlePartitionedCallOp()
358 entry.getSecond().cast<BlockArgument>().getArgNumber(); in HandlePartitionedCallOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 388 interleaveComma(block.getArguments(), *p, [&](BlockArgument arg) { in Print()
470 auto verify_operand_types = [&](BlockArgument block_arg, in Verify()
601 llvm::ArrayRef<BlockArgument> ReplicateOp::GetReplicatedBlockArguments() { in GetReplicatedBlockArguments()
607 llvm::ArrayRef<BlockArgument> ReplicateOp::GetPackedBlockArguments() { in GetPackedBlockArguments()
612 bool ReplicateOp::IsReplicatedBlockArgument(BlockArgument block_arg) { in IsReplicatedBlockArgument()
618 bool ReplicateOp::IsPackedBlockArgument(BlockArgument block_arg) { in IsPackedBlockArgument()
626 BlockArgument block_arg, unsigned replica) { in GetReplicaOperandIndexForBlockArgument()
636 Value ReplicateOp::GetReplicaOperandForBlockArgument(BlockArgument block_arg, in GetReplicaOperandForBlockArgument()
650 BlockArgument block_arg) { in GetOperandsForBlockArgument()
|
| D | tf_device_ops.td | 286 llvm::ArrayRef<BlockArgument> GetPackedBlockArguments();
287 llvm::ArrayRef<BlockArgument> GetReplicatedBlockArguments();
288 bool IsReplicatedBlockArgument(BlockArgument block_arg);
289 bool IsPackedBlockArgument(BlockArgument block_arg);
290 unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica);
291 Value GetReplicaOperandForBlockArgument(BlockArgument block_arg, unsigned replica);
292 MutableArrayRef<OpOperand> GetOperandsForBlockArgument(BlockArgument block_arg);
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | BufferResultsToOutParams.cpp | 22 SmallVectorImpl<BlockArgument> &appendedEntryArgs) { in updateFuncOp()
61 ArrayRef<BlockArgument> appendedEntryArgs) { in updateReturnOps()
129 SmallVector<BlockArgument, 6> appendedEntryArgs; in runOnOperation()
|
| D | BufferDeallocation.cpp | 206 (definingBlock == parentBlock && value.isa<BlockArgument>())) { in introduceCopies()
233 if (auto blockArg = value.dyn_cast<BlockArgument>()) in introduceCopies()
247 void introduceBlockArgCopy(BlockArgument blockArg) { in introduceBlockArgCopy()
|
| D | NormalizeMemRefs.cpp | 179 BlockArgument oldMemRef = funcOp.getArgument(argIndex); in areMemRefsNormalizable()
363 BlockArgument newMemRef = in normalizeFuncOpMemRefs()
365 BlockArgument oldMemRef = funcOp.getArgument(argIndex + 1); in normalizeFuncOpMemRefs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_driver.cc | 191 void QuantizeArg(BlockArgument arg, QuantParams params);
200 void RequantizeArg(BlockArgument arg, RequantizeState *state);
221 QuantState &GetArgQuantState(BlockArgument arg) { in GetArgQuantState()
235 RequantizeState &GetArgRequantizeState(BlockArgument arg) { in GetArgRequantizeState()
248 void InitializeArgState(BlockArgument arg, Value in, in InitializeArgState()
380 llvm::DenseMap<BlockArgument, int> arg_states_;
384 llvm::SmallVector<BlockArgument, 4> args_;
520 void QuantizationDriver::QuantizeArg(BlockArgument arg, QuantParams params) { in QuantizeArg()
564 void QuantizationDriver::RequantizeArg(BlockArgument arg, in RequantizeArg()
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | Liveness.cpp | 53 for (BlockArgument argument : block->getArguments()) { in BlockInfoBuilder()
192 currentBlock = value.cast<BlockArgument>().getOwner(); in resolveLiveness()
275 for (BlockArgument argument : block->getArguments()) in print()
289 auto blockArg = value.cast<BlockArgument>(); in print()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 69 using mlir::BlockArgument;
179 Status AddArgumentNode(BlockArgument arg, unsigned index,
186 StatusOr<std::unique_ptr<NodeDef>> GetArgumentNode(BlockArgument arg,
202 llvm::DenseMap<BlockArgument, Node*> args_;
211 BlockArgument arg, unsigned index, llvm::StringRef name) { in GetArgumentNode()
310 auto input_arg = src.cast<BlockArgument>(); in AddEdgeBetweenNodes()
394 bool IsEntryFunctionArg(BlockArgument arg) { in IsEntryFunctionArg()
401 Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index, in AddArgumentNode()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | InliningUtils.cpp | 155 [&](BlockArgument arg) { return !mapper.contains(arg); })) in inlineRegion()
257 BlockArgument regionArg = entryBlock->getArgument(i); in inlineRegion()
337 BlockArgument regionArg = entryBlock->getArgument(i); in inlineCall()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | buffer_reuse_pass.cc | 202 ArrayRef<BlockArgument> arguments) { in find_reuse_candiates()
215 for (BlockArgument old_buffer : arguments) { in find_reuse_candiates()
281 for (BlockArgument arg : f.getArguments()) { in get_buffer_arguments()
|