- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for KernelDef (0.38 sec)
-
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) -
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) -
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) -
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) -
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) -
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)