Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 213 for attr_ (0.04 sec)

  1. tensorflow/compiler/jit/node_matchers_test.cc

          "to be 0 but found 1");
    }
    
    TEST(NodeMatchers, Attrs) {
      Scope root = Scope::NewRootScope().ExitOnError();
      Output enter = ops::internal::Enter(
          root.WithOpName("enter"),
          ops::Placeholder(root.WithOpName("data"), DT_FLOAT), "frame_name",
          ops::internal::Enter::Attrs{}.IsConstant(true));
      EXPECT_THAT(enter.node(), NodeWith(Attr("is_constant", true)));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 14:43:57 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/shape_inference.cc

                  value.add_int64_val(dim.size());
                }
              }
              AddNodeAttr("value", value, &const_def);
              for (auto const& attr : n->attrs()) {
                if (*attr.first.begin() == '_') {
                  AddNodeAttr(attr.first, attr.second, &const_def);
                }
              }
    
              TF_ASSIGN_OR_RETURN(Node * const_node, graph->AddNode(const_def));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc

      const char* key = pair.getName().data();
      const auto attr = pair.getValue();
    
      if (mlir::isa<::mlir::IntegerAttr>(attr)) {
        fbb->Int(key, mlir::dyn_cast<mlir::IntegerAttr>(attr).getInt());
        return success();
      }
    
      if (mlir::isa<::mlir::FloatAttr>(attr)) {
        fbb->Double(key, mlir::dyn_cast<mlir::FloatAttr>(attr).getValueAsDouble());
        return success();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

            // Same type repeated "repeats" times.
            Attribute attr = attrs_[output_arg.type_attr()];
            if (!attr)
              return InvalidArgument("Missing attribute '", output_arg.type_attr(),
                                     "' required for output '", output_arg.name(),
                                     "'");
            TypedAttr type_attr = mlir::dyn_cast<TypedAttr>(attr);
            if (!type_attr)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

      let description = [{
        The `attr` operation stores TF op's attribute, which doesn't support
        arithmetic operations.
    
        Example:
    
        ```mlir
        %1 = "tfr.constant"() { value: i32 } : () -> !tfr.attr
        %2 = "tfr.constant"() { value: [i32, f32] } : () -> !tfr.attr
        %3 = tfr.constant [i32, f32] -> !tfr.attr
        %4 = tfr.constant f32 -> !tfr.attr
        ```
      }];
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/aot/tfcompile.bzl

        attrs = {
            "model_name": attr.string(),
            "srcs": attr.label_list(mandatory = True, allow_files = True),
            "header_out": attr.output(),
            "cmd": attr.string(),
            "tfcompile_tool": attr.label(cfg = "exec", executable = True, allow_files = True),
            "tfcompile_graph": attr.label(allow_single_file = True),
            "tfcompile_config": attr.label(allow_single_file = True),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 19:18:08 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  7. tools/istio-iptables/pkg/log/nflog.go

    		if attrs.UID != nil {
    			uid = *attrs.UID
    		}
    		if attrs.GID != nil {
    			gid = *attrs.GID
    		}
    		inDev, outDev := "", ""
    		if attrs.InDev != nil {
    			ii, err := net.InterfaceByIndex(int(*attrs.InDev))
    			if err == nil {
    				inDev = ii.Name
    			}
    		}
    		if attrs.OutDev != nil {
    			ii, err := net.InterfaceByIndex(int(*attrs.OutDev))
    			if err == nil {
    				outDev = ii.Name
    			}
    		}
    		iptablesTrace.WithLabels(
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Thu Jun 06 15:59:40 UTC 2024
    - 2.8K bytes
    - Viewed (0)
  8. tensorflow/cc/framework/scope.cc

        const Operation& colocate_with_op) const {
      std::unordered_set<string> current_constraints(colocation_constraints_);
      const AttrSlice attrs = colocate_with_op.node()->attrs();
      std::vector<string> node_constraints;
      if (TryGetNodeAttr(attrs, kColocationAttrName, &node_constraints)) {
        for (const string& entry : node_constraints) {
          StringPiece s(entry);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_patterns.td

        (TFL_TransposeOp
          (TFL_ResizeBilinearOp
            (TFL_TransposeOp $input,
              (Arith_ConstantOp
                ConstantAttr<RankedI32ElementsAttr<[4]>,"{0, 2, 3, 1}">)),
            (Arith_ConstantOp:$output_size (GetI32DenseAttr (GetAsVectorAttr<"output"> $attrs))),
            (GetCompositeAttributeAs<"align_corners", "BoolAttr"> $attrs),
            ConstBoolAttrTrue,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc

        auto operand_size_attr = builder.getDenseI32ArrayAttr(operand_sizes);
        NamedAttrList attrs(send_op->getAttrs());
        attrs.set(send_op.getOperandSegmentSizeAttr(), operand_size_attr);
    
        AddOperandAndRewriteAs<XlaSendTPUEmbeddingGradientsOp>(send_op, dedup_op,
                                                               attrs, &builder);
      }
      return success();
    }
    
    void RewriteTPUEmbeddingOps::runOnOperation() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 22:55:42 UTC 2024
    - 4.5K bytes
    - Viewed (0)
Back to top