- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 32 for lookupSymbol (0.22 sec)
-
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
print_buffer(*subgraph, i++, input, nullptr); std::cout << '\'' << subgraph_name << "' outputs:\n"; mlir::Operation *terminator = nullptr; if (subgraph->name()) { if (auto fn = module.lookupSymbol<FuncOp>(subgraph->name()->str())) terminator = fn.back().getTerminator(); } i = 0; for (auto output : *subgraph->outputs()) { print_buffer(*subgraph, i, output, [&](int i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
// It's here so xla::Parameters that are created form block.getArguments will // have the proper shapes. TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); mlir::func::FuncOp main = module_op.lookupSymbol<mlir::func::FuncOp>("main"); mlir::Block& block = main.getRegion().front(); xla::XlaBuilder builder("main"); // Create xla_params. std::vector<xla::XlaOp> xla_params;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} auto symbol_name = mlir::cast<FlatSymbolRefAttr>(named_attr.getValue()).getValue(); auto module = op->getParentOfType<ModuleOp>(); mlir::Operation *symbol_op = module.lookupSymbol(symbol_name); if (!symbol_op) { return op->emitError() << "'tf_saved_model.bound_input' attribute must " "reference a valid symbol, got invalid symbol '"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
)mlir"; TEST_F(LiftAsFunctionCallTest, LiftedFunctionSucceeds) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleLifted); ASSERT_TRUE(module_op); auto composite_dot_general_fn = module_op->lookupSymbol<func::FuncOp>("composite_dot_general_fn_1"); ASSERT_THAT(composite_dot_general_fn, NotNull()); auto dot_general_op = FindOperationOfType<mlir::stablehlo::DotGeneralOp>( composite_dot_general_fn);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
&decomposed_partitioned_call_callees); } void StackOpsDecompositionPass::runOnOperation() { auto module = getOperation(); auto main = module.lookupSymbol<func::FuncOp>("main"); if (!main) return; if (failed(DecomposeStackOps(&main.front(), module))) { signalPassFailure(); } } } // namespace namespace TF {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
compile_op_operands.emplace_back(shape_op.getResult()); } FlatSymbolRefAttr func_attr = cluster_func.getFuncAttr(); func::FuncOp func = cluster_func->getParentOfType<ModuleOp>().lookupSymbol<func::FuncOp>( func_attr.getValue()); std::string txt_module; if (failed(EncapsulateFuncAndSerialize( module_name.empty() ? "unknown_graph" : module_name.str(), func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
if (!op->hasAttr(kShapeInferenceGraph)) { return false; } ModuleOp module = op->getParentOfType<ModuleOp>(); func::FuncOp func = module.lookupSymbol<func::FuncOp>( op.getShapeInferenceGraphAttr().getRootReference()); if (func == nullptr) return false; std::vector<_XlaRecvAtHostOp> xla_recv_at_host_ops;
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/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
Builder* builder, mlir::tf_device::ClusterFuncOp cluster_func) { // Look up function definition from module. mlir::func::FuncOp func = cluster_func->getParentOfType<ModuleOp>() .lookupSymbol<mlir::func::FuncOp>(cluster_func.getFunc()); bool use_spmd = false; if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(kUseSpmdAttr)) use_spmd = use_spmd_attr.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
&decomposed_partitioned_call_callees); } void TensorListOpsDecompositionPass::runOnOperation() { auto module = getOperation(); auto main = module.lookupSymbol<func::FuncOp>("main"); if (!main) return; if (failed(DecomposeTensorListOps(&main.front(), module))) { signalPassFailure(); } } } // namespace namespace TF {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
static bool IsValidTFLiteMlirModule(ModuleOp module) { MLIRContext* context = module.getContext(); // Verify that module has a function named main. FuncOp main_fn = module.lookupSymbol<FuncOp>("main"); if (!main_fn) { int entry_func_count = 0; for (auto fn : module.getOps<FuncOp>()) { auto attrs = fn->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)