| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 470 auto verify_operand_types = [&](BlockArgument block_arg, in Verify()
473 if (failed(VerifyCompatibleTypes(block_arg.getType(), op_operand_type))) in Verify()
476 << ") and block argument " << block_arg.getArgNumber() << " (" in Verify()
477 << block_arg.getType() << ") to have compatible types"; in Verify()
481 for (auto block_arg : block.getArguments()) { in Verify() local
482 if (block_arg.getArgNumber() < num_replicated_block_args) { in Verify()
483 for (int32_t i = n * block_arg.getArgNumber(), e = i + n; i < e; ++i) in Verify()
484 if (failed(verify_operand_types(block_arg, i))) return failure(); in Verify()
486 const int32_t idx = block_arg.getArgNumber() - num_replicated_block_args + in Verify()
488 if (failed(verify_operand_types(block_arg, idx))) return failure(); in Verify()
[all …]
|
| D | tf_device_ops.td | 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/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_invariant_op_hoisting.cc | 81 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant() local
82 if (block_arg.getOwner() != replicate_block) return; in MakeShapeOpInvariant()
85 block_arg, /*replica=*/0)); in MakeShapeOpInvariant()
103 if (auto block_arg = read_var_op.resource().dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant() local
104 if (block_arg.getOwner() != replicate_block) return; in MakeShapeOpInvariant()
109 replicate_op.GetReplicaOperandForBlockArgument(block_arg, in MakeShapeOpInvariant()
|
| D | tpu_space_to_depth_pass.cc | 135 auto block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast() local
137 while (block_arg || cast_op_input) { in HandleCast()
138 if (block_arg) { in HandleCast()
140 HandleFuncOp(block_arg.getOwner()->getParentOp()); in HandleCast()
141 block_arg = nullptr; in HandleCast()
150 block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast()
453 BlockArgument block_arg, in HandleHostReplicatedInputs() argument
460 replicate.GetOperandsForBlockArgument(block_arg); in HandleHostReplicatedInputs()
473 block_arg.setType(space_to_depth.getType()); in HandleHostReplicatedInputs()
487 if (auto block_arg = input.value().dyn_cast<BlockArgument>()) { in HandleCluster() local
[all …]
|
| D | annotate_parameter_replication.cc | 75 auto block_arg = operand.dyn_cast<BlockArgument>(); in runOnOperation() local
76 if (block_arg && block_arg.getOwner() == &replicate.GetBody()) { in runOnOperation()
78 if (mirrored_replicate_args.count(block_arg.getArgNumber()) == 0) { in runOnOperation()
|
| D | tensor_device_copy_conversion.cc | 47 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction() local
50 if (block_arg.getParentRegion() != &func_op.getRegion()) { in runOnFunction()
54 block_arg.getArgNumber(), kTFDeviceAttr)) { in runOnFunction()
|
| D | tpu_variable_runtime_reformatting.cc | 165 auto block_arg = arg.dyn_cast<BlockArgument>(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
166 if (!block_arg || block_arg.getOwner() != &replicate.GetBody()) continue; in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
167 assert(replicate_arg_to_execute_arg.count(block_arg.getArgNumber()) == 0 && in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
169 replicate_arg_to_execute_arg[block_arg.getArgNumber()] = in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
198 const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
201 if (replicate.IsReplicatedBlockArgument(block_arg)) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
215 replicate.GetReplicaOperandForBlockArgument(block_arg, i), in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
|
| D | cross_host_transfer.cc | 96 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction() local
99 if (block_arg.getParentRegion() != &func_op.getRegion()) continue; in runOnFunction()
102 block_arg.getArgNumber(), kArgDeviceAttr)) { in runOnFunction()
|
| D | tpu_host_computation_expansion.cc | 47 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
48 return block_arg.getOwner()->getParentOp(); in GetOpOfValue()
|
| D | tpu_dynamic_padding_mapper.cc | 65 if (auto block_arg = operand_and_idx.value().dyn_cast<BlockArgument>()) { in GetRemappedReplicatedInputIndices() local
66 if (block_arg.getOwner() == replicate_block) { in GetRemappedReplicatedInputIndices()
68 replicated_input_indices_attr[block_arg.getArgNumber()] in GetRemappedReplicatedInputIndices()
|
| D | replicate_to_island.cc | 166 for (auto& block_arg : replicate_op.GetBody().getArguments()) in ExpandReplicateIntoReplicas() local
167 mapping.map(block_arg, in ExpandReplicateIntoReplicas()
168 replicate_op.GetReplicaOperandForBlockArgument(block_arg, i)); in ExpandReplicateIntoReplicas()
|
| D | tpu_dynamic_layout_pass.cc | 245 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in HandleCompileAndExecutes() local
248 if (maybe_replicate != block_arg.getParentRegion()->getParentOp() || in HandleCompileAndExecutes()
250 execute_launch, compile_launch, block_arg, in HandleCompileAndExecutes()
|
| D | resource_op_lifting_cleanup.cc | 261 auto block_arg = ret->getOperand(result_idx).dyn_cast<BlockArgument>(); in ForwardCommonArgToOutput() local
262 if (!block_arg) { in ForwardCommonArgToOutput()
268 common_arg_index = block_arg.getArgNumber(); in ForwardCommonArgToOutput()
269 } else if (common_arg_index.getValue() != block_arg.getArgNumber()) { in ForwardCommonArgToOutput()
|
| D | cluster_tf_ops_pass.cc | 125 if (BlockArgument block_arg = value.dyn_cast<BlockArgument>()) { in GetFunctionMetadatas() local
127 block_arg.getArgNumber(), kTFDeviceAttr)) { in GetFunctionMetadatas()
|
| D | tpu_cluster_formation.cc | 469 Value block_arg = std::get<1>(input_and_block_arg); in ReplicateCluster() local
470 mlir::replaceAllUsesInRegionWith(input->getResult(0), block_arg, in ReplicateCluster()
476 pi.setOperand(use.getOperandNumber(), block_arg); in ReplicateCluster()
|
| D | tpu_extract_head_tail_outside_compilation.cc | 62 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
63 return block_arg.getOwner()->getParentOp(); in GetOpOfValue()
|
| D | tpu_extract_outside_compilation.cc | 214 auto block_arg = v.cast<BlockArgument>(); in GetExternalOperands() local
215 if (block_arg.getParentRegion() == op->getParentRegion()) in GetExternalOperands()
|
| /external/clang/test/SemaCXX/ |
| D | instantiate-blocks.cpp | 6 T block_arg; in foo() local
11 …block_arg = arg; // expected-error {{variable is not assignable (missing __block type specifier)}} in foo()
12 return block_arg+arg; }; in foo()
|
| /external/llvm-project/clang/test/SemaCXX/ |
| D | instantiate-blocks.cpp | 6 T block_arg; in foo() local
11 …block_arg = arg; // expected-error {{variable is not assignable (missing __block type specifier)}} in foo()
12 return block_arg+arg; }; in foo()
|
| /external/llvm-project/clang/test/CodeGenCXX/ |
| D | instantiate-blocks.cpp | 20 T block_arg; in foo() local
25 return byref_block_arg + block_arg + arg; }; in foo()
|
| /external/clang/test/CodeGenCXX/ |
| D | instantiate-blocks.cpp | 20 T block_arg; in foo() local
25 return byref_block_arg + block_arg + arg; }; in foo()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_variables.cc | 59 auto block_arg = op.resource().template dyn_cast<BlockArgument>(); in GetGlobalTensor() local
60 if (!block_arg) return nullptr; in GetGlobalTensor()
61 int index = block_arg.getArgNumber(); in GetGlobalTensor()
|
| D | initialize_variables.cc | 53 auto block_arg = op.resource().template dyn_cast<BlockArgument>(); in GetGlobalTensor() local
54 if (!block_arg) return nullptr; in GetGlobalTensor()
55 int index = block_arg.getArgNumber(); in GetGlobalTensor()
|
| /external/llvm-project/clang/test/SemaOpenCL/ |
| D | invalid-block.cl | 37 …int (^const bl2)(bl_t block_arg) = ^() { // expected-error{{declaring function parameter of type '…
38 …return block_arg(); // expected-error{{implicit declaration of function 'block_arg' is invalid in …
|