- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 44 for output_arg (0.13 sec)
-
tensorflow/compiler/mlir/lite/tests/end2end/disallow_stateful_partitioned_call.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 24 20:05:09 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/function-resource-args.mlir
// CHECK: function // CHECK: signature // CHECK: input_arg // CHECK: type: DT_RESOURCE // CHECK: input_arg // CHECK: type: DT_RESOURCE // CHECK: output_arg // CHECK: type: DT_RESOURCE // CHECK: ret // Check _resource_arg_unique_id for each argument. Since they alias each other, // both values are 0. // CHECK: arg_attr
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/graph-as-function.mlir
// CHECK-NEXT: name: "function0" // CHECK-NEXT: input_arg { // CHECK-NEXT: name: "function0" // CHECK: input_arg { // CHECK-NEXT: name: "function01" // CHECK: output_arg { // CHECK-NEXT: name: "function02" // CHECK: node_def { // CHECK-NEXT: name: "[[NAME:[^"]*]]" // CHECK-NEXT: op: "Identity" // CHECK-NEXT: input: "function0" // CHECK: ret {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.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/tfr/python/tfr_gen.py
op_name = name.qn[1] op_def, _ = self._op_defs.lookup(op_name) if len(op_def.output_arg) == 1: return {_get_type_from_proto(op_def.output_arg[0])}, None return ({tuple(_get_type_from_proto(arg) for arg in op_def.output_arg)}, None) elif f_type == (TFRTypes.PY_BUILTIN_FUNC,): assert name.is_simple() if name == QN('range'):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_test.cc
op { name: "Foo" input_arg { name: "images" description: "Images to process." } input_arg { name: "dim" description: "Description for dim." type: DT_FLOAT } output_arg { name: "output" description: "Description for output." type: DT_FLOAT } attr { name: "T" type: "type" description: "Type for images" allowed_values { list {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 20:04:30 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model_test.cc
in.mutable_composite_tensor()->add_components()->set_name("input1:0"); in.mutable_composite_tensor()->add_components()->set_name("input2:0"); TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"]; out.mutable_composite_tensor()->add_components()->set_name("output2:0"); out.mutable_composite_tensor()->add_components()->set_name("output1:0"); AddSignatureDefToSavedModelBundle(signature_def, "signature_def",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 13:30:31 UTC 2022 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
EXPECT_EQ(f1_rewritten->signature().input_arg(1).type(), DT_RESOURCE); ASSERT_EQ(f1_rewritten->signature().output_arg_size(), 1); EXPECT_EQ(f1_rewritten->signature().output_arg(0).type(), DT_BOOL); // Check node "if" input and output edges. auto node_name_index = g->BuildNodeNameIndex(); const Node *if_node = node_name_index.at("if"); ASSERT_NE(if_node, nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
" signature {" " name: 'AddVariablesFunction'" " input_arg {" " name: 'var'" " type: DT_RESOURCE" " }" " output_arg {" " name: 'sum'" " type: DT_FLOAT" " }" " }" " node_def {" " name: 'read0'" " op: 'ReadVariableOp'" " input: 'var'"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
// Add derived "result_segment_sizes" attr to the created operation. // TODO(b/146937733): Don't use <void> here. set_segment_sizes_attr(output_ranges, op_def->output_arg(), mlir::OpTrait::AttrSizedResultSegments< void>::getResultSegmentSizeAttr()); } } return success(); } private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0)