- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 47 for identity_op (0.27 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K 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/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
builder.setInsertionPointToStart(&wrapper_island_op.GetBody()); auto identity_op = builder.create<TF::IdentityOp>( name_loc, /*result_types=*/main_file_prefix_arg.getType(), /*input=*/main_file_prefix_arg); builder.create<tf_executor::YieldOp>(name_loc, identity_op.getResult()); return wrapper_island_op; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
if (cluster_name.has_value()) continue; if (auto identity_op = llvm::dyn_cast_or_null<TF::IdentityOp>(op)) { auto identity_input = identity_op.getInput(); auto output = identity_op.getOutput(); output.replaceAllUsesWith(identity_input); identity_op.erase(); } } } } void TpuV1BridgeExecutorIslandCoarsening::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
while (true) { if (auto block_arg = llvm::dyn_cast<BlockArgument>(val)) { return block_arg.getArgNumber(); } if (auto identity_op = llvm::dyn_cast<TF::IdentityOp>(val.getDefiningOp())) { val = identity_op.getOperand(); } else { return -1; } } }; for (auto call_op :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
.bazelrc
# https://github.com/tensorflow/tensorflow/issues/48919. # Users can still include debug info for a specific kernel, e.g. with: # --config=dbg --per_file_copt=+tensorflow/core/kernels/identity_op.*@-g # Since this .bazelrc file is synced between the tensorflow/tensorflow repo and # the openxla/xla repo, also include debug info for files under xla/. build:dbg --per_file_copt=+.*,-tensorflow.*,-xla.*@-g0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 17:12:54 UTC 2024 - 52.9K 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/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/tf2xla/internal/passes/tpu_cluster_formation.cc
template <typename T> std::tuple<T, llvm::SmallVector<mlir::TF::IdentityOp, 4>> GetSingleUserOfType( OpResult result) { llvm::SmallVector<mlir::TF::IdentityOp, 4> identity_ops; do { Operation* user = result.hasOneUse() ? *result.getUsers().begin() : nullptr; if (auto t = llvm::dyn_cast_or_null<T>(user)) { return std::make_tuple(t, identity_ops); } else if (auto identity =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)