- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetIndexAttr (0.22 sec)
-
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
&guaranteed_constant)) { return false; } return guaranteed_constant; } // Finds the `index` of an _Arg or _Retval node. Status GetIndexAttr(const Node& n, int num_args, int* index) { TF_RETURN_IF_ERROR(GetNodeAttr(n.attrs(), "index", index)); if (*index < 0 || *index >= num_args) { return errors::InvalidArgument("Invalid ", n.type_string(), " number ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
for (int i = res.index(), j = 0; i < op->getNumResults(); i++, j++) { auto index = builder.create<mlir::arith::ConstantOp>( op->getLoc(), builder.getIndexAttr(j)); auto element_op = builder.create<GetElementOp>( op->getLoc(), unconstrainted_tensor_type, new_op.getResult(res.index()), index.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
if (!preceding_build_list) { return failure(); } int64_t num_tensors = preceding_build_list.getNumOperands(); rewriter.replaceOpWithNewOp<arith::ConstantOp>( gl_op, rewriter.getIndexAttr(num_tensors)); return success(); } }; class BuildConstantListAsAttr : public OpRewritePattern<BuildListOp> { using OpRewritePattern<BuildListOp>::OpRewritePattern; public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
Value lhs_shape = rewriter->create<shape::ShapeOfOp>(loc, lhs); Value rhs_shape = rewriter->create<shape::ShapeOfOp>(loc, rhs); Value const_neg2 = rewriter->create<arith::ConstantOp>(loc, rewriter->getIndexAttr(-2)); auto shape_type = shape::ShapeType::get(rewriter->getContext()); auto lhs_splitted = rewriter->create<shape::SplitAtOp>( loc, TypeRange{shape_type, shape_type}, lhs_shape, const_neg2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)