- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for _disable_call_shape_inference (0.45 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/legalize_tfg.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-call.pbtxt
type: DT_INT32 } } ret { key: "a" value: "a_0" } attr { key: "_disable_call_shape_inference" value { b: true } } } } # CHECK: func @main
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-library.pbtxt
# CHECK-LABEL: func @main() { # CHECK: "tf.LegacyCall"() <{_disable_call_shape_inference = true, f = @foo0}> {device = ""} # CHECK: "tf.LegacyCall"() <{_disable_call_shape_inference = false, f = @bar0}> {device = ""} # CHECK-LABEL: func private @foo0() # CHECK: "tf.LegacyCall"() <{_disable_call_shape_inference = false, f = @bar0}> {device = ""}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-resource-args.pbtxt
} } attr { key: "shared_name" value { s: "x" } } } node { name: "func_call" op: "test_func_name" input: "x" input: "x" attr { key: "_disable_call_shape_inference" value { b: true } } } library { function { signature { name: "test_func_name" input_arg { name: "a_0" type: DT_RESOURCE }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-name-bug.pbtxt
min_consumer: 12 } # Verify that functions from the library are properly imported. # CHECK-LABEL: func @main() { # CHECK: "tf.LegacyCall"() <{_disable_call_shape_inference = false, f = @foo110}> {device = ""} # CHECK: "tf.LegacyCall"() <{_disable_call_shape_inference = false, f = @foo111}> {device = ""} # CHECK-LABEL: func private @foo110()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-custom-operation.pbtxt
# the names are matching between the function definition and the uses / call # site (a numerical suffix may be appended). # CHECK: "tf.LegacyCall"(%outputs) <{_disable_call_shape_inference = false, f = @foo0}> {device = ""}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/convert_to_legacy_compile_and_replicate_attributes.mlir
// CHECK-LABEL: func @convert_to_legacy_attribute func.func @convert_to_legacy_attribute(%arg0: tensor<*xf32>, %arg1: tensor<f32>, %arg2: tensor<f32>) -> tensor<*xf32> attributes {tf._disable_call_shape_inference = true, tf.signature.is_stateful} { // CHECK-NOT: _replication_info = %0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getSuccessors(), new_regions); inner_op = rewriter.create(state); } else { bool disable_call_shape_inference = false; if (op->hasAttr("_disable_call_shape_inference")) { disable_call_shape_inference = op->getAttrOfType<BoolAttr>("_disable_call_shape_inference") .getValue(); } inner_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
$config, $config_proto, $executor_type), (CallOp $f, $args), [(ArgTypesMatchCallee $op, $args, $f)]>; } // The extra attr on this op is _disable_call_shape_inference, which we ignore // in the bridge. def : Pat<(TF_LegacyCallOp:$op $args, FlatSymbolRefAttr:$f, $attr), (CallOp $f, $args), [(ArgTypesMatchCallee $op, $args, $f)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
mlir::SymbolRefAttr val = mlir::SymbolRefAttr::get(builder_.getContext(), node_type_name); result.addAttribute("f", val); if (!result.attributes.get("_disable_call_shape_inference")) { result.addAttribute("_disable_call_shape_inference", builder_.getBoolAttr(false)); } } auto composite_control_flow_op = [&](const std::string& name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)