Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for UpdateBuilder (0.15 sec)

  1. tensorflow/cc/ops/const_op.h

      auto cast_builder = NodeBuilder(cast_op_name, "Cast")
                              .Input(orig_const)
                              .Attr("DstT", DataTypeToEnum<DstT>::v());
      scope.UpdateBuilder(&cast_builder);
      Node* ret;
      scope.UpdateStatus(cast_builder.Finalize(scope.graph(), &ret));
      if (!scope.ok()) return Output();
      scope.UpdateStatus(scope.DoShapeInference(ret));
      return Output(ret, 0);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 17 09:17:01 UTC 2020
    - 2.8K bytes
    - Viewed (0)
  2. tensorflow/cc/ops/const_op.cc

      const string unique_name = scope.GetUniqueNameForOp("Const");
      auto builder = NodeBuilder(unique_name, "Const")
                         .Attr("value", value)
                         .Attr("dtype", dtype);
      scope.UpdateBuilder(&builder);
      scope.UpdateStatus(builder.Finalize(graph, &ret));
      if (!scope.ok()) return Output();
    
      scope.UpdateStatus(scope.DoShapeInference(ret));
      if (!scope.ok()) return Output();
    
      return Output(ret);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 03 20:58:54 UTC 2017
    - 2.7K bytes
    - Viewed (0)
  3. tensorflow/cc/framework/scope.cc

      if (ok()) {
        GraphDef graph_def;
        graph()->ToGraphDef(&graph_def);
        UpdateStatus(ConvertGraphDefToGraph(opts, std::move(graph_def), g));
      }
      return *impl()->status_;
    }
    
    void Scope::UpdateBuilder(NodeBuilder* builder) const {
      std::vector<Node*> control_inputs;
      for (const auto& op : impl()->control_deps_) {
        control_inputs.push_back(op.node());
      }
      builder->ControlInputs(control_inputs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  4. tensorflow/cc/ops/while_loop.cc

      std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input});
      const string unique_name = scope.GetUniqueNameForOp("Merge");
      NodeBuilder builder = NodeBuilder(unique_name, "Merge").Input(input_list);
      scope.UpdateBuilder(&builder);
    
      Node* merge_node;
      TF_RETURN_IF_ERROR(builder.Finalize(scope.graph(), &merge_node));
      TF_RETURN_IF_ERROR(scope.DoShapeInference(merge_node));
      *merge_output = Output(merge_node, 0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  5. tensorflow/cc/framework/scope.h

      // START_SKIP_DOXYGEN
    
      /// Update the builder with properties accumulated in this scope. Does not set
      /// status().
      // TODO(skyewm): NodeBuilder is not part of public API
      void UpdateBuilder(NodeBuilder* builder) const;
      // END_SKIP_DOXYGEN
    
      CompositeOpScopes GetCompositeOpScopes(const string& composite_op_name) const;
    
      bool ok() const;
    
      // TODO(skyewm): Graph is not part of public API
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 09:08:33 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/cc_op_gen.cc

        strings::StrAppend(&body, spaces, ".Attr(\"", graph_attr.name(), "\", ",
                           attr_name, ")\n");
      }
      strings::StrAppend(&body, "  ;\n");
      strings::StrAppend(&body, "  ", scope_str, ".UpdateBuilder(&builder);\n");
      strings::StrAppend(&body, "  ", scope_str, ".UpdateStatus(builder.Finalize(",
                         scope_str, ".graph(), &ret));\n");
      strings::StrAppend(&body, "  ", return_on_error, "\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 17:22:47 UTC 2023
    - 17K bytes
    - Viewed (0)
  7. tensorflow/cc/gradients/array_grad.cc

          ::tensorflow::NodeBuilder(unique_name, "ConcatOffset")
              .Input(::tensorflow::ops::AsNodeOut(scope, op.input(dim_index)))
              .Input(::tensorflow::ops::AsNodeOutList(scope, shapes.output));
      scope.UpdateBuilder(&builder);
      ::tensorflow::Node* concat_offset_node;
      scope.UpdateStatus(builder.Finalize(scope.graph(), &concat_offset_node));
      scope.UpdateStatus(scope.DoShapeInference(concat_offset_node));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/deadness_analysis_test.cc

      auto builder = NodeBuilder(unique_name, "_SwitchN")
                         .Input(_data)
                         .Input(_output_index)
                         .Attr("num_outs", num_outs);
      scope.UpdateBuilder(&builder);
      scope.UpdateStatus(builder.Finalize(scope.graph(), &ret));
      if (!scope.ok()) return;
      scope.UpdateStatus(scope.DoShapeInference(ret));
      for (int32_t i = 0; i < ret->num_outputs(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
Back to top