- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 211 for getIpr (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/guarantee_all_funcs_one_use.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
auto func_ops = module_op.getOps<func::FuncOp>(); auto main_func_itr = absl::c_find_if(func_ops, [&main_func_id](auto func_op) { return func_op.getName() == main_func_id; }); if (main_func_itr == func_ops.end()) return {}; return *main_func_itr; } func::FuncOp GetSaveFuncOp(ModuleOp module_op) { for (auto func_op : module_op.getOps<func::FuncOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
// functions and avoid converting ConstOps inside initializer functions. for (auto func_op : module_op.getOps<func::FuncOp>()) { // Do not unfreeze constants under these functions. if (func_op.getSymName().contains("while_body")) continue; if (func_op.getSymName().contains("while_cond")) continue; absl::c_copy_if(func_op.getOps<TF::ConstOp>(), std::back_inserter(target_const_ops),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
// input token in the parent op or a new token for the entry computation. auto token = StringAttr::get(ctx, tensorflow::kXlaTokenArgNodeName); for (Operation& op : region->getOps()) { // Only communication related ops that needs to have token should have the // extra attribute. if (!ops_with_tokens.contains(&op)) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf_while.cc
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeWhilePass) void RunOnFunction(func::FuncOp func); void runOnOperation() override { for (auto op : getOperation().getOps<func::FuncOp>()) RunOnFunction(op); } }; } // namespace // Inserts call to the given function into the 'region'. void CreateRegionWithCall(func::FuncOp func, Region& region, Location loc) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
device_mgr_.get(), Env::Default(), /*config=*/nullptr, TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr); flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); } FunctionLibraryRuntime* flr_; std::unique_ptr<DeviceMgr> device_mgr_; std::unique_ptr<FunctionLibraryDefinition> lib_def_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
.append("duplicate exported name '", exported_name, "'") .attachNote(p.first->getSecond()->getLoc()) .append("previously seen here"); } } } for (auto func : module.getOps<func::FuncOp>()) { const bool is_exported = IsExported(func); if (is_exported && !func.isPublic()) { return func.emitError() << "exported function @" << func.getName() << " should be public";
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/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
// failure. template <typename OpT> LogicalResult GetOp(Region* region, OpT* result) { *result = {}; for (auto op : region->getOps<OpT>()) { if (*result) return op.emitError("should be unique within a function"); *result = op; } return success(); } LogicalResult RunOnRegion(Region* region) { RecvTPUEmbeddingActivationsOp recv_op; if (failed(GetOp(region, &recv_op))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
})) return replicated_input.emitOpError() << "expects all inputs from 'tf.TPUPartitionedInputV2' ops"; const auto metadata_iter = replicated_input->getBlock()->getOps<TF::TPUReplicateMetadataOp>(); TF::TPUReplicateMetadataOp metadata; if (!metadata_iter.empty()) metadata = *(metadata_iter.begin()); auto first_partitioned_input = llvm::cast<TF::TPUPartitionedInputV2Op>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.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)