Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 35 for getNamedAttr (0.18 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

        attributes.emplace_back(builder.getNamedAttr(
            "backend_config", BuildVhloStringV1Attr(op->backend_config, builder)));
        attributes.emplace_back(builder.getNamedAttr(
            "called_computations", BuildVhloArrayV1Attr({}, builder)));
        attributes.emplace_back(builder.getNamedAttr(
            "operand_layouts", BuildVhloArrayV1Attr({}, builder)));
        attributes.emplace_back(builder.getNamedAttr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

        }
      }
      attrs.push_back(rewriter.getNamedAttr("lhs_quantization_axis",
                                            activation_quantization_axis));
      attrs.push_back(rewriter.getNamedAttr("rhs_quantization_axis",
                                            activation_quantization_axis));
      attrs.push_back(rewriter.getNamedAttr("output_quantization_axis",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

          const auto& value = m[key];
          switch (value.GetType()) {
            case flexbuffers::FBT_INT: {
              auto attr = value.AsInt64();
              auto named_attr = builder->getNamedAttr(
                  key, builder->getIntegerAttr(builder->getIntegerType(64), attr));
              attrs.push_back(named_attr);
              break;
            }
            case flexbuffers::FBT_VECTOR_BOOL: {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc

            ArrayRef<Value>{},
            ArrayRef<NamedAttribute>{
                builder.getNamedAttr("device", device_attr),
                builder.getNamedAttr("container",
                                     variable_v2_op.getContainerAttr()),
                builder.getNamedAttr("shared_name",
                                     builder.getStringAttr(variable_name))});
        for (Operation *user :
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc

      mlir::Builder b(module.getContext());
      auto producer =
          b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer()));
      auto min_consumer = b.getNamedAttr(
          "min_consumer", b.getI32IntegerAttr(versions.min_consumer()));
      auto bad_consumers = b.getNamedAttr(
          "bad_consumers",
          b.getI32ArrayAttr(llvm::ArrayRef<int32_t>(
              versions.bad_consumers().data(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc

        SmallVector<NamedAttribute> dump_attributes{
            rewriter.getNamedAttr("log_dir_path",
                                  rewriter.getStringAttr(folder_name)),
            rewriter.getNamedAttr("file_name", rewriter.getStringAttr(file_name)),
            // The op is disabled by default. Otherwise, values will be saved
            // during calibration.
            rewriter.getNamedAttr("enabled", rewriter.getBoolAttr(enabled)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 22:55:22 UTC 2024
    - 13K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc

              rewriter.getNamedAttr("id", rewriter.getStringAttr(aggregator_id)),
              rewriter.getNamedAttr(
                  "calibration_method",
                  rewriter.getI32IntegerAttr(calib_opts_.calibration_method())),
              rewriter.getNamedAttr("num_bins",
                                    rewriter.getI32IntegerAttr(effective_num_bins)),
              rewriter.getNamedAttr(
                  "min_percentile",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc

          mlir::ModuleOp::create(mlir::UnknownLoc::get(&context));
      mlir::Builder builder(*module_ref);
      (*module_ref)
          ->setAttr("tf.devices",
                    builder.getDictionaryAttr(builder.getNamedAttr(
                        "bad_device", builder.getDictionaryAttr({}))));
    
      mlir::TF::RuntimeDevices devices;
      EXPECT_TRUE(mlir::failed(GetDevicesFromOp(*module_ref, &devices)));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc

        auto key = key_value.getName();
        auto value = key_value.getValue();
        op_attr_array.push_back(builder.getArrayAttr({key, value}));
      }
    
      result.attributes.push_back(
          builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array)));
    
      // TODO(tfrt-devs): support func attributes in tfrt_fallback_sync.
      if (options.has_func_attr) {
        llvm::SmallVector<mlir::Attribute, 4> op_func_attr_array;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/utils/device_util.cc

        if (device->device_type() == DEVICE_GPU) {
          auto metadata = ParseGpuDeviceMetadata(*device, &builder);
          devices.push_back(builder.getNamedAttr(name, metadata));
        } else {
          auto metadata = builder.getUnitAttr();
          devices.push_back(builder.getNamedAttr(name, metadata));
        }
      }
    
      op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.4K bytes
    - Viewed (0)
Back to top