Home
last modified time | relevance | path

Searched refs:func_body (Results 1 – 10 of 10) sorted by relevance

/external/llvm-project/mlir/test/Bindings/Python/dialects/
Dstd.py25 func_body = module.body.operations[0].regions[0].blocks[0]
26 subview = func_body.operations[6]
/external/llvm/utils/
Dupdate_test_checks.py208 func_body = func_dict[checkprefix][func_name].splitlines()
213 func_body = genericize_check_lines(func_body)
233 for func_line in func_body:
Dupdate_llc_test_checks.py113 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/llvm-project/llvm/utils/
Dupdate_mir_test_checks.py207 func_body): argument
210 func_body.pop(0)
212 if not func_body:
217 first_line = func_body[0]
225 for func_line in func_body:
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc711 void UpdateFunctionType(OpBuilder& builder, FuncOp func, Block& func_body) { in UpdateFunctionType() argument
712 auto new_argument_types = llvm::to_vector<4>(func_body.getArgumentTypes()); in UpdateFunctionType()
714 llvm::to_vector<4>(func_body.getTerminator()->getOperandTypes()); in UpdateFunctionType()
747 Block& func_body = func.front(); in RewriteFunction() local
749 builder.setInsertionPointToStart(&func_body); in RewriteFunction()
754 rewrite_block ? func_body.addArgument(token_type) in RewriteFunction()
761 {/*region_idx=*/llvm::None, init_token, &func_body.front()}}; in RewriteFunction()
817 if (rewrite_block) UpdateFunctionType(builder, func, func_body); in RewriteFunction()
/external/llvm-project/llvm/utils/UpdateTestChecks/
Dcommon.py484 func_body = str(func_dict[checkprefix][func_name]).splitlines()
488 output_lines.append('%s %s: %s' % (comment_marker, checkprefix, func_body[0]))
489 for func_line in func_body[1:]:
498 func_body = generalize_check_lines(func_body, is_analyze, vars_seen, global_vars_seen)
513 for func_line in func_body:
/external/vulkan-validation-layers/scripts/
Ddispatch_table_helper_generator.py200 func_body = ' { ' + return_type + ' };'
201 decl = decl.replace (';', func_body)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_sharding_identification.mlir206 …%1:2 = "tf.StatefulPartitionedCall"(%arg0){f= @func_body, config="", config_proto="", executor_typ…
217 // CHECK-LABEL: func @func_body
218 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_body(%arg0: tensor<*xi32>)-> tensor<*xi32> {
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1689 const auto& func_body = func->node_def(); in ReachableFunctions() local
1690 std::for_each(func_body.begin(), func_body.end(), process_node); in ReachableFunctions()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc418 const FunctionBody* func_body = flr0_->GetFunctionBody(handle); in TEST_F() local
419 for (const Node* node : func_body->graph->nodes()) { in TEST_F()