Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for KernelDef (0.38 sec)

  1. tensorflow/cc/framework/fuzzing/op_fuzzing.bzl

            name,
            op_def_src,
            api_def_srcs = [],
            kernel_deps = []):
        """
        Generates fuzzers for several groups of ops.
    
        For each one we need the corresponding OpDef, ApiDef and KernelDef,
        since they all can contain constraints for the inputs.
    
        Args:
            name: the name of the fuzz artifact
            op_def_src: op definitions
            api_def_srcs: api definitions
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 07 19:14:57 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.cc

        // If user doesn't provide a serialized KernelDef, use the kernel builder
        // to build a new one.
        tensorflow::kernel_factory::OpKernelRegistrar(
            builder->cc_builder->Build(), name,
            std::make_unique<tensorflow::KernelBuilderFactory>(builder));
    
        TF_SetStatus(status, TF_OK, "");
        return;
      }
    
      tensorflow::KernelDef* kernel_def = new tensorflow::KernelDef();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_device.cc

      XlaOpRegistry::RegisterCompilationKernels();
      XlaDeviceOpRegistrations* registrations = new XlaDeviceOpRegistrations;
      for (const KernelDef* jit_def : XlaOpRegistry::DeviceKernels(
               jit_device,
               /*include_compilation_only_kernels=*/false)) {
        KernelDef* def = new KernelDef(*jit_def);
        const std::unordered_set<std::string>* constant_inputs =
            XlaOpRegistry::CompileTimeConstantInputArgNames(def->op());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 21:05:42 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc

    // in sync.
    TEST_F(LegalizationOpConfigTest, MlirLoweringWithoutXlaKernel) {
      tensorflow::XlaOpRegistry::RegisterCompilationKernels();
      std::vector<const tensorflow::KernelDef*> kernel_defs =
          tensorflow::XlaOpRegistry::DeviceKernels(
              tensorflow::DEVICE_CPU_XLA_JIT,
              /*include_compilation_only_kernels=*/true);
    
      std::set<std::string> xla_op_kernels;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 30 03:31:01 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  5. tensorflow/c/kernels.h

    // registration fails, the given status will be populated.
    //
    // This method is the same as TF_RegisterKernelBuilder except it takes in a
    // serialized KernelDef, and uses it for registration, instead of building a new
    // one. Users can choose to not provide a serialized KernelDef and in that case
    // it's identical to TF_RegisterKernelBuilder.
    TF_CAPI_EXPORT extern void TF_RegisterKernelBuilderWithKernelDef(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  6. tensorflow/c/kernels_test.cc

          .Output("output1: uint8")
          .Attr("SomeDataTypeAttr: type");
    
      TF_KernelBuilder* builder = TF_NewKernelBuilder(
          op_name, device_name, &MyCreateFunc, &MyComputeFunc, &MyDeleteFunc);
    
      KernelDef kernel_def;
      kernel_def.set_op(op_name);
      kernel_def.set_device_type(device_name);
      std::string kernel_def_str = kernel_def.SerializePartialAsString();
    
      {
        TF_Status* status = TF_NewStatus();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
Back to top