Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 41 for op_registry (0.33 sec)

  1. tensorflow/compiler/jit/partially_decluster_pass.cc

          continue;
        }
    
        DeviceType device_type("");
        TF_RETURN_IF_ERROR(
            DeviceNameToDeviceType(n->assigned_device_name(), &device_type));
        TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type,
                                              n->def(), &input_mtypes,
                                              &output_mtypes));
        for (const Edge* e : n->out_edges()) {
          Node* dst = e->dst();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/scope.cc

    Scope Scope::NewRootScope() {
      Graph* graph = new Graph(OpRegistry::Global());
      ShapeRefiner* refiner =
          new ShapeRefiner(graph->versions(), graph->op_registry());
      return Scope(new Impl(graph, new Status, new Impl::NameMap, refiner,
                            /* disable_shape_inference */ false));
    }
    
    Scope Scope::DisabledShapeInferenceScope() {
      Graph* graph = new Graph(OpRegistry::Global());
      ShapeRefiner* refiner =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

    Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib,
                       std::unique_ptr<Graph>* result) {
      auto graph = std::make_unique<Graph>(OpRegistry::Global());
      TF_RETURN_IF_ERROR(s.ToGraph(graph.get()));
      FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib);
    
      // Assign all nodes to the CPU device.
      static const char* kCpuDevice = "/job:localhost/replica:0/task:0/cpu:0";
      for (Node* n : graph->nodes()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/shape_inference.cc

                       const tensorflow::FunctionLibraryDefinition* fnlib_def,
                       GraphShapeInfo* shape_info) {
      ShapeRefiner shape_refiner(graph->versions(), graph->op_registry());
      shape_refiner.set_require_shape_inference_fns(false);
      // TODO(dlibenzi): Verify if it is worth trying to infer shaped within
      // functions. Some functions can be called at multiple locations with
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

              e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":",
              e->dst_input());
        }
      }
    
      auto output = std::make_unique<Graph>((*graph)->op_registry());
      TF_RETURN_WITH_CONTEXT_IF_ERROR(
          EncapsulateSubgraphsInFunctions(
              kXlaClusterIdAttr, **graph, RewriteSubgraph,
              /*reuse_existing_functions=*/true, &output, flib_def),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/build_xla_ops_pass.cc

      DeviceType device_type("");
      TF_RETURN_IF_ERROR(
          DeviceNameToDeviceType(n->assigned_device_name(), &device_type));
      TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(),
                                            device_type, n->def(), &input_mtypes,
                                            &output_mtypes));
      return output_mtypes;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  7. platforms/documentation/docs/src/snippets/native-binaries/cunit/groovy/libs/cunit/2.1-2/include/CUnit/TestDB.h

     *  its destruction will be handled by the framework.  ppRegistry may not be 
     *  NULL (checked by assertion), but *ppRegistry can be NULL (in which case the
     *  function has no effect).  Note that *ppRegistry will be set to NULL on return.
     *
     *  @param ppRegistry Address of a pointer to the registry to destroy (non-NULL).
     */
    
    CU_EXPORT 
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Mon Nov 27 17:53:42 UTC 2023
    - 40.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/utils/tftext_utils.cc

        }
      }
      return failure();
    }
    
    bool IsTFTextRegistered(const tensorflow::OpRegistry* op_registery) {
      const std::vector<std::string> kTFTextOps = {
          "WhitespaceTokenizeWithOffsets",
      };
      for (const auto& iter : kTFTextOps) {
        if (op_registery->LookUp(iter)) {
          return true;
        }
      }
      return false;
    }
    
    }  // namespace TFL
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.4K bytes
    - Viewed (0)
  9. pkg/registry/core/service/ipallocator/controller/repair_test.go

    	if err := r.runOnce(); err != nil {
    		t.Fatal(err)
    	}
    	if !ipregistry.updateCalled || ipregistry.updated == nil || ipregistry.updated.Range != cidr.String() || ipregistry.updated != ipregistry.item {
    		t.Errorf("unexpected ipregistry: %#v", ipregistry)
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sat Sep 30 15:46:06 UTC 2023
    - 21.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc

        auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg1, 0);
        std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
        TF_CHECK_OK(s.ToGraph(g.get()));
        FunctionDef *xla_fdef = fdl.add_function();
        TF_CHECK_OK(GraphToFunctionDef(*g, "f3", xla_fdef));
      }
      FunctionLibraryDefinition fld(OpRegistry::Global(), fdl);
    
      // Build the XLA computation graph.
      // "arg0" (T=DT_RESOURCE), "arg1" (T=DT_INT32)
    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