Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 36 for output_arg (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/lite/tests/end2end/disallow_stateful_partitioned_call.pbtxt

          name: "function"
          input_arg {
            name: "inputs"
            type: DT_FLOAT
          }
          input_arg {
            name: "statefulpartitionedcall_args_1"
            type: DT_RESOURCE
          }
          output_arg {
            name: "identity"
            type: DT_FLOAT
          }
          is_stateful: true
        }
        node_def {
          name: "Identity"
          op: "Identity"
          input: "inputs"
          attr {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 24 20:05:09 UTC 2023
    - 2.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-as-function.pbtxt

          name: "function"
          input_arg {
            name: "inputs"
            type: DT_FLOAT
          }
          input_arg {
            name: "statefulpartitionedcall_args_1"
            type: DT_RESOURCE
          }
          output_arg {
            name: "identity"
            type: DT_FLOAT
          }
          is_stateful: true
        }
        node_def {
          name: "Identity"
          op: "Identity"
          input: "inputs"
          attr {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 24 00:18:34 UTC 2023
    - 5K bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top