- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 28 for identity_op (0.13 sec)
-
tensorflow/compiler/jit/xla_compile_util_test.cc
#include "tensorflow/core/kernels/ops_testutil.h" #include "tensorflow/core/tpu/tpu_defs.h" namespace tensorflow { namespace { TEST_F(OpsTestBase, CreateSingleOpGraph) { TF_EXPECT_OK(NodeDefBuilder("identity_op", "Identity") .Input(FakeInput(DT_FLOAT)) .Attr("T", DT_FLOAT) .Finalize(node_def())); TF_EXPECT_OK(InitOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
if (!classes_attr) { // Attempt to parse "_class" from the IdentityOp that follows VariableV2. // For read-only reference variables, IdentityOp should be the only user of // VariableV2. auto identity_op = op->getUsers().begin(); classes_attr = identity_op->getAttrOfType<ArrayAttr>(kClassAttr); if (!classes_attr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
struct TensorDeviceCopyConversionPass : public impl::TensorDeviceCopyConversionPassBase< TensorDeviceCopyConversionPass> { void runOnOperation() override; }; // Folds tf.IdentityOp and tf.IdentityNOp if op device and the argument devices // from the defining ops match. void TensorDeviceCopyConversionPass::runOnOperation() { func::FuncOp func_op = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/split_into_island_per_op_pass.cc
auto identity = builder.create<TF::IdentityOp>(island.getLoc(), operand.getType(), operand); yield.setOperand(0, identity.getOutput()); } else { auto identity_n = builder.create<TF::IdentityNOp>( island.getLoc(), yield.getOperandTypes(), yield.getOperands()); for (const auto& it : llvm::enumerate(identity_n.getResults()))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
class IdentityOp; class IdentityNOp; // Returns if a value corresponds to a constant, returns the matched constant // as an attribute. template <typename AttrT> bool GetValueAsConstant(Value val, AttrT &attr) { while (auto result = mlir::dyn_cast<OpResult>(val)) { Operation *op = result.getOwner(); if (!isa<IdentityOp>(op) && !isa<IdentityNOp>(op)) break;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.cc
// clang-format off // go/keep-sorted start TF::AvgPoolOp, TF::ConcatOp, TF::ConcatV2Op, TF::ExpandDimsOp, TF::IdentityNOp, TF::IdentityOp, TF::MaxPoolOp, TF::PadV2Op, TF::RankOp, TF::ReshapeOp, TF::SelectOp, TF::SelectV2Op, TF::ShapeNOp,
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/tensorflow/transforms/tpu_identity_pruning.cc
} // Removes Identity/IdentityN ops from a region and forwards its operands to its // results. void RemoveIdentityFromRegion(Region& region) { region.walk([](Operation* op) { if (isa<TF::IdentityOp, TF::IdentityNOp>(op)) { op->replaceAllUsesWith(op->getOperands()); op->erase(); } }); } void TPUIdentityPruning::runOnOperation() { SmallVector<tf_device::ClusterOp, 4> clusters;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir
func.func @while_main(%arg0: tensor<?x256x256xf32>) -> (tensor<i32>, tensor<256x256xf32>, tensor<?x256x256xf32>) attributes {tf.entry_function = {inputs = "input", outputs = "Identity,Identity_1,Identity_2"}} { %cst = arith.constant dense<1.000000e+00> : tensor<256x256xf32> %cst_0 = arith.constant dense<0> : tensor<i32> %cst_1 = arith.constant dense<-1> : tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.cc
#include "mlir/IR/Matchers.h" // from @llvm-project #include "mlir/IR/TypeUtilities.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project namespace mlir { namespace TF { class IdentityOp; class IdentityNOp; // Returns the RankedTensorType for the given operand. TensorFlow constant ops // may have non-static shape because the shape is not propagated during constant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/remove_identity_op_pattern.h
// shape information which is necessary for correct pattern matching in this // pass. struct RemoveIdentity : public OpRewritePattern<TF::IdentityOp> { using OpRewritePattern<TF::IdentityOp>::OpRewritePattern; LogicalResult matchAndRewrite(TF::IdentityOp identity, PatternRewriter &rewriter) const override; }; } // namespace quant } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 15 06:13:49 UTC 2023 - 1.7K bytes - Viewed (0)