Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 21 for getNamedAttr (0.13 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/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)
  5. 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)
  6. tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc

                 const FunctionLibraryDefinition& function_library) override {
        mlir::Builder b(module.getContext());
        auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(0));
        auto min_consumer = b.getNamedAttr("min_consumer", b.getI32IntegerAttr(0));
        auto bad_consumers =
            b.getNamedAttr("bad_consumers", b.getI32ArrayAttr({1, 2, 3, 4}));
    
        module->setAttr("tf.versions",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 08:25:30 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

            if (attr.getName() == kFuncDeviceAttr) {
              resultAttrForOp[i] = builder.getNamedAttr(
                  kFuncDeviceAttr,
                  StringAttr::get(func.getContext(), it->getSecond()));
              found = true;
              break;
            }
          }
          if (!found) {
            resultAttrForOp.push_back(builder.getNamedAttr(
                kFuncDeviceAttr,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc

          llvm::ArrayRef<Value>{compilation_key},
          llvm::ArrayRef<NamedAttribute>{
              builder->getNamedAttr("index",
                                    builder->getI64IntegerAttr(execute_arg_index)),
              builder->getNamedAttr("is_output", builder->getBoolAttr(false))});
    }
    
    // Builds a TPUCopyWithLayoutOp with the given get_layout op and input.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        if (!graph) continue;
        Builder b(&context);
        auto producer = b.getNamedAttr(
            "producer", b.getI32IntegerAttr(graph.getVersion().getProducer()));
        auto min_consumer = b.getNamedAttr(
            "min_consumer",
            b.getI32IntegerAttr(graph.getVersion().getMinConsumer()));
        auto bad_consumers =
            b.getNamedAttr("bad_consumers",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

        attributes.push_back(rewriter.getNamedAttr("fused_activation_function",
                                                   rewriter.getStringAttr("TANH")));
        // cell_clip.
        attributes.push_back(
            rewriter.getNamedAttr("cell_clip", rewriter.getF32FloatAttr(0.0)));
        // proj_clip.
        attributes.push_back(
            rewriter.getNamedAttr("proj_clip", rewriter.getF32FloatAttr(0.0)));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
Back to top