- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 79 for identity_op (0.24 sec)
-
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/transforms/lower_static_tensor_list.cc
} }; struct ConvertIdentity : public OpConversionPattern<TF::IdentityOp> { using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::IdentityOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { Value input = adaptor.getOperands()[0]; rewriter.replaceOpWithNewOp<TF::IdentityOp>( op, input.getType(), adaptor.getOperands(), op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.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/tensorflow/transforms/resource_device_inference.cc
device_attr, op, result); if (failed(res)) return WalkResult::interrupt(); } else if (auto identity = dyn_cast<IdentityOp>(op)) { LLVM_DEBUG(dump("Visiting ", identity)); // Try to construct IdentityOp's attribute from recorded assignment. if (!GetDeviceAttr(op).empty()) return WalkResult::advance();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K 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/transforms/prepare_tf.cc
// of Identity op is only used by TF ops. struct RemoveIdentity : public OpRewritePattern<TF::IdentityOp> { using OpRewritePattern<TF::IdentityOp>::OpRewritePattern; LogicalResult matchAndRewrite(TF::IdentityOp identity, PatternRewriter &rewriter) const override { // Replace the op with the input if input and result have the same type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K 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) -
tensorflow/c/eager/c_api_test_util.cc
TFE_OpAddInput(op, b, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status); TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); return op; } TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "Identity", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)