- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for getCallee (0.16 sec)
-
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
const llvm::StringMap<Attribute>& derived_attrs) const { // Create the new op Location loc = call_op.getLoc(); rewriter.setInsertionPointAfter(call_op); std::string tf_op_name = GetTFOpName(call_op.getCallee()); OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); Operation* new_op = rewriter.create(new_state); if (materialize_derived_attrs_) { for (const auto& attr : derived_attrs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
if (!MatchCallArgs(first_call.value(), second_call.value(), arg_matcher)) return; can_transform = true; callee_names = {first_call.value().getCallee(), second_call.value().getCallee()}; } }; // Transform IfRegionOp to IfOp. LogicalResult RegionControlFlowToFunctional::ConvertIfOp( SymbolTableCollection& symbol_table, IfRegionOp if_region) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
new_result_types.push_back(token.getType()); auto new_call = builder.create<func::CallOp>( call.getLoc(), new_result_types, new_symbol ? *new_symbol : call.getCallee(), new_operands); for (auto results : llvm::zip(call.getResults(), new_call.getResults())) std::get<0>(results).replaceAllUsesWith(std::get<1>(results)); call.erase(); return new_call.getResults().back();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
// The inliner only inlines the TFR call op. bool changed = false; auto walk_result = func.walk([&](CallOp call_op) { auto callee = table.lookup<TFRFuncOp>(call_op.getCallee()); if (!callee || callee.isExternal()) return WalkResult::advance(); // Record the boundary of the inlined operations. The inlined operation will // be inserted between these two operations.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (auto call_site = mlir::dyn_cast<mlir::CallSiteLoc>(unwrapped_inst_loc)) { if (auto name_loc = mlir::dyn_cast<mlir::NameLoc>( GetLocationWithoutOpType(call_site.getCallee()))) { llvm::StringRef original_node_name, original_func_name; std::tie(original_node_name, original_func_name) = name_loc.getName().strref().split('@');
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/pick_subgraphs.cc
const auto& impl_inference_device_type = GetInferenceDeviceTypeForOp(impl); if (*impl_inference_device_type == preferred_inference_device_type) { if (call.getCallee() != impl.getName()) { // We need to rebuild the call op. :( builder->setInsertionPoint(call); auto new_call = builder->create<func::CallOp>(call.getLoc(), impl,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 24 15:10:02 UTC 2022 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
} }; } // namespace // Checks if an op calls a composite function and all the inputs and outputs are // quantized. bool IsQuantizedCompositeFunction(func::CallOp call_op) { if (!call_op.getCallee().starts_with("quantized_")) { return false; } bool has_quantized_types = false; for (Value operand : call_op.getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
internal/grid/grid_test.go
started := time.Now() dl, _ := ctx.Deadline() if testing.Verbose() { fmt.Println(GetCaller(ctx).Name, "Server deadline:", time.Until(dl)) } <-ctx.Done() serverCanceled <- time.Since(started) if testing.Verbose() { fmt.Println(GetCaller(ctx).Name, "Server Context canceled with", ctx.Err(), "after", time.Since(started)) } return nil },
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Fri Jun 07 15:51:52 UTC 2024 - 36.4K bytes - Viewed (0) -
pkg/registry/core/service/portallocator/controller/repair_test.go
"k8s.io/kubernetes/pkg/registry/core/service/portallocator" ) type mockRangeRegistry struct { getCalled bool item *api.RangeAllocation err error updateCalled bool updated *api.RangeAllocation updateErr error } func (r *mockRangeRegistry) Get() (*api.RangeAllocation, error) { r.getCalled = true return r.item, r.err }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat Sep 30 15:46:06 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
EXPECT_EQ(entry_funcs.size(), 2); EXPECT_EQ(entry_funcs[0].getSymName(), "entry_func_1"); EXPECT_EQ(entry_funcs[1].getSymName(), "entry_func_2"); } TEST(CallGraphUtilTest, GetCallees) { const char *const code = R"mlir( func.func @entry_func(%arg0: tensor<i32>) -> tensor<i32> attributes {tf_saved_model.initializer_type = ""} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0)