- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for tensorlist (0.15 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
}); return has_variant_input || has_variant_output; } // There are 2 standard tf ops which are not TensorList ops that may take as // input a tensorlist. These are tf.AddN and tf.ZeroesLike. Since the runtime // implementation of a tensorlist are not compatible between tf and tflite // we cannot use tflite tensorlist kernels until these cases are handled. bool IsNonTensorListVariantOp(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
if (a.TypeId() == tensorflow::TypeIndex::Make<::tensorflow::TensorList>()) { TF_RETURN_IF_ERROR(ValidateVariantType<::tensorflow::TensorList>(a)); *out = ::tensorflow::TensorList(); return ::tensorflow::TensorListBinaryAdd( cc_ctx, *a.get<::tensorflow::TensorList>(), *b.get<::tensorflow::TensorList>(), out->get<::tensorflow::TensorList>(), cc_binary_add);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.td
"legalized by this pass, then the IR won't be changed so that " "tensorlist ops can pass through (default false).">, Option< "default_to_single_batch_", "default-to-single-batch", "bool", "false", "When specified to true, if the tensorlist ops has unspecified batch " "size, this pass will assume that the batch size is one to proceed "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
OpBuilder builder(list); llvm::SmallVector<int64_t, 8> element_shape; // Infer TensorList element shape from the return type first, and then from // the const element shape operand. We first check the return type because // shape inference might have successfully inferred the element shape from // write operations on the TensorList. if (failed(GetElementShapeFromResultType(list.getType(), &element_shape))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
// meaningless value -1 for non resource outputs. out_desc.input_index = it != output_to_input_alias.end() ? it->getSecond() : -1; // MLIR-based TF-Compiler bridge doesn't support tensorlist output yet. // TODO(ycao): Support tensorlist-type output. out_desc.is_tensor_list = false; } // XLA computation always uses Tuple shape. *xla_output_shape = xla::ShapeUtil::MakeTupleShape(shapes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.h
// Creates a pass that adds control dependencies to keep the relative execution // order of operations with side effects frozen. std::unique_ptr<OperationPass<func::FuncOp>> CreatePinOpsWithSideEffectsPass(); // Legalize TensorList Ops iff all of them are supported. std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTensorListPass(); // Reduce the type precision of some tensor types if all values within that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 21:29:34 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir
// RUN: tf-opt "-tfl-lower-static-tensor-list=allow-tensorlist-pass-through default-to-single-batch" -split-input-file %s | FileCheck %s // ----- // CHECK-LABEL: tensorlistConst func.func @tensorlistConst(%arg0 : tensor<1xi32>) -> tensor<2x3xi32> { // CHECK-DAG: %[[ELEMENT0:.*]] = "tf.Const"() <{value = dense<[0, 1, 2]> : tensor<3xi32>}> : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 39.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
} for (const XlaOutputDescription& descr : compilation_result->outputs) { if (descr.type == DT_VARIANT) { return errors::Unimplemented( "Support for TensorList crossing the XLA/TF boundary " "is not implemented"); } } std::vector<TensorShape> output_tensor_shapes; output_tensor_shapes.reserve(ctx->num_outputs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
mlir::TF::RecvTPUEmbeddingActivationsOp::getOperationName(), context)); supported_ops->insert(OperationName( mlir::TF::SendTPUEmbeddingGradientsOp::getOperationName(), context)); } // Stack, TensorList and TensorArray ops are rewritten during the second phase // of the bridge (compilation of TPUCompile op). They would not match any // legalization/canonicalization pattern and have to be manually added to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// Materialize the MlirPassthroughOp by replacing it with the MLIR module // attached as an attribute. std::unique_ptr<OperationPass<func::FuncOp>> CreateMaterializePassthroughOpPass(); // Replicates the TensorList init op by undoing some CSE needed for correct // shape assignment in shape_inference. std::unique_ptr<OperationPass<func::FuncOp>> CreateReplicateTensorListInitOpsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0)