Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for node_def (0.21 sec)

  1. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc

    absl::StatusOr<FunctionDef> TFRDecomposeContext::ExpandNode(
        const NodeDef& node_def, StringPiece func_name) {
      const OpDef* op_def;
      TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef(node_def.op(), &op_def));
      DataTypeVector input_dtys, output_dtys;
      TF_RETURN_IF_ERROR(InputTypesForNode(node_def, *op_def, &input_dtys));
      TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, &output_dtys));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:34:43 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold_utils.cc

      absl::StatusOr<OpKernelRunner> runner = OpKernelRunner::Create(
          node_def->get()->op(), node_def->get()->name(), host_cpu, operands.size(),
          [&](tensorflow::AttrValueMap* attr_value_map) {
            *attr_value_map = node_def->get()->attr();
            return absl::OkStatus();
          },
          fallback_state.device_manager(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc

        const FunctionLibraryDefinition& flib_def) {
      for (const std::string& func_name : flib_def.ListFunctionNames()) {
        const FunctionDef* func_def = flib_def.Find(func_name);
        for (const NodeDef& node_def : func_def->node_def()) {
          if (node_def.op() == kPartitionedCall) return true;
        }
      }
      return false;
    }
    
    }  // namespace
    
    bool IsSupportedByNonReplicatedBridge(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 12:22:33 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h

    // ShapedType for the leading values with ShapedType in the results of the
    // nodes. Set it to true if the returned NodeDef will be executed by the linked
    // TF Eager runtime.
    absl::StatusOr<std::unique_ptr<NodeDef>> ConvertTFDialectOpToNodeDef(
        mlir::Operation* inst, llvm::StringRef name,
        bool ignore_unregistered_attrs);
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 2.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.h

    #include "tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/graph.pb.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/graph/graph.h"
    
    namespace tensorflow {
    namespace tf2xla {
    namespace v2 {
    
    // Given an MLIR module, returns a GraphDef.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.h

    #include "tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/graph.pb.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/graph/graph.h"
    
    namespace tensorflow {
    
    ABSL_DEPRECATED("Use tensorflow::tf2xla::api::ConvertMlirToGraphdef instead.")
    // Given an MLIR module, returns a GraphDef.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 3.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/export_utils.h

    // dialect back into a TensorFlow valid op name.
    absl::StatusOr<llvm::StringRef> GetTensorFlowOpName(llvm::StringRef);
    
    // Converts an MLIR operation to TensorFlow NodeDef with given node name. This
    // name should be unique to the graph it is being inserted into.
    absl::StatusOr<std::unique_ptr<NodeDef>> GetOperationNodeDef(
        mlir::Operation* inst, llvm::StringRef name);
    
    // Converts MLIR attributes with values to their tensorflow equivalent.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.9K bytes
    - Viewed (0)
  8. pkg/kubelet/runonce_test.go

    	// TODO: Factor out "stats.Provider" from Kubelet so we don't have a cyclic dependency
    	volumeStatsAggPeriod := time.Second * 10
    	kb.resourceAnalyzer = stats.NewResourceAnalyzer(kb, volumeStatsAggPeriod, kb.recorder)
    	nodeRef := &v1.ObjectReference{
    		Kind:      "Node",
    		Name:      string(kb.nodeName),
    		UID:       types.UID(kb.nodeName),
    		Namespace: "",
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Apr 18 06:59:54 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  9. pkg/proxy/kubemark/hollow_proxy.go

    				OOMScoreAdj:      ptr.To[int32](0),
    			},
    
    			Client:      client,
    			Proxier:     &FakeProxier{},
    			Broadcaster: broadcaster,
    			Recorder:    recorder,
    			NodeRef: &v1.ObjectReference{
    				Kind:      "Node",
    				Name:      nodeName,
    				UID:       types.UID(nodeName),
    				Namespace: "",
    			},
    		},
    	}
    }
    
    func (hp *HollowProxy) Run() error {
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Apr 22 05:08:41 UTC 2024
    - 2.9K bytes
    - Viewed (0)
Back to top