Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 66 for op_kernel (0.32 sec)

  1. 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)
  2. 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)
  3. 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(&params_, 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)
  4. tensorflow/c/test_op1.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/core/framework/op.h"
    #include "tensorflow/core/framework/op_kernel.h"
    
    namespace tensorflow {
    
    REGISTER_OP("TestCApi1").Doc(R"doc(Used to test C API)doc");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 06 02:40:19 UTC 2018
    - 875 bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/c/test_op.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/core/framework/op.h"
    #include "tensorflow/core/framework/op_kernel.h"
    
    namespace tensorflow {
    
    REGISTER_OP("TestCApi").Doc(R"doc(Used to test C API)doc");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Sep 13 00:16:08 UTC 2016
    - 874 bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top