Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 41 for Dtype (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

    shared_name: the name by which this variable is referred to.
    dtype and shape: attributes representing the data type and shape held in the
      variable.
    
    Example:
        resource_variable_ops.var_handle_op(
              dtype=dtypes.int32, shape=[8, 16], container="foo", shared_name="bar")
      returns a handle for a variable with name "bar" in container "foo", and the
      variable holds a tensor of shape [8, 16] and dtype int32.
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/math_grad.cc

    REGISTER_NO_GRADIENT_OP("Floor");
    
    // Conjugate helper function returns the conjugate of an Output if it
    // is complex valued.
    Output ConjugateHelper(const Scope& scope, const Output& out) {
      DataType dtype = out.type();
      if (dtype == DT_COMPLEX64 || dtype == DT_COMPLEX128) {
        return Conj(scope, out);
      } else {
        return out;
      }
    }
    
    // TODO(andydavis) Add control dependencies to gradient functions (as needed).
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir

      // CHECK: "tf.TensorArrayV3"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:40:22 UTC 2024
    - 68.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        return false;
      }
    
      if (n->type_string() != "AssignAddVariableOp" &&
          n->type_string() != "AssignSubVariableOp") {
        return false;
      }
    
      DataType dtype;
      if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) {
        return false;
      }
    
      Node* const_input = nullptr;
      for (const Edge* e : n->in_edges()) {
        if (!e->IsControlEdge() && e->src()->IsConstant()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc

        LLVM_DEBUG(llvm::dbgs() << "Zero point value should be a tensor type. Got: "
                                << zero_points_type << ".\n");
        return failure();
      }
    
      if (Type zero_points_element_type = zero_points_type.getElementType();
          !mlir::isa<IntegerType>(zero_points_element_type)) {
        LLVM_DEBUG(llvm::dbgs() << "Zero point should be an integer type. Got: "
                                << zero_points_element_type << ".\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/deadness_analysis.cc

        if (node->type_string() == "Const" && must_be_true) {
          const TensorProto* proto = nullptr;
          TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "value", &proto));
    
          Tensor tensor(proto->dtype());
          TF_RET_CHECK(tensor.FromProto(*proto));
    
          *predicate = tensor.scalar<bool>()() ? MakeTrue() : MakeFalse();
          return absl::OkStatus();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_test.cc

      TF_OperationDescription* init(string type) {
        // Construct op_name to match the name used by REGISTER_OP in the
        // ATTR_TEST_REGISTER calls above.
        string op_name = "CApiAttributesTestOp";
        if (type.find("list(") == 0) {
          op_name += "List";
          type = type.replace(0, 5, "");
          type = type.replace(type.size() - 1, 1, "");
        }
        op_name += type;
        return TF_NewOperation(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

    }
    
    // Return true if the passed quantized type is unsigned.
    bool QuantizedTypeIsUnsigned(Type type) {
      return TypeSwitch<Type, bool>(type)
          .Case<mlir::TF::Qint8Type>([](Type) { return false; })
          .Case<mlir::TF::Qint16Type>([](Type) { return false; })
          .Case<mlir::TF::Qint32Type>([](Type) { return false; })
          .Case<mlir::TF::Quint8Type>([](Type) { return true; })
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py

        save.save(
            model,
            self._input_saved_model_path,
            signatures=model.matmul.get_concrete_function(
                tensor_spec.TensorSpec(
                    shape=input_shape, dtype=dtypes.float32, name='input_tensor'
                )
            ),
        )
    
        def data_gen() -> repr_dataset.RepresentativeDataset:
          for _ in range(50):
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 51.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/tests/tpu_cluster_formation.mlir

    // Test cluster with bad `num_cores_per_replica` attribute.
    !rtype = tensor<!tf_type.resource<tensor<10x3xf32>>>
    func.func @replication_with_model_parallelism(%arg0: !rtype, %arg1: !rtype, %arg2: !rtype, %arg3: !rtype) -> (tensor<10x3xf32>) {
      %2 = "tf.TPUReplicatedInput"(%arg0, %arg2) : (!rtype, !rtype) -> !rtype
      %3 = "tf.TPUReplicatedInput"(%arg1, %arg3) : (!rtype, !rtype) -> !rtype
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 53.9K bytes
    - Viewed (0)
Back to top