Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 216 for _kernel (0.14 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py

            self._bias = None
            if bias_shape is not None:
              self._bias = array_ops.constant(
                  np.random.uniform(size=bias_shape), dtype=dtypes.float32
              )
    
            self._kernel = np.random.uniform(size=y_shape).astype('f4')
            self._min = (-0.8, -0.8, -0.9)
            self._max = (0.9, 0.9, 1.0)
    
          @def_function.function(
              input_signature=[
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 18.2K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/scope.h

      /// of setting the status on the scope.
      Scope ExitOnError() const;
    
      /// Return a new scope. All ops created with the new scope will have
      /// kernel_label as the value for their '_kernel' attribute;
      Scope WithKernelLabel(const string& kernel_label) const;
    
      // The following functions are for scope object consumers.
    
      /// Return a unique name, using default_name if an op name has not been
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 09:08:33 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  3. tensorflow/cc/framework/scope.cc

      for (const auto& op : impl()->control_deps_) {
        control_inputs.push_back(op.node());
      }
      builder->ControlInputs(control_inputs);
    
      if (!impl()->kernel_label_.empty()) {
        builder->Attr("_kernel", impl()->kernel_label_);
      }
    
      if (!impl()->colocation_constraints_.empty()) {
        std::vector<string> constraints(impl()->colocation_constraints_.begin(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  4. tensorflow/c/kernels.cc

          c_kernel_ =
              (*create_func)(reinterpret_cast<TF_OpKernelConstruction*>(ctx));
        } else {
          c_kernel_ = nullptr;
        }
      }
    
      void Compute(OpKernelContext* ctx) override {
        (*compute_func_)(c_kernel_, reinterpret_cast<TF_OpKernelContext*>(ctx));
      }
    
      ~COpKernel() override {
        if (delete_func_ != nullptr) {
          (*delete_func_)(c_kernel_);
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  5. tensorflow/c/kernels.h

    // Plugin should implement TF_InitKernel to register kernels. This function
    // should register all kernels in a plugin.
    void TF_InitKernel();
    
    // Allocates a new kernel builder and returns a pointer to it.
    //
    // If non-null, TensorFlow will call create_func when it needs to instantiate
    // the kernel. The pointer returned by create_func will be passed to
    // compute_func and delete_func, thereby functioning as a "this" pointer for
    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/compiler/jit/xla_device_ops.h

    #include "tensorflow/core/kernels/data/iterator_ops.h"
    #include "tensorflow/core/kernels/data/optional_ops.h"
    #include "tensorflow/core/kernels/data/options_dataset_op.h"
    #include "tensorflow/core/kernels/data/prefetch_dataset_op.h"
    #include "tensorflow/core/kernels/fifo_queue.h"
    #include "tensorflow/core/kernels/function_ops.h"
    #include "tensorflow/core/kernels/identity_op.h"
    #include "tensorflow/core/kernels/resource_variable_ops.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 23 19:28:25 UTC 2021
    - 17.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      }
      tensorflow::OpKernel* op_kernel_raw;
      status = params_.function_library->CreateKernel(props, &op_kernel_raw);
      if (!status.ok()) {
        return op_->emitRemark()
               << "failed to create tf2xla kernel: " << status.ToString();
      }
      // Transfer ownership of the kernel to a local smart pointer.
      auto op_kernel = absl::WrapUnique(op_kernel_raw);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  8. 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)
  9. tensorflow/compiler/jit/kernels/xla_ops.cc

              /*may_alias_resource_update=*/false, &kernel, &pjrt_client,
              &pjrt_executable);
        } else {
          status = CompileToLocalExecutable(
              ctx, function_, has_ref_vars_, platform_info_, args, compile_mode,
              /*may_alias_resource_update=*/false, &client, &kernel, &executable);
        }
        if (compile_mode != DeviceCompileMode::kLazy ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
  10. src/cmd/vendor/github.com/google/pprof/internal/elfexec/elfexec.go

    			return b, err
    		}
    	}
    	return nil, nil
    }
    
    // kernelBase calculates the base for kernel mappings, which usually require
    // special handling. For kernel mappings, tools (like perf) use the address of
    // the kernel relocation symbol (_text or _stext) as the mmap start. Additionally,
    // for obfuscation, ChromeOS profiles have the kernel image remapped to the 0-th page.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri May 31 19:48:28 UTC 2024
    - 13.9K bytes
    - Viewed (0)
Back to top