Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 33 for NamedAttribute (0.17 sec)

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

      if (auto device_attr =
              func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr))
        *node_def->mutable_device() = device_attr.getValue().str();
    
      llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs =
          mlir::function_interface_impl::getArgAttrs(func, index);
      absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
    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

      if (auto device_attr =
              func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr))
        *node_def->mutable_device() = device_attr.getValue().str();
    
      llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs =
          mlir::function_interface_impl::getArgAttrs(func, index);
      absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
    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/transforms/xla_call_module_deserialization.cc

                  << " but enclosing XlaCallModule has a function list of size "
                  << function_list.size();
              return WalkResult::interrupt();
            }
    
            llvm::SmallVector<NamedAttribute> new_config;
            // Copy the attributes in the current config except `called_index`.
            for (auto attr : backend_config) {
              if (attr.getName() != kCalledIndexAttrName) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/converter_gen.cc

    // Signature:
    // void mlir::BuiltinOptions{id}ToAttributes(
    //     tflite::BuiltinOptions{id}Union op_union,
    //     mlir::Builder builder,
    //     llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes);
    //
    // where id is an empty string if builtin_options_id is 1, or builtin_options_id
    // otherwise.
    static void EmitBuiltinOptionsToAttributes(const RecordKeeper &record_keeper,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 15:05:28 UTC 2023
    - 23.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

        const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) {
      // We apply all result attributes at once to avoid excessive allocations when
      // we have many result values.
      llvm::SmallVector<std::vector<NamedAttribute>, 8> result_attrs;
      {
        llvm::SmallVector<DictionaryAttr, 8> tmp;
        func.getAllResultAttrs(tmp);
    
        for (const auto& res : tmp) {
          result_attrs.push_back(res.getValue().vec());
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

                                rewriter.getIntegerType(64)),
          padding_nums);
      return padding_attr;
    }
    
    template <typename UniformQuantizedConvolutionOp>
    FailureOr<SmallVector<NamedAttribute>> ConvertToMhloConvolutionOpAttrs(
        UniformQuantizedConvolutionOp op, PatternRewriter &rewriter) {
      // TODO(b/261005147): Update the lowering logic after migration to mhlo
      // ConvolutionDimensionNumbers.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc

          llvm::ArrayRef<Type>{RankedTensorType::get({ShapedType::kDynamic},
                                                     builder->getIntegerType(64))},
          llvm::ArrayRef<Value>{compilation_key},
          llvm::ArrayRef<NamedAttribute>{
              builder->getNamedAttr("index",
                                    builder->getI64IntegerAttr(execute_arg_index)),
              builder->getNamedAttr("is_output", builder->getBoolAttr(false))});
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc

        auto bad_consumers =
            b.getNamedAttr("bad_consumers", b.getI32ArrayAttr({1, 2, 3, 4}));
    
        module->setAttr("tf.versions",
                        b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>(
                            {producer, min_consumer, bad_consumers})));
    
        return run_status_;
      }
    
      Status run_status_;
    };
    
    FunctionDef XTimesTwo() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 08:25:30 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc

        for (auto [value, aggregator_id] :
             llvm::zip_equal(quantizable_values, aggregator_ids)) {
          // ID attribute will have empty value for now.
          SmallVector<NamedAttribute, 5> attributes{
              rewriter.getNamedAttr("id", rewriter.getStringAttr(aggregator_id)),
              rewriter.getNamedAttr(
                  "calibration_method",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

      auto rendezvous_name = StringAttr::get(context, formatted_key);
      auto rendezvous_name_attr = NamedAttribute(
          StringAttr::get(context, xla::kXlaHostTransferRendezvousNameAttr),
          rendezvous_name);
    
      auto host_handler_name_value =
          StringAttr::get(context, host_handler_name.str());
      auto host_handler_name_attr = NamedAttribute(
          StringAttr::get(context, xla::kXlaHostTransferHandlerNameAttr),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
Back to top