Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for output_arg (0.17 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/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)
  8. 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)
  9. 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)
  10. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

            auto cf_consider_fn = [disable_constant_folding](const Node* n) {
              if (disable_constant_folding) return false;
              for (const auto& output_arg : n->op_def().output_arg()) {
                if (output_arg.type() == DT_VARIANT) {
                  return false;
                }
              }
              return true;
            };
            GraphOptimizer::Options graph_optimizer_options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
Back to top