Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for output_arg (0.38 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/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)
  3. tensorflow/cc/tools/freeze_saved_model_test.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 07 13:30:31 UTC 2022
    - 21.7K bytes
    - Viewed (0)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/cc/framework/cc_op_gen.cc

      strings::StrAppend(out, "  }\n\n");
    
      for (int i = 0; i < op_info.graph_op_def.output_arg_size(); ++i) {
        const string arg_range = strings::StrCat(
            "_outputs_range[\"", op_info.graph_op_def.output_arg(i).name(), "\"]");
        if (op_info.is_list_output[i]) {
          strings::StrAppend(out, "  for (int32 i = ", arg_range, ".first; i < ",
                             arg_range, ".second; ++i)\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 17:22:47 UTC 2023
    - 17K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/xla_cluster_util.cc

        const OpRegistrationData* op_reg;
        if (OpRegistry::Global()->LookUp(call_target.name(), &op_reg).ok()) {
          const OpDef& op = op_reg->op_def;
          if (absl::c_any_of(op.output_arg(), [](const OpDef::ArgDef arg) {
                return arg.is_ref();
              })) {
            return true;
          }
          continue;
        }
    
        callee_ref_nodes.clear();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/cc_op_gen_util.cc

      for (int i = 0; i < graph_op_def.output_arg_size(); ++i) {
        // ApiDef arguments must be in the same order as in OpDef since
        // we initialize ApiDef based on OpDef.
        const auto& arg = graph_op_def.output_arg(i);
        const auto& api_def_arg(api_def.out_arg(i));
        CHECK_EQ(arg.name(), api_def_arg.name());
    
        bool is_list = ArgIsList(arg);
        output_types.push_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  10. platforms/software/dependency-management/src/test/groovy/org/gradle/api/internal/runtimeshaded/RuntimeShadedJarCreatorTest.groovy

            when:
            relocatedJarCreator.create(outputJar, [jarFile1, jarFile2, jarFile3, jarFile4, jarFile5, jarFile6, inputDirectory])
    
            then:
    
            TestFile[] contents = tmpDir.testDirectory.listFiles().findAll { it.isFile() }
            contents.length == 1
            contents[0] == outputJar
            handleAsJarFile(outputJar) { JarFile file ->
                List<JarEntry> entries = file.entries() as List
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed May 22 13:39:49 UTC 2024
    - 20.8K bytes
    - Viewed (0)
Back to top