- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 52 for _output_shapes (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/string-attr.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 29 04:41:05 UTC 2021 - 713 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/device_conversion.mlir
%arg1: tensor<1x3xf32> {tf_saved_model.index_path = [0]}) -> (tensor<3x3xf32> {tf_saved_model.index_path = []}) { // CHECK: {{%.*}} = corert.get_op_handler %arg0 "/device:GPU:0" %2 = "tf.MatMul"(%arg0, %arg1) {T = f32, _output_shapes = ["tfshape$dim { size: 3 } dim { size: 3 }"], device = "/device:GPU:0", transpose_a = false, transpose_b = false} : (tensor<3x1xf32>, tensor<1x3xf32>) -> tensor<3x3xf32> func.return %2 : tensor<3x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 645 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/device-arg-retval-attr.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Dec 07 17:45:22 UTC 2020 - 1.6K bytes - Viewed (0) -
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/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)