- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 33 for identity_op (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
rewriter.setInsertionPointAfter(op); auto const_op = rewriter.create<TF::ConstOp>(op.getLoc(), new_type, tensor_proto_attr); auto new_identity_op = rewriter.create<TF::IdentityOp>( op->getLoc(), const_op.getType(), const_op); return new_identity_op.getResult(); } Operation* LogicsForUniformDequanization(PatternRewriter& rewriter,
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/quantization/tensorflow/cc/constant_fold_test.cc
Operation* op_to_fold = FindOperationOfType<TF::MulOp>(test_func); SmallVector<Value> results = ConstantFoldOpIfPossible(op_to_fold); EXPECT_THAT(results, SizeIs(1)); // No constant-folding since the IdentityOp has `TF_NoConstantFold` trait. auto mul_op = dyn_cast_or_null<TF::MulOp>(results[0].getDefiningOp()); EXPECT_THAT(mul_op, NotNull()); // Even though the preceding CastOp is foldable, it shouldn't be folded since
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
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/convert_control_to_data_outputs.cc
auto chain_arg = func.getArgument(chain_index++); auto src_identity = builder_chain_src.create<TF::IdentityOp>( chain_arg.getLoc(), chain_arg.getType(), chain_arg); auto chain_src_island = CreateIsland(src_identity, {}, builder_chain_src); auto sink_identity = builder_chain_sink.create<TF::IdentityOp>( chain_arg.getLoc(), chain_arg.getType(), chain_arg); auto chain_sink_island =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
// quantizable but is necessary to make the conversion successful. bool IsAlwaysAllowlistedOp(Operation *op) { return llvm::isa< // clang-format off // go/keep-sorted start TF::ConstOp, TF::IdentityOp, TF::PartitionedCallOp, TF::StatefulPartitionedCallOp // go/keep-sorted end // clang-format on >(op); } // LINT.IfChange
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/passes.h
// as GraphDef. std::unique_ptr<OperationPass<ModuleOp>> CreateMergeInitializerFunctionOpsToMainPass(); // Creates a pass that moves & merges the "@tf_quant__save" function to "@main" // function. A new `IdentityOp` will be created. It will have control dependency // to the save function and returns the file_prefix argument (typed // `tensor<!tf_type.string>`). The file_prefix argument, which can be identified
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tests/fallback_to_flex_ops_default.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/tfl_while_outline.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 13.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
// TODO(b/269195256#comment19) change the warning for Identity op to error // when issue with input graph is resolved. Possible issue with python layer // inserting Identity op incorrectly. if (isa<TF::IdentityOp>(op)) { op->emitWarning("TF/XLA TPU bridge input check: found invalid op. ") << op->getName() << " can't be both xla and non-xla"; return true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0)