- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 51 for GetInputs (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
if (op->getOperands().size() != 2) return rewriter.notifyMatchFailure( op, "only match for the case where operands is of size 2"); auto keys = op.getInputs()[0]; auto indices = op.getInputs()[1]; auto keys_ty = mlir::dyn_cast_or_null<ShapedType>(keys.getType()); auto indices_ty = mlir::dyn_cast_or_null<ShapedType>(indices.getType()); if (!keys_ty || !keys_ty.hasStaticShape() ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
constexpr int kValues = 0; constexpr int kRowSplits = 1; if (func.getFunctionType().getInputs().size() != func.getFunctionType().getResults().size()) { return func.emitError() << "Mismatched number of inputs and outputs."; } int row_splits = func.getFunctionType().getInputs().size() - kRowSplits; if (row_splits == 0) { auto input_values = GetInputType(func, kValues);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
dyn_cast_or_null<TF::TPUReplicatedInputOp>(op)) { // TODO(b/267700110): Handle multiple input/output cases. rewriter.replaceOp(replicated_input_op, replicated_input_op.getInputs()); } else if (auto replicated_output_op = dyn_cast_or_null<TF::TPUReplicatedOutputOp>(op)) { // TODO(b/267700110): Handle multiple input/output cases.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc
// this is a forced/hard-coded constraint. auto qbarrier = rewriter.create<QuantizeCastOp>(op.getLoc(), quantizedType, op.getInputs()); rewriter.replaceOpWithNewOp<DequantizeCastOp>(op, converter.input_type, qbarrier.getResult()); return false; } }; class ConstFakeQuantRewrite
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
llvm::SmallVectorImpl<std::string>* var_handle_shared_names) { Block& block = function.front(); auto func_type = function.getFunctionType(); auto func_arg_types = llvm::to_vector<4>(func_type.getInputs()); llvm::SmallDenseMap<llvm::StringRef, int> var_arg_index_by_name; for (auto var_handle_op : llvm::make_early_inc_range(block.getOps<TF::VarHandleOp>())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
build-logic/documentation/src/main/groovy/gradlebuild/docs/GradleJavadocsPlugin.java
task.setTitle("Gradle API " + version); Javadocs javadocs = extension.getJavadocs(); // TODO: This should be part of Javadoc task task.getInputs().file(javadocs.getJavadocCss()) .withPropertyName("stylesheetFile") .withPathSensitivity(PathSensitivity.NAME_ONLY);
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu May 30 04:10:18 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/fake_quant_utils.h
// constants and the tf.FakeQuantWithMinMaxVarsOp. FetchAttrType min_value, max_value; if (!fetch_min_max_(tf_op, min_value, max_value)) { return failure(); } Value input = tf_op.getInputs(); int quant_dim = -1; auto input_type = mlir::cast<ShapedType>(input.getType()); if (PerAxis) { if (!input_type.hasRank()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
LogicalResult matchAndRewrite(stablehlo::ReduceWindowOp reduce_op, PatternRewriter& rewriter) const override { MLIRContext* ctx = reduce_op.getContext(); ValueRange inputs = reduce_op.getInputs(); // Only handle binary reduce ops for now if (inputs.size() != 1) return failure(); Value reduce_input = inputs[0]; RankedTensorType reduce_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc
FunctionType func_type = func.getFunctionType(); if (llvm::ArrayRef(return_types) == func_type.getResults()) return; auto updated_type = FunctionType::get(func.getContext(), func_type.getInputs(), return_types); func.setType(updated_type); } // TODO(jpienaar): Remove when recursive side-effect modeling is added. bool IsSideEffectFree(func::FuncOp func) { return !func.getBody()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} } else { assert(input_sharding_type == xla::OpSharding::OTHER); if (partitioned_input.getInputs().size() != num_cores_per_replica) return tiled_sharding_mismatched( partitioned_input.getInputs().size()); for (int i = 0; i < sharding.tile_assignment_devices_size(); ++i) { const int assigned_logical_device =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)