Searched refs:func_body (Results 1 – 7 of 7) sorted by relevance
| /external/llvm/utils/ |
| D | update_test_checks.py | 208 func_body = func_dict[checkprefix][func_name].splitlines()
213 func_body = genericize_check_lines(func_body)
233 for func_line in func_body:
|
| D | update_llc_test_checks.py | 113 func_body = func_dict[checkprefix][func_name].splitlines()
114 output_lines.append('; %s: %s' % (checkprefix, func_body[0]))
115 for func_line in func_body[1:]:
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 777 Block& func_body) { in UpdateFunctionType() argument
778 auto new_argument_types = llvm::to_vector(func_body.getArgumentTypes()); in UpdateFunctionType()
780 llvm::to_vector(func_body.getTerminator()->getOperandTypes()); in UpdateFunctionType()
813 Block& func_body = func.front(); in RewriteFunction() local
815 builder.setInsertionPointToStart(&func_body); in RewriteFunction()
820 rewrite_block ? func_body.addArgument(token_type, func.getLoc()) in RewriteFunction()
827 {/*region_idx=*/llvm::None, init_token, &func_body.front()}}; in RewriteFunction()
896 if (rewrite_block) UpdateFunctionType(builder, func, func_body); in RewriteFunction()
|
| /external/vulkan-validation-layers/scripts/ |
| D | dispatch_table_helper_generator.py | 200 func_body = ' { ' + return_type + ' };'
201 decl = decl.replace (';', func_body)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_sharding_identification.mlir | 206 …%1:2 = "tf.StatefulPartitionedCall"(%arg0){f= @func_body, config="", config_proto="", executor_typ…
217 // CHECK-LABEL: func @func_body
218 func.func @func_body(%arg0: tensor<*xi32>)-> (tensor<*xi32>, tensor<*xi1>) {
246 …%0 = "tf.PartitionedCall"(%arg0) {f= @func_body, config="", config_proto="", executor_type=""} : (…
250 func.func @func_body(%arg0: tensor<*xi32>)-> tensor<*xi32> {
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 1751 const auto& func_body = func->node_def(); in ReachableFunctions() local
1752 std::for_each(func_body.begin(), func_body.end(), process_node); in ReachableFunctions()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 401 const FunctionBody* func_body = flr0_->GetFunctionBody(handle); in TEST_F() local
402 for (const Node* node : func_body->graph->nodes()) { in TEST_F()
|