- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for while_loop (0.14 sec)
-
tensorflow/c/c_api.cc
#include "tensorflow/c/experimental/filesystem/modular_filesystem.h" #include "tensorflow/cc/framework/gradients.h" #include "tensorflow/cc/framework/ops.h" #include "tensorflow/cc/framework/scope_internal.h" #include "tensorflow/cc/ops/while_loop.h" #include "tensorflow/cc/saved_model/loader.h" #include "tensorflow/core/distributed_runtime/server_lib.h" #include "tensorflow/core/framework/logging.h" #include "tensorflow/core/framework/op_gen_lib.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
LogicalResult matchAndRewrite(WhileRegionOp while_op, PatternRewriter &rewriter) const override { auto &body_block = while_op.getBody().front(); auto &cond_block = while_op.getCond().front(); bool changed = false; for (int op_idx : llvm::seq<int>(0, while_op.getNumOperands())) { auto body_arg = body_block.getArgument(op_idx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
while_op.getInput().getTypes(), while_op.getOutput().getTypes(), while_op.getBody().getArgumentTypes()); return PropagateShapeToRegions(compatible_types, {&while_op.getCond(), &while_op.getBody()}, max_iterations); } return PropagateShapeToRegions(while_op.getInput().getTypes(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
SmallVector<Value, 4> new_body_yield; SmallVector<bool, 4> removed_operand(while_op.getNumOperands(), false); llvm::SmallVector<Type, 4> types; new_operands.reserve(while_op.getNumOperands()); new_body_yield.reserve(while_op.getNumOperands()); types.reserve(while_op.getNumOperands()); // Remove block arguments not used in either cond or body. This leaves the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto new_while = rewriter.create<TF::WhileRegionOp>( while_op.getLoc(), while_op->getResultTypes(), while_op->getOperands(), /*parallel_iterations=*/10, /*is_stateless=*/false, /*shape_invariant=*/false); new_while.getCond().takeBody(while_op.getCond()); new_while.getBody().takeBody(while_op.getBody()); ReplaceReturnOp(new_while.getCond(), rewriter);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)