- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for output_arg (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
// attributes. for (const tensorflow::OpDef::ArgDef& output_arg : op_def_->output_arg()) { int original_size = state_->types.size(); if (!output_arg.number_attr().empty()) { // Same type repeated "repeats" times. Attribute repeats_attr = attrs_[output_arg.number_attr()]; if (!repeats_attr) return InvalidArgument("Missing attribute '", output_arg.number_attr(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/renderers/op_implementation_renderer.cc
void OpImplementationRenderer::RenderExecutionListOp() { ArgView output_arg = op_.OnlyOutput(); Statement("int num_retvals = $0.size()", output_arg.VariableName()); Statement("return " + Call(op_.VariableName(), "Execute", {output_arg.VariableName(), "&num_retvals"})); } void OpImplementationRenderer::RenderExecutionSingleOutput() { ArgView output_arg = op_.OnlyOutput(); Statement("int num_retvals = 1");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 05:51:40 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
"'raw_outputs/class_predictions' type: DT_FLOAT } input_arg: { name: " "'anchors' type: DT_FLOAT } output_arg: { name: " "'TFLite_Detection_PostProcess' type: DT_FLOAT } output_arg: { name: " "'TFLite_Detection_PostProcess:1' type: DT_FLOAT } output_arg: { name: " "'TFLite_Detection_PostProcess:2' type: DT_FLOAT } output_arg: { name: " "'TFLite_Detection_PostProcess:3' type: DT_FLOAT } attr : { name: "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-custom-operation.pbtxt
node { name: "_tf.foo" op: "foo" input: "Constant" } library { function { signature { name: "foo" input_arg { name: "arg" type: DT_INT32 } output_arg { name: "return_value" type: DT_INT32 } } ret { key: "return_value" value: "arg" } } } versions { producer: 62 min_consumer: 12 }
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/graphdef2mlir/graph-function-call.pbtxt
value { s: "cluster" } } } library { function { signature { name: "test_func_name" input_arg { name: "a_0" type: DT_INT32 } output_arg { name: "a" type: DT_INT32 } } ret { key: "a" value: "a_0" } attr { key: "_disable_call_shape_inference" value { b: true
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
signature { name: "test_func_name" input_arg { name: "a_0" type: DT_RESOURCE } input_arg { name: "a_1" type: DT_RESOURCE } output_arg { name: "a" type: DT_RESOURCE } } resource_arg_unique_id { key: 0 value: 0 } resource_arg_unique_id { key: 1 value: 0 }
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/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/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/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/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)