| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 475 int32_t op_operand_idx) -> LogicalResult { in Verify()
485 for (auto block_arg : block.getArguments()) { in Verify() local
616 bool ReplicateOp::IsReplicatedBlockArgument(BlockArgument block_arg) { in IsReplicatedBlockArgument()
622 bool ReplicateOp::IsPackedBlockArgument(BlockArgument block_arg) { in IsPackedBlockArgument()
630 BlockArgument block_arg, unsigned replica) { in GetReplicaOperandIndexForBlockArgument()
640 Value ReplicateOp::GetReplicaOperandForBlockArgument(BlockArgument block_arg, in GetReplicaOperandForBlockArgument()
654 BlockArgument block_arg) { in GetOperandsForBlockArgument()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_invariant_op_hoisting.cc | 51 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant() local
73 if (auto block_arg = read_var_op.resource().dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant() local
|
| D | tpu_space_to_depth_pass.cc | 89 auto block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast() local
407 BlockArgument block_arg, in HandleHostReplicatedInputs()
441 if (auto block_arg = input.value().dyn_cast<BlockArgument>()) { in HandleCluster() local
485 if (auto block_arg = arg.dyn_cast<mlir::BlockArgument>()) { in GetBlockArgNum() local
549 if (auto block_arg = conv2d_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
558 if (auto block_arg = pad_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
|
| D | annotate_parameter_replication.cc | 76 auto block_arg = operand.dyn_cast<BlockArgument>(); in runOnOperation() local
|
| D | tensor_device_copy_conversion.cc | 68 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction() local
|
| D | tpu_host_computation_expansion.cc | 45 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
|
| D | tpu_variable_runtime_reformatting.cc | 174 auto block_arg = arg.dyn_cast<BlockArgument>(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
207 const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
|
| D | replicate_to_island.cc | 172 for (auto& block_arg : replicate_op.GetBody().getArguments()) in ExpandReplicateIntoReplicas() local
|
| D | tpu_dynamic_layout_pass.cc | 252 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in HandleCompileAndExecutes() local
|
| D | cluster_tf_ops_pass.cc | 125 if (BlockArgument block_arg = value.dyn_cast<BlockArgument>()) { in GetFunctionMetadatas() local
|
| D | resource_op_lifting_cleanup.cc | 261 auto block_arg = ret->getOperand(result_idx).dyn_cast<BlockArgument>(); in ForwardCommonArgToOutput() local
|
| D | tpu_extract_head_tail_outside_compilation.cc | 63 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
|
| D | tpu_cluster_formation.cc | 488 Value block_arg = std::get<1>(input_and_block_arg); in ReplicateCluster() local
|
| D | tpu_extract_outside_compilation.cc | 327 auto block_arg = v.cast<BlockArgument>(); in GetStaticExternalOperands() local
|
| /external/clang/test/SemaCXX/ |
| D | instantiate-blocks.cpp | 6 T block_arg; in foo() local
|
| /external/clang/test/CodeGenCXX/ |
| D | instantiate-blocks.cpp | 20 T block_arg; in foo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | cross_device_transfer.cc | 96 if (BlockArgument block_arg = value.dyn_cast<BlockArgument>()) { in GetDevice() local
|
| /external/tensorflow/tensorflow/core/ir/ |
| D | ops.cc | 421 Type block_arg = std::get<1>(arg_types.value()); in verifyBody() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | rank_specialization.cc | 233 Value block_arg, operand; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 1197 if (auto block_arg = parent.dyn_cast<mlir::BlockArgument>()) { in MapTensorListResultToArgument() local
|