- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 111 for _output_shapes (1.13 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/tf_tpu_embedding_ops.mlir
tf_executor.fetch } func.return } // CHECK: name: "RecvTPUEmbedding" // CHECK-NEXT: op: "RecvTPUEmbeddingActivations" // CHECK-NEXT: attr { // CHECK-NEXT: key: "_output_shapes" // CHECK-NEXT: value { // CHECK-NEXT: list { // CHECK-NEXT: shape { // CHECK-NEXT: dim { // CHECK-NEXT: size: 512 // CHECK-NEXT: } // CHECK-NEXT: dim {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/tf-legacy-call.mlir
tf_executor.fetch %arg0 : tensor<*xi32> } func.return %0 : tensor<*xi32> } // CHECK: node { // CHECK: name: "tf.LegacyCall" // CHECK-NEXT: op: "foo0" // CHECK: attr { // CHECK-NEXT: key: "_output_shapes" // CHECK-NEXT: value { // CHECK-NEXT: list { // CHECK-NEXT: shape { // CHECK: attr { // CHECK-NEXT: key: "_tpu_replicate" // CHECK-NEXT: value { // CHECK-NEXT: s: "cluster"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_import_options.h
// general. struct MLIRImportOptions { // If true, functionalize the input graph before importing it into MLIR. bool upgrade_legacy = false; // Whether to unconditionally use the shape set via _output_shapes on import. bool unconditionally_use_set_output_shapes = false; // Apply default attributes from the op definition to the loaded op. bool add_default_attributes = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 20 13:19:26 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
// `ignore_unregistered_attrs` argument is set to true, the attributes which are // not in the op registry will be ignored. If the `ignore_unregistered_attrs` // argument is not set to true, _output_shapes attribute is added to nodes with // ShapedType for the leading values with ShapedType in the results of the // nodes. Set it to true if the returned NodeDef will be executed by the linked // TF Eager runtime.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
end++; if (begin != end) { mlir::TF::ResultShapeRange output_shapes = { mlir::TF::ResultShapeIterator(begin), mlir::TF::ResultShapeIterator(end)}; SetShapeAttribute("_output_shapes", output_shapes, attributes); } } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
%4 = "tf.Add"(%2, %arg2) : (tensor<?x8x10xf32>, tensor<8x10xf32>) -> tensor<?x8x10xf32> %5 = "tf.Add"(%arg1, %arg2) : (tensor<8x10xf32>, tensor<8x10xf32>) -> tensor<8x10xf32> %6 = "tf.Const"() {_output_shapes = ["tfshape$"], device = "/device:CPU:0", dtype = f32, value = dense<1.000000e+00> : tensor<f32>} : () -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc
llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> unconditionally_use_set_output_shapes( "tf-enable-unconditionally-use-set-output-shapes-on-import", llvm::cl::desc("Enable using the _output_shapes unconditionally on import " "(temporary)"), llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> enable_soft_placement( "tf-enable-soft-placement-on-import",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 10 20:59:50 UTC 2023 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-as-function.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 24 00:18:34 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h
bool restrict_functionalization_to_compiled_nodes = false; // If true, enables shape inference on input. // TODO(jpienaar): This will be removed shortly. bool enable_shape_inference = true; // _output_shapes is an unregistered attribute which is used during // GraphConstructor::ConvertGraph to override shapes. It is unfortunately // not always set correctly (which is undesirable and should be addressed)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 04:56:10 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
// Create a bad output shape attr. AttrValue shape_attr; TensorShapeProto* shape_proto = shape_attr.mutable_list()->add_shape(); shape_proto->add_dim()->set_size(1); builder.Attr("_output_shapes", shape_attr); TF_RETURN_IF_ERROR(builder.Finalize(&node)); return CreateSingleOpGraph(node, {}, {DataType::DT_INT32}); } absl::Status BuildHloFromGraph(Graph& graph, bool use_output_shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0)