- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 66 for op_kernel (0.14 sec)
-
tensorflow/compiler/jit/xla_device.cc
}); } } void XlaDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { VLOG(2) << "XlaDevice::Compute " << op_kernel->name() << ":" << op_kernel->type_string(); ShowXlaDeviceDeprecationWarning(jit_device_name_.type_string()); op_kernel->Compute(context); } void XlaDevice::ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,
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/c/kernels/bitcast_op_test.cc
std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); ASSERT_TRUE(status.ok()) << status.ToString(); OpKernelContext::Params params; DummyDevice dummy_device(nullptr); params.device = &dummy_device; params.op_kernel = kernel.get(); gtl::InlinedVector<TensorValue, 4> inputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
return failure(); params_.inputs = inputs; params_.op_kernel = op_kernel.get(); llvm::SmallVector<tensorflow::AllocatorAttributes, 4> output_attr( op_->getNumResults()); params_.output_attr_array = output_attr.data(); tensorflow::OpKernelContext op_context(¶ms_, op_->getNumResults()); device_->Compute(params_.op_kernel, &op_context); status = op_context.status(); if (!status.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/c/test_op1.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 06 02:40:19 UTC 2018 - 875 bytes - Viewed (0) -
tensorflow/c/kernels.cc
namespace tensorflow { namespace { // An OpKernel whose methods delegate to C function pointers. class COpKernel : public OpKernel { public: explicit COpKernel(OpKernelConstruction* ctx, void* (*create_func)(TF_OpKernelConstruction*), void (*compute_func)(void*, TF_OpKernelContext*), void (*delete_func)(void*))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op_test.cc
std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); ASSERT_TRUE(status.ok()) << status.ToString(); OpKernelContext::Params params; DummyDevice dummy_device(nullptr); params.device = &dummy_device; params.op_kernel = kernel.get(); AllocatorAttributes alloc_attrs; params.output_attr_array = &alloc_attrs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
~XlaDevice() override; Allocator* GetAllocator(AllocatorAttributes attr) override TF_LOCKS_EXCLUDED(mu_); void Compute(OpKernel* op_kernel, OpKernelContext* context) override; void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, AsyncOpKernel::DoneCallback done) override; Status Sync() override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/test_op.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 13 00:16:08 UTC 2016 - 874 bytes - Viewed (0) -
tensorflow/compiler/jit/tf_to_hlo_compiler.h
#define TENSORFLOW_COMPILER_JIT_TF_TO_HLO_COMPILER_H_ #include <memory> #include <vector> #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { class TfToHloCompiler { public: TfToHloCompiler() = default; virtual ~TfToHloCompiler() = default; // Compiles a Tensorflow `function` to an HloModuleProto stored in the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/test_opkernels.cc
==============================================================================*/ #include "tensorflow/core/framework/common_shape_fns.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h" namespace tensorflow { namespace tf_mlrt { REGISTER_OP("TestAsyncIdentity") .Input("in: T") .Output("out: T") .Attr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 1.7K bytes - Viewed (0)