Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 82 for attr_value (0.74 sec)

  1. tensorflow/compiler/jit/encapsulate_util.cc

    Status AppendToListAttr(Node* n, const string& attr_name, const string& value) {
      std::vector<T> attr_value;
      Status s = GetNodeAttr(n->attrs(), attr_name, &attr_value);
      if (!s.ok() && s.code() != error::NOT_FOUND) {
        return s;
      }
    
      n->ClearAttr(attr_name);
      attr_value.push_back(value);
      n->AddAttr(attr_name, attr_value);
      return absl::OkStatus();
    }
    
    // Replaces attribute value.
    template <typename T>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc

        }
      }
      return true;
    }
    
    bool HasSymbolRefAttr(Operation* op) {
      for (const auto& attr : op->getAttrs()) {
        Attribute attr_value = attr.getValue();
        if (mlir::isa<SymbolRefAttr>(attr_value)) {
          return true;
        } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
          if (!array_attr.empty() &&
              mlir::isa<SymbolRefAttr>(*array_attr.begin())) {
            return true;
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h

    #include "llvm/ADT/StringRef.h"
    #include "mlir/IR/Operation.h"  // from @llvm-project
    #include "tensorflow/compiler/mlir/tensorflow/utils/export_utils.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/framework/node_def_util.h"
    #include "tensorflow/core/framework/op_def_builder.h"
    #include "tensorflow/core/platform/status.h"
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 2.4K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function.cc

                                      const void* proto, size_t proto_len,
                                      TF_Status* status) {
      tensorflow::AttrValue attr_value;
      if (!attr_value.ParseFromArray(proto, proto_len)) {
        status->status = InvalidArgument(
            "Unparseable AttrValue proto passed to "
            "TF_FunctionSetAttrValueProto");
        return;
      }
    
      auto fdef_or = func->record->mutable_fdef();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc

          mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func",
          func_type, {});
    
      func.addEntryBlock();
      mlir::StringAttr attr_value = builder->getStringAttr("MaxUnpooling2D");
      func->setAttr("tf._implements", attr_value);
      return func;
    }
    
    func::FuncOp createMaxUnpoolingFunc(
        mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 29 21:02:21 UTC 2022
    - 7.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      }
      return true;
    }
    
    bool HasSymbolRefAttr(Operation* op) {
      for (const auto& attr : op->getAttrs()) {
        Attribute attr_value = attr.getValue();
        if (mlir::isa<SymbolRefAttr>(attr_value)) {
          return true;
        } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
          if (!array_attr.empty() &&
              mlir::isa<SymbolRefAttr>(*array_attr.begin())) {
            return true;
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/convert_attr.cc

        case AttrValue::kI:
          return builder->getI64IntegerAttr(value.i());
        case AttrValue::kS:
          return builder->getStringAttr(value.s());
        case AttrValue::kF:
          return builder->getFloatAttr(builder->getF32Type(), value.f());
        case AttrValue::kB:
          return builder->getBoolAttr(value.b());
        case AttrValue::kType: {
          mlir::Type type;
          TF_RETURN_IF_ERROR(ConvertDataType(value.type(), *builder, &type));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  8. tensorflow/c/kernels/bitcast_op_test.cc

                       TensorShape expected_shape, error::Code expected_code) {
      Status status;
      NodeDef def;
      def.set_op("Bitcast");
      def.set_device(DEVICE_CPU);
    
      AttrValue typeAttr;
      SetAttrValue(input_tensor->dtype(), &typeAttr);
    
      AttrValue outTypeAttr;
      SetAttrValue(out_type, &outTypeAttr);
    
      (*def.mutable_attr())["T"] = typeAttr;
      (*def.mutable_attr())["type"] = outTypeAttr;
    
      def.add_input(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/device_compilation_profiler.h

    #define TENSORFLOW_COMPILER_JIT_DEVICE_COMPILATION_PROFILER_H_
    
    #include <cstdint>
    #include <string>
    
    #include "tensorflow/compiler/jit/xla_compile_util.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    
    namespace tensorflow {
    
    // Tracks statistics for device compilation and uses these to determine whether
    // the given cluster should be compiled or not.
    class DeviceCompilationProfiler : public ResourceBase {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 3.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc

    #include "tensorflow/compiler/mlir/quantization/stablehlo/quantization_config.pb.h"
    #include "tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.pb.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/op.h"
    #include "tensorflow/core/framework/op_kernel.h"
    #include "tensorflow/core/framework/op_requires.h"
    #include "tensorflow/core/framework/shape_inference.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top