- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 33 for identity_op (0.15 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/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) -
CONTRIBUTING.md
`--per_file_copt` bazel option. For example, if you want to debug the Identity op, which are in files starting with `identity_op`, you can run ```bash bazel build --config=dbg --per_file_copt=+tensorflow/core/kernels/identity_op.*@-g //tensorflow/tools/pip_package:build_pip_package ```
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 11:45:51 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_test.cc
EXPECT_TRUE(compilation_result == nullptr); EXPECT_TRUE(xla_executable == nullptr); } TEST_F(OpsTestBase, CompileSingleOpSuccess) { 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 Mar 12 06:33:33 UTC 2024 - 19.8K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
} } // Checks if an operation can have TPU devices propagated through. bool IsSupportedOpToSetDevice(Operation& op) { return IsSupportedExecutorOp(op) || isa<TF::IdentityOp, TF::IdentityNOp, TF::ShapeOp>(op); } // Finds nonconflicting TPU device for an operation from its operands. If an // operand has no device or a non TPU device, or if there are conflicting
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/tensorflow/analysis/resource_alias_analysis.cc
// in the Island body. if (value == island.getControl()) break; value = island.GetYield().getOperand(res_index); } else if (isa<IdentityNOp, IdentityOp>(op)) { value = op->getOperand(res_index); } else if (auto call = dyn_cast<CallOpInterface>(op)) { func::FuncOp func = dyn_cast<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
callbacks/preload.go
} if results, ok := joinIdentityMap[utils.ToStringKey(fieldValues...)]; ok { joinKey := utils.ToStringKey(joinFieldValues...) identityMap[joinKey] = append(identityMap[joinKey], results...) } } _, foreignValues = schema.GetIdentityFieldValuesMap(tx.Statement.Context, joinResults, joinRelForeignFields) } else { for _, ref := range rel.References {
Registered: Wed Jun 12 16:27:09 UTC 2024 - Last Modified: Wed Jun 12 10:52:33 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
callee.getArgument(use.getOperandNumber()), module, infer_from_op); if (type_from_callee.has_value()) return type_from_callee; } else if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(use.getOwner())) { auto type_from_alias = GetElementTypeFromAccess( use.getOwner()->getResult(use.getOperandNumber()), module, infer_from_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0)