Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 41 for type_attr (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

                                   "'");
          TypeAttr type_attr = mlir::dyn_cast<TypeAttr>(attr);
          if (!type_attr)
            return InvalidArgument("Attribute '", output_arg.type_attr(),
                                   "' required for output '", output_arg.name(),
                                   "' isn't a type attribute");
          state_->types.push_back(UnrankedTensorType::get(type_attr.getValue()));
        } else if (!output_arg.type_list_attr().empty()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
      (*node_def->mutable_attr())["T"] = type_attr;
    
      AttrValue index_attr;
      index_attr.set_i(index);
      (*node_def->mutable_attr())["index"] = index_attr;
    
      if (auto device_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
      (*node_def->mutable_attr())["T"] = type_attr;
    
      AttrValue index_attr;
      index_attr.set_i(index);
      (*node_def->mutable_attr())["index"] = index_attr;
    
      if (auto device_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  4. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc

                std::string type = "DT_UINT8";
    
                if (arg.type() != DT_INVALID) {
                  type = DataType_Name(arg.type());
                } else if (!arg.type_attr().empty()) {
                  OpDef_AttrDef attr =
                      *FindAttr(arg.type_attr(), op_info.graph_op_def);
                  if (attr.has_default_value() &&
                      attr.default_value().value_case() == AttrValue::kType) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 27 16:26:51 UTC 2024
    - 13K bytes
    - Viewed (0)
  5. tensorflow/cc/framework/cc_op_gen_util.cc

      }
      return string(name);
    }
    
    void InferArgAttributes(const OpDef::ArgDef& arg,
                            std::unordered_map<string, string>* inferred_attrs) {
      if (!arg.type_attr().empty()) {
        gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name());
      } else if (!arg.type_list_attr().empty()) {
        gtl::InsertIfNotPresent(inferred_attrs, arg.type_list_attr(), arg.name());
      }
      if (!arg.number_attr().empty()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

        if (matchPattern(cst_tensor_op.getArg(), m_Constant(&array))) {
          llvm::DenseSet<Type> all_types;
          for (auto it : array) {
            TypedAttr typed_attr = it.dyn_cast<TypedAttr>();
            if (!typed_attr) return failure();
            all_types.insert(typed_attr.getType());
          }
          if (all_types.size() != 1) return failure();
          ShapedType new_out_type = RankedTensorType::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc

      llvm::SmallVector<Attribute, 2> shape_attrs;
      llvm::SmallVector<Attribute, 2> type_attrs;
      for (Type type : dataset_types) {
        shape_attrs.push_back(
            TF::ShapeAttr::get(builder.getContext(), mlir::cast<ShapedType>(type)));
        type_attrs.push_back(TypeAttr::get(getElementTypeOrSelf(type)));
      }
    
      auto anonymous_iterator = builder.create<AnonymousIteratorV3Op>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfr/passes/decompose.cc

              }
              attribute = TypeAttr::get(type);
            }
            Value attr_cst;
            // Wrap these special attributes as a special TFR constant, so the SSA
            // value has a valid type to be used as TFR function argument. These
            // attributes are not expected to be manipulated by the lowering passes.
            if (mlir::isa<TypeAttr>(attribute) || mlir::isa<ArrayAttr>(attribute) ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

          /*asymmetric_quantize_inputs=*/mlir::BoolAttr(),
          /*input_to_input_intermediate=*/mlir::TypeAttr(),
          /*input_to_forget_intermediate=*/mlir::TypeAttr(),
          /*input_to_cell_intermediate=*/mlir::TypeAttr(),
          /*input_to_output_intermediate=*/mlir::TypeAttr(),
          /*effective_hidden_scale_intermediate=*/mlir::TypeAttr());
    
      // Cast the static shaped lstm result to FuncOp's signature -
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/quantization/quantization_context.cc

            input_specs.push_back(original_input_specs[i]);
          } else if (requantize.pos == RequantizeState::ON_OUTPUT) {
            input_specs.push_back(TypeAttr::get(requantize.params));
          } else {
            input_specs.push_back(TypeAttr::get(state.params));
          }
        }
        op->setAttr("input_specs", ArrayAttr::get(context, input_specs));
    
        llvm::SmallVector<Attribute, 4> output_specs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 08 01:38:03 UTC 2024
    - 13.1K bytes
    - Viewed (0)
Back to top