Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for mutable_attr (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

    // executed on TPU or GPU devices, which do not have `Cast` registered as a
    // runtime OpKernel.
    void RemoveIdentityCast(NodeDef* node_def) {
      auto attr = node_def->mutable_attr();
      if (node_def->op() == "Cast" && attr->contains("SrcT") &&
          attr->contains("DstT") &&
          attr->at("SrcT").type() == attr->at("DstT").type() &&
          attr->contains("Truncate") && !attr->at("Truncate").b()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      TF_RETURN_IF_ERROR(ConvertAttributes(attr.getAttrs().getValue(),
                                           /*attrs_to_ignore=*/{}, remove_ref_type,
                                           value->mutable_func()->mutable_attr()));
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::StringAttr& attr, AttrValue* value) {
      absl::string_view attr_value(attr.getValue().data(), attr.getValue().size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_function.cc

        return;
      }
    
      auto fdef_or = func->record->mutable_fdef();
      if (!fdef_or.ok()) {
        status->status = fdef_or.status();
        return;
      }
    
      (*(fdef_or.value()->mutable_attr()))[string(attr_name)] = attr_value;
    
      status->status = absl::OkStatus();
    }
    
    void TF_FunctionGetAttrValueProto(TF_Function* func, const char* attr_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental.cc

      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
      OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr());
    
      const tensorflow::OpRegistrationData* op_reg_data;
      status->status =
          tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data);
      if (!status->status.ok()) return;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.cc

    // an attribute.
    void AnnotateEagerRuntimeConstructionContext(
        tensorflow::FunctionDef& function_def) {
      tensorflow::AttrValue value;
      SetAttrValue("kEagerRuntime", &value);
      (*function_def.mutable_attr())["_construction_context"] = value;
    }
    
    }  // namespace
    
    extern "C" {
    
    TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top