Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for OP_REQUIRES_OK (0.14 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc

        string func_name;
        string node_name;
        OP_REQUIRES_OK(ctx, ctx->GetAttr("log_dir_path", &log_dir_path));
        OP_REQUIRES_OK(ctx, ctx->GetAttr("enabled", &enabled_));
        OP_REQUIRES_OK(ctx, ctx->GetAttr("file_name", &file_name));
        OP_REQUIRES_OK(ctx, ctx->GetAttr("func_name", &func_name));
        OP_REQUIRES_OK(ctx, ctx->GetAttr("node_name", &node_name));
        OP_REQUIRES_OK(ctx, ctx->env()->RecursivelyCreateDir(log_dir_path));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 03:12:17 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc

          : OpKernel(context) {
        OP_REQUIRES_OK(context, context->GetAttr("id", &id_));
    
        int calibration_method_value;
        int num_bins;
        float min_percentile;
        float max_percentile;
        OP_REQUIRES_OK(context, context->GetAttr("calibration_method",
                                                 &calibration_method_value));
        OP_REQUIRES_OK(context, context->GetAttr("num_bins", &num_bins));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc

        std::string output_file_path;
        OP_REQUIRES_OK(context,
                       context->GetAttr("output_file_path", &output_file_path));
        OP_REQUIRES_OK(context, context->env()->NewWritableFile(output_file_path,
                                                                &output_file_));
    
        OP_REQUIRES_OK(context, context->GetAttr("ids", &ids_));
        OP_REQUIRES_OK(context, context->GetAttr("calibration_methods",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_device_ops.cc

                 << type_string() << " on an XLA device. This should never happen.";
    }
    
    XlaAssignVariableOp::XlaAssignVariableOp(OpKernelConstruction* c)
        : OpKernel(c) {
      OP_REQUIRES_OK(c, c->GetAttr("dtype", &dtype_));
    }
    
    void XlaAssignVariableOp::Compute(OpKernelContext* context) {
      OP_REQUIRES(context, dtype_ == context->input(1).dtype(),
                  errors::InvalidArgument(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 3K bytes
    - Viewed (0)
  5. tensorflow/compiler/aot/aot_only_var_handle_op.cc

      void Compile(XlaOpKernelContext* context) override;
    
     private:
      string name_;
    };
    
    XlaAotOnlyVarHandleOp::XlaAotOnlyVarHandleOp(OpKernelConstruction* c)
        : XlaOpKernel(c) {
      OP_REQUIRES_OK(c, c->GetAttr("shared_name", &name_));
    }
    
    void XlaAotOnlyVarHandleOp::Compile(XlaOpKernelContext* context) {
      // Look for a resource of the same name. TF also keys that on the container
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 09:57:04 UTC 2024
    - 3K bytes
    - Viewed (0)
Back to top