- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 56 for identity_op (0.3 sec)
-
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) -
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/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/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) -
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)