| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_sparsetensors.cc | 139 for (const auto block_arg : main_func.getArguments()) { in CreateComponentTensorsFromSparseTensors() local
185 for (auto block_arg : main_func.getArguments()) { in runOnOperation() local
236 for (auto block_arg : main_func.getArguments()) { in runOnOperation() local
|
| D | device_utils.cc | 51 if (auto block_arg = val.dyn_cast<mlir::BlockArgument>()) { in DeviceId() local
|
| D | cluster_function_conversion.cc | 78 if (auto block_arg = operand.dyn_cast<mlir::BlockArgument>()) { in AttachRetvalLayouts() local
|
| D | mesh_propagation.cc | 55 mlir::LogicalResult ExtractMeshFromBlockArgument(mlir::BlockArgument block_arg, in ExtractMeshFromBlockArgument()
137 if (auto block_arg = operand_value.dyn_cast<mlir::BlockArgument>()) { in ExtractMeshFromOperand() local
|
| D | layout_parsing.cc | 187 auto block_arg = operand.dyn_cast<mlir::BlockArgument>(); in ExtractLayoutFromOperand() local
|
| D | shape_utils.cc | 62 auto block_arg = input_value.get().dyn_cast<mlir::BlockArgument>(); in ExtractGlobalInputShape() local
|
| D | dtensor_allreduce_sum_optimization.cc | 289 mlir::BlockArgument block_arg; in CheckWhileLoopOptimizationCriteria() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 525 int32_t op_operand_idx) -> LogicalResult { in verify()
535 for (auto block_arg : block.getArguments()) { in verify() local
624 bool ReplicateOp::IsReplicatedBlockArgument(BlockArgument block_arg) { in IsReplicatedBlockArgument()
630 bool ReplicateOp::IsPackedBlockArgument(BlockArgument block_arg) { in IsPackedBlockArgument()
638 BlockArgument block_arg, unsigned replica) { in GetReplicaOperandIndexForBlockArgument()
648 Value ReplicateOp::GetReplicaOperandForBlockArgument(BlockArgument block_arg, in GetReplicaOperandForBlockArgument()
662 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 | 90 auto block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast() local
408 BlockArgument block_arg, in HandleHostReplicatedInputs()
442 if (auto block_arg = input.value().dyn_cast<BlockArgument>()) { in HandleCluster() local
486 if (auto block_arg = arg.dyn_cast<mlir::BlockArgument>()) { in GetBlockArgNum() local
550 if (auto block_arg = conv2d_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
559 if (auto block_arg = pad_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
|
| D | annotate_parameter_replication.cc | 77 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 runOnOperation() 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 | 118 auto block_arg = arg.dyn_cast<BlockArgument>(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
151 const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
|
| D | replicate_to_island.cc | 168 for (auto& block_arg : replicate_op.GetBody().getArguments()) in ExpandReplicateIntoReplicas() local
|
| D | tpu_dynamic_layout_pass.cc | 228 if (auto block_arg = input.dyn_cast<BlockArgument>()) { in HandleCompileAndExecutes() local
|
| D | cluster_tf_ops_pass.cc | 126 if (BlockArgument block_arg = value.dyn_cast<BlockArgument>()) { in GetFunctionMetadatas() local
|
| D | resource_op_lifting_cleanup.cc | 265 auto block_arg = ret->getOperand(result_idx).dyn_cast<BlockArgument>(); in ForwardCommonArgToOutput() local
|
| D | tpu_extract_head_tail_outside_compilation.cc | 64 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
|
| D | tpu_cluster_formation.cc | 504 Value block_arg = std::get<1>(input_and_block_arg); in ReplicateCluster() 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/dtensor/mlir/expansions/ |
| D | resource_spmd_expander.cc | 192 auto block_arg = input_resource_value.dyn_cast<mlir::BlockArgument>(); in ExpandOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | cross_device_transfer.cc | 97 if (BlockArgument block_arg = value.dyn_cast<BlockArgument>()) { in GetDevice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | copy_cleanup_pass.cc | 137 auto block_arg = copy.getTarget().dyn_cast<BlockArgument>(); in RemoveCopyIfTargetIsFunctionArg() local
|