- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for tensorlist (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} }; while (!worklist.empty()) { tensorlist = worklist.top(); worklist.pop(); // TensorLists are semantically immutable. For example, TensorListSetItem // takes a TensorList as input and produces a TensorList as output. So to // traverse modifications to TensorList and verify that all elements written
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/tensorflow/transforms/tf_passes.td
let summary = "Replicate TensorList init ops for correct shape assignments in shape inference"; let description = [{ If we pass same TensorList to a while op as multiple arguments or just use the same TensorList at multiple places and assign different TensorListSetItem to elements of TensorList, the shape inference is then
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
namespace { // If the input of TensorListGetItemOp is TensorListFromTensorOp and the // TensorListFromTensorOp is only used by TensorListGetItemOp (not modified by // other TensorList ops), we can convert it to a GatherOp. class ConvertTensorListGetItemOpOfTensorListFromTensorOpToGather : public OpRewritePattern<TensorListGetItemOp> { using OpRewritePattern<TensorListGetItemOp>::OpRewritePattern;
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/translate/import_model.cc
shape_inference::InferenceContext* shape_context = shape_refiner_->GetContext(&node); return shape_ic(shape_context); } // Treat TensorList init ops specially here as the op requires knowing its // element dtype. // TODO(jpienaar): Reconsider post refactoring shape functions. if (node.type_string() == "TensorListReserve" ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)