- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for OP_REQUIRES_OK (0.16 sec)
-
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) -
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) -
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) -
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) -
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)