Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 35 for getNamedAttr (0.5 sec)

  1. 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)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc

    //      align_corners = false, device = "", half_pixel_centers = true}
    LogicalResult SmuggleOp(Operation* op, PatternRewriter& rewriter) {
      auto call_target =
          rewriter.getNamedAttr("call_target_name", op->getName().getIdentifier());
      SmallVector<NamedAttribute> attrs{op->getAttrs()};
      attrs.push_back(call_target);
      auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 19 01:14:04 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. 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)
  6. tensorflow/compiler/mlir/tensorflow/utils/convert_attr.cc

          for (const auto& func_attr : value.func().attr()) {
            TF_ASSIGN_OR_RETURN(auto attr,
                                ConvertAttributeValue(func_attr.second, builder));
            attrs.push_back(builder->getNamedAttr(func_attr.first, attr));
          }
          auto func_attrs = builder->getDictionaryAttr(attrs);
          return mlir::TF::FuncAttr::get(builder->getContext(), value.func().name(),
                                         func_attrs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

        return llvm::SmallVector<mlir::NamedAttribute, 4>{
            builder.getNamedAttr("then_branch", then_attr),
            builder.getNamedAttr("else_branch", else_attr),
            // TODO(b/139667752): Analyze statelessness correctly
            builder.getNamedAttr("is_stateless", builder.getBoolAttr(false))};
      }
      if (auto* opts = options.AsWhileOptions()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  8. 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)
  9. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

            llvm::ArrayRef<Value>{},
            llvm::ArrayRef<NamedAttribute>{
                builder->getNamedAttr(kDeviceAttr, builder->getStringAttr(device)),
                builder->getNamedAttr("container", builder->getStringAttr("")),
                builder->getNamedAttr(
                    "shared_name",
                    builder->getStringAttr(GetRandomStateVariableName()))}));
      }
      return state_vars;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        auto inputs = b.getNamedAttr("inputs", b.getStringAttr(ss.str()));
        s.clear();
        llvm::interleave(ret_nodes, ss, node_name, ",");
        auto outputs = b.getNamedAttr("outputs", b.getStringAttr(ss.str()));
        s.clear();
        llvm::interleave(specs.control_outputs, ss, ",");
        auto control_outputs =
            b.getNamedAttr("control_outputs", b.getStringAttr(ss.str()));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top