- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 85 for identity_op (0.26 sec)
-
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) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_main_function.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 15.4K 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/ir/tf_arith_ops_folder.h
auto is_valid_broadcasting = [](ShapedType operand_ty, ShapedType identity_ty, ShapedType result_ty) -> bool { // Scalar identity is broadcastable to any operand shape, we only need to // check that operand has the same shape as a result. bool scalar_identity = identity_ty.hasRank() && identity_ty.getRank() == 0; if (scalar_identity) return operand_ty == result_ty;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-defs.pbtxt
} } node_def { name: "Identity_1" op: "Identity" input: "cond_true1" attr { key: "T" value { type: DT_INT32 } } } ret { key: "cond_true2" value: "Identity:output:0" } ret { key: "cond_true3" value: "Identity_1:output:0" } } function {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 24 00:20:25 UTC 2020 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/tf_identity_propagation.mlir
%0 = "tf.Identity"(%arg0) {_XlaSharding = ""} : (tensor<i32>) -> tensor<i32> // CHECK: return %[[OUTPUT]] func.return %0 : tensor<i32> } // CHECK-LABEL: func @identity_n // CHECK-SAME: (%[[ARG0:.*]]: tensor<i32>, %[[ARG1:.*]]: tensor<f32>) func.func @identity_n(%arg0: tensor<i32>, %arg1: tensor<f32>) -> (tensor<i32>, tensor<f32>) { // CHECK-NOT: "tf.IdentityN"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Mar 23 23:34:42 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
void runOnOperation() override; }; // Returns the first value in the chain of operands, which is not defined by a // tf.IdentityOp or a tf.ReadVariableOp. Value SkipIdentityAndReadVariable(Value v) { while (auto op = v.getDefiningOp()) { if (!isa<TF::IdentityOp, TF::ReadVariableOp>(op)) break; v = op->getOperand(0); } return v; } void AnnotateParameterReplicationPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
callbacks/associations.go
} } cacheKey := utils.ToStringKey(relPrimaryValues...) if len(relPrimaryValues) != len(rel.FieldSchema.PrimaryFields) || !identityMap[cacheKey] { if cacheKey != "" { // has primary fields identityMap[cacheKey] = true } distinctElems = reflect.Append(distinctElems, rv) } } } if elems.Len() > 0 {
Registered: Wed Jun 12 16:27:09 UTC 2024 - Last Modified: Tue Apr 11 03:06:13 UTC 2023 - 14.3K 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)