| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 471 int32_t op_operand_idx) -> LogicalResult { in Verify()
481 for (auto block_arg : block.getArguments()) { in Verify() local
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()
|
| /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
103 if (auto block_arg = read_var_op.resource().dyn_cast<BlockArgument>()) { in MakeShapeOpInvariant() local
|
| D | tpu_space_to_depth_pass.cc | 135 auto block_arg = cast_input.dyn_cast<mlir::BlockArgument>(); in HandleCast() local
453 BlockArgument block_arg, in HandleHostReplicatedInputs()
487 if (auto block_arg = input.value().dyn_cast<BlockArgument>()) { in HandleCluster() local
531 if (auto block_arg = arg.dyn_cast<mlir::BlockArgument>()) { in GetBlockArgNum() local
595 if (auto block_arg = conv2d_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
604 if (auto block_arg = pad_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution() local
|
| D | tensor_device_copy_conversion.cc | 47 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction() local
|
| D | annotate_parameter_replication.cc | 75 auto block_arg = operand.dyn_cast<BlockArgument>(); in runOnOperation() local
|
| D | tpu_host_computation_expansion.cc | 47 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
|
| D | cross_host_transfer.cc | 96 if (BlockArgument block_arg = arg.dyn_cast<BlockArgument>()) { in runOnFunction() local
|
| D | tpu_variable_runtime_reformatting.cc | 165 auto block_arg = arg.dyn_cast<BlockArgument>(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
198 const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
|
| D | tpu_dynamic_padding_mapper.cc | 65 if (auto block_arg = operand_and_idx.value().dyn_cast<BlockArgument>()) { in GetRemappedReplicatedInputIndices() local
|
| D | replicate_to_island.cc | 166 for (auto& block_arg : replicate_op.GetBody().getArguments()) in ExpandReplicateIntoReplicas() local
|
| D | tpu_dynamic_layout_pass.cc | 245 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 | 62 if (auto block_arg = value.dyn_cast<BlockArgument>()) in GetOpOfValue() local
|
| D | tpu_extract_outside_compilation.cc | 214 auto block_arg = v.cast<BlockArgument>(); in GetExternalOperands() local
|
| D | tpu_cluster_formation.cc | 469 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/llvm-project/clang/test/SemaCXX/ |
| D | instantiate-blocks.cpp | 6 T block_arg; in foo() local
|
| /external/llvm-project/clang/test/CodeGenCXX/ |
| D | instantiate-blocks.cpp | 20 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/lite/transforms/ |
| D | legalize_variables.cc | 59 auto block_arg = op.resource().template dyn_cast<BlockArgument>(); in GetGlobalTensor() local
|
| D | initialize_variables.cc | 53 auto block_arg = op.resource().template dyn_cast<BlockArgument>(); in GetGlobalTensor() local
|