- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 537 for funcMap (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
// Builder builder(func_op.getContext()); func_op->setAttr( kTfInputShapesAttr, ArrayAttr::get(func_op.getContext(), updated_input_shapes_attr)); } } // Validates func ops. Returns `failure` if the function is invalid. LogicalResult ValidateFuncOp(func::FuncOp func_op) { auto input_shapes_attr = func_op->getAttrOfType<ArrayAttr>(kTfInputShapesAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
return failure(result.wasInterrupted()); } void ConvertToLegacyCompileAndReplicateAttributesPass::runOnOperation() { func::FuncOp func_op = getOperation(); if (failed(ConvertToLegacyAttributes(func_op))) return signalPassFailure(); } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertToLegacyCompileAndReplicateAttributesPass() { return std::make_unique<ConvertToLegacyCompileAndReplicateAttributesPass>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/tensor_array_side_effect_analysis.cc
mlir::func::FuncOp func_op) { for (mlir::Operation& op : func_op.front()) { if (!mlir::isMemoryEffectFree(&op) && !IsTensorArrayOp(&op)) return false; } return true; } TensorArraySideEffectAnalysis::TensorArraySideEffectAnalysis( mlir::ModuleOp module) { for (auto func_op : module.getOps<mlir::func::FuncOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
} }; // Returns true iff func_op has either no Region or the body has no Blocks. bool IsFuncOpEmpty(func::FuncOp func_op) { return func_op->getNumRegions() == 0 || func_op.getBody().empty(); } // Gets the GraphOp from the function op. Returns an empty op iff it doesn't // exist. GraphOp GetGraphOpFromFuncOp(func::FuncOp func_op) { if (IsFuncOpEmpty(func_op)) return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/mark_functions_noinline.cc
const StringSet<> noinline_functions = GetNoinlineFunctionsSet(); func::FuncOp func_op = getOperation(); Builder builder(&getContext()); // Adds the `tf._noinline = true` attribute to the function if the name // matches. if (noinline_functions.contains(func_op.getSymName())) { func_op->setAttr(kTfNoinlineAttr, builder.getBoolAttr(true)); LLVM_DEBUG(llvm::dbgs()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 18 02:52:57 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kArgumentWithUnknownDims); ASSERT_TRUE(module_op); auto func_op = module_op->lookupSymbol<func::FuncOp>("unknown_dims_arg"); ASSERT_THAT(func_op, NotNull()); ASSERT_THAT(func_op.getNumArguments(), Eq(1)); EXPECT_TRUE(HasRankOf(func_op.getArgument(0), /*rank=*/2)); } TEST_F(AttrsAndConstraintsTest, HasRankOfReturnsFalseForUnknownRank) {
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/quantization/tensorflow/ops/tf_quantize_op.cc
} Value func_input_arg; // Creates a function. func::FuncOp func_op = PrepareFunctionRegister( rewriter, input_val, result_type, unique_func_name, func_input_arg); // Fills the body. Operation* last_op_in_func = quantization_operations_func(rewriter, func_op.getOperation(), func_input_arg, result_type, quant_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
Block &nested_block = nested_module->getRegion(0).front(); for (func::FuncOp func_op : llvm::make_early_inc_range(nested_block.getOps<func::FuncOp>())) { if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) { nested_block.getOperations().remove(func_op.getOperation()); symbol_table.insert(func_op.getOperation()); } } nested_module->erase(); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
}; // Folds tf.IdentityOp and tf.IdentityNOp if op device and the argument devices // from the defining ops match. void TensorDeviceCopyConversionPass::runOnOperation() { func::FuncOp func_op = getOperation(); auto should_fold_op_func = [&func_op](const Value &arg, const StringAttr &op_device) { // In TFRT TPU, tensor transfer is handled specifically by D2H and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
EXPECT_TRUE(module_op_ref); return module_op_ref; } mlir::MLIRContext ctx_{}; }; func::FuncOp GetMainFuncOp(ModuleOp module_op) { for (auto func_op : module_op.getOps<func::FuncOp>()) { if (func_op.getSymName() == "main") { return func_op; } } return {}; } TEST_F(ConvertAssetArgsTest, ConvertsSingleAssetArg) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0)