- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 129 for created (0.22 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
const llvm::SmallVector<int64_t>& expected_packed_shape, const llvm::SmallVector<int8_t>& expected_packed_values) { MLIRContext context; OwningOpRef<ModuleOp> module(ModuleOp::create(UnknownLoc::get(&context))); OpBuilder builder(&module->getBodyRegion()); context.loadDialect<TF::TensorFlowDialect>(); Value value = CreateConstValue<int8_t>(builder, module->getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
mlir::MLIRContext context(registry); auto module = mlir::parseSourceFile<mlir::ModuleOp>(saved_model_mlir_path, &context); ASSERT_TRUE(module); // Create a cost recorder with fake cost records. auto expected_op_cost_map = GetOpCostMap(module.get()); EXPECT_EQ(expected_op_cost_map.size(), 1); unsigned int seed = 23579; for (auto& [op_key, cost] : expected_op_cost_map) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/common/controller.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 09:51:28 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr(); MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym)); } builder.setInsertionPoint(cluster_func_op); auto xla_launch_op = builder.create<TF::XlaLaunchOp>( cluster_func_op.getLoc(), cluster_func_op.getResultTypes(), /*constants=*/ValueRange({}), ValueRange(non_resource_args), ValueRange(resource_args), cluster_func_op.getFuncAttr());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc
// original const and the qbarrier that led to the quantization. auto fusedLoc = rewriter.getFusedLoc( {qbarrier.getArg().getDefiningOp()->getLoc(), qbarrier.getLoc()}); auto newConstOp = rewriter.create<arith::ConstantOp>( fusedLoc, newConstValueType, cast<TypedAttr>(newConstValue)); rewriter.replaceOpWithNewOp<StorageCastOp>(qbarrier, qbarrier.getType(), newConstOp);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/trim_functions_tf.cc
return WalkResult::advance(); }); if (walk_result.wasInterrupted()) return signalPassFailure(); } } } // namespace // Creates an instance of the TensorFlow Lite dialect TrimFunctions /// pass. std::unique_ptr<OperationPass<ModuleOp>> CreateTrimFunctionsPass() { return std::make_unique<TrimFunctionsPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 07 21:08:41 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/post_quantize.cc
populateWithGenerated(patterns); if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) { signalPassFailure(); } } } // namespace // Creates an instance of the TensorFlow dialect PostQuantize pass. std::unique_ptr<OperationPass<func::FuncOp>> CreatePostQuantizePass() { return std::make_unique<PostQuantizePass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
// TODO: b/310291615 - find a better way for multi-platform support. if (call_op_has_platform_index_arg) { arg_mapper.map(func_op.getArgument(0), builder.create<mhlo::ConstantOp>( func_op.getLoc(), builder.getI16IntegerAttr(0))); } for (auto [func_arg, operand] : llvm::zip_equal( func_op.getArguments().take_back(call_op.getNumOperands()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/jit/cluster_scoping_pass.cc
node->AddAttr(kXlaInternalScopeAttr, scope); } // NB! We append a new scope as suffix to the _XlaInternalScope attribute // instead of overriding the old value. In other words, appending scope B to // scope A creates the conjunction of the scopes A and B (i.e, A & B) and, // in effect, the node gets both the old and new scopes. As a unique scope // disallows a node being merged with nodes in other scopes, the scope
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/eval_util.cc
// attribute of the MLIR op. The assigned device might be remote, not // available during compilation or compilation only device for on demand // execution which may create a recursion if used for constant folding. constexpr char kHostCpu[] = "/job:localhost/replica:0/task:0/CPU:0"; TFE_OpSetDevice(op, kHostCpu, status); RETURN_FAILURE_IF_ERROR(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 13 06:02:14 UTC 2023 - 5.7K bytes - Viewed (0)