- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 44 for tensorlist (0.26 sec)
-
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> legalize_custom_tensor_list_ops( "legalize-custom-tensor-list-ops", llvm::cl::desc("Convert \"tf.TensorList*\" ops to \"tfl.custom_op\"" "if they can all be supported."), llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> serialize_stablehlo_ops( "serialize-stablehlo-ops",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
llvm::cl::location(emit_custom_ops), llvm::cl::init(false)); // NOLINTNEXTLINE static opt<bool, true> lower_tensor_list_ops_flag( "lower-tensor-list-ops", llvm::cl::desc("Lower the TensorList ops within the TFLite dialect"), llvm::cl::location(lower_tensor_list_ops), llvm::cl::init(false)); // NOLINTNEXTLINE static opt<bool, true> strip_debug_info_flag(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K 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) -
tensorflow/c/kernels_test.cc
*tensor_proto = TensorProtoHelpers::GenerateTensorProto(); CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrTensor", v); } TEST_F(TestKernelAttr, TensorList) { struct TensorProtoHelpers { public: static ::tensorflow::TensorProto GenerateTensorProto1() { ::tensorflow::TensorProto tensor_proto; tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"return getElementTypeOrSelf(element_type());">; let extraClassDeclaration = [{ // Returns type of the TensorList element produced by this op. TensorType element_type() { return handle_dtype().getSubtypes()[0]; } // Returns data type of the result handle. Returned type contains type of // the TensorList element as a subtype. VariantType handle_dtype() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
LogicalResult matchAndRewrite(Operation *op, PatternRewriter &rewriter) const override { auto addn_op = cast<AddNOp>(op); // TODO(hinsu): Support variant with TensorList type. tf.AddV2 doesn't // support variant type so variant types require special handling. if (mlir::isa<VariantType>(getElementTypeOrSelf(addn_op.getType()))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0)