- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for call_op (0.22 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
if (method.ok() && method->has_static_range_ptq()) return true; } TF::PartitionedCallOp call_op = dyn_cast_or_null<TF::PartitionedCallOp>(op); return call_op && call_op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) && call_op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue() == llvm::StringRef(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
for (auto& function_use : function_uses.value()) { auto call_op = function_use.getUser(); auto caller_func = call_op->getParentOfType<func::FuncOp>(); if (!caller_func) return failure(); builder.setInsertionPoint(call_op); for (auto [lifted_op, arg_idx] : lifted_op_and_arg_idx) { auto new_op = builder.clone(*lifted_op, mapping); call_op->insertOperands(arg_idx, new_op->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
return method.has_weight_only_ptq(); } bool IsWeightOnlyQuantizableOp(const Operation& op) { if (auto call_op = dyn_cast<TF::XlaCallModuleOp>(op)) { StringRef entry_function_name = GetEntryFunctionName(call_op); absl::StatusOr<Method> quantization_method = GetQuantizationMethod(call_op); return ContainsConvOrDot(entry_function_name) && quantization_method.ok() && quantization_method->has_weight_only_ptq();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); Operation* call_op = FindOperationOfType<TF::XlaCallModuleOp>(main_fn); EXPECT_FALSE(IsHybridQuantizedOp(call_op)); } constexpr absl::string_view kModuleDotGeneralFullyConnected = R"mlir( module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
max_iterations); } else if (auto call_op = dyn_cast<CallOpInterface>(op)) { if (auto func = dyn_cast<func::FuncOp>(call_op.resolveCallable(&symbol_table_))) { PropagateConstantToCallee(call_op, func, module); FailureOr<bool> failure_or_converged = PropagateShapeToFunctions( module, call_op.getArgOperands().getTypes(), {func}, max_iterations);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
op_builder.setInsertionPointToStart(®ion.front()); auto call_op = op_builder.create<mlir::func::CallOp>( loc, func.getFunctionType().getResults(), func.getSymName(), region.getArguments()); op_builder.create<mlir::TFL::YieldOp>(loc, call_op.getResults()); } void InlineStablehloOpRegion(mlir::Region& region, mlir::func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto get_call_index = [&](mlir::Block& b) -> std::optional<int> { if (b.getOperations().size() != 2) return std::nullopt; if (auto call_op = dyn_cast<mlir::func::CallOp>(b.front())) return subgraph_index_map_.at(call_op.getCallee().str()); return std::nullopt; }; auto body_subgraph_index = get_call_index(op.getBody().front());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto inputs = func_ty.getInputs(); block->addArguments(inputs, SmallVector<Location>(inputs.size(), loc)); mlir::func::CallOp call_op = rewriter.create<mlir::func::CallOp>( loc, func, func_ty.getResults(), block->getArguments()); rewriter.create<mhlo::ReturnOp>(loc, call_op.getResults()); } //===----------------------------------------------------------------------===// // Op converters.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
// only executes ops with no side effects. mutable llvm::SmallDenseMap<Operation*, bool> is_pure_function_; }; bool OpSideEffectCollector::IsCallToPureFunction(Operation* callOp) const { auto call = llvm::dyn_cast<CallOpInterface>(callOp); if (!call) return false; // not a call func::FuncOp func_op = dyn_cast<func::FuncOp>(call.resolveCallable( &symbol_table_collection_)); return IsPureFunction(func_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
src/runtime/testdata/testprogcgo/coro.go
// license that can be found in the LICENSE file. //go:build goexperiment.rangefunc && !windows package main /* #include <stdint.h> // for uintptr_t void go_callback_coro(uintptr_t handle); static void call_go(uintptr_t handle) { go_callback_coro(handle); } */ import "C" import ( "fmt" "iter" "runtime/cgo" ) func init() { register("CoroCgoIterCallback", func() { println("expect: OK")
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri May 17 19:46:10 UTC 2024 - 3.4K bytes - Viewed (0)