- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 40 for Dtype (0.04 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)