Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 93 for attr_val (0.2 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

            float_func.emitError("The attr_map attribute is malformed");
            return failure();
          }
          identifier_to_attr.insert(
              {llvm::StringRef(std::string(key_and_value_pair[1])),
               inner_op.getAttr(
                   llvm::StringRef(std::string(key_and_value_pair[1])))});
        }
      }
    
      // Set the attributes for ops with the attr_map attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/cc_op_gen_util.h

    #define TENSORFLOW_CC_FRAMEWORK_CC_OP_GEN_UTIL_H_
    
    #include <string>
    #include <unordered_map>
    #include <utility>
    #include <vector>
    
    #include "tensorflow/core/framework/api_def.pb.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/attr_value_util.h"
    #include "tensorflow/core/framework/op_def_util.h"
    #include "tensorflow/core/framework/op_gen_lib.h"
    #include "tensorflow/core/framework/tensor.pb.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc

              device.has_job && device.job == jobType) {
            for (const auto& attr : node->attrs()) {
              auto attr_key = attr.first;
              auto attr_value = attr.second;
              if (attr_key == attrKey &&
                  attr_value.value_case() == AttrValue::kType &&
                  attr_value.type() == DT_RESOURCE) {
                return true;
                break;
              }
            }
          }
        }
      }
      return false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 12:22:33 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

                                         IntegerAttr attr_value, Operation* op,
                                         ModuleOp& module) {
      const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name);
      if (ex_attr_value == nullptr) {
        module->setAttr(attr_name, attr_value);
        return success();
      }
      if (ex_attr_value == attr_value) {
        return success();
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver.mlir

      %0 = "tf.Conv2D"(%output, %cst) <{data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "SAME", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true}> {attr_map = "0:strides,1:use_cudnn_on_gpu,2:padding,3:explicit_paddings,4:dilations", device = ""} : (tensor<1x3x4x3xf32>, tensor<2x3x3x2xf32>) -> tensor<1x2x2x2xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/utils/convert_attr.h

    #define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_UTILS_CONVERT_ATTR_H_
    
    #include "mlir/IR/Attributes.h"  // from @llvm-project
    #include "mlir/IR/Builders.h"  // from @llvm-project
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tsl/platform/statusor.h"
    
    namespace tensorflow {
    
    using tsl::StatusOr;
    
    // Converts non func AttrValue proto into an MLIR attribute. Func attribute is
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 1.6K bytes
    - Viewed (0)
  9. tensorflow/c/eager/abstract_op_attrs.h

    #ifndef TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    #define TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    
    #include "absl/container/inlined_vector.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    
    // Attributes of an op.
    class AbstractOpAttrs {
     protected:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 26 22:20:27 UTC 2021
    - 2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/tests/propagate_quantize_type.mlir

        %0 = "tf.Identity"(%cst) : (tensor<2x1024xi8>) -> tensor<2x1024xi8>
        %1 = "tf.Cast"(%0) {Truncate = false} : (tensor<2x1024xi8>) -> tensor<2x1024xf32>
        %2 = "tf.MatMul"(%arg0, %1) {attr_map = "0:transpose_a,1:transpose_a", device = "", transpose_a = false, transpose_b = false} : (tensor<1x2x2x2xf32>, tensor<2x1024xf32>) -> tensor<*xf32>
        %3 = "tf.Mul"(%2, %cst_0) : (tensor<*xf32>, tensor<f32>) -> tensor<*xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 6.6K bytes
    - Viewed (0)
Back to top