- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 136 for Dtype (0.07 sec)
-
tensorflow/c/eager/c_api_experimental_test.cc
// First create a simple identity function. TF_Graph* function_graph = TF_NewGraph(); TF_OperationDescription* arg_descr = TF_NewOperation(function_graph, "Placeholder", "arg"); TF_SetAttrType(arg_descr, "dtype", TF_INT32); TF_Status* status = TF_NewStatus(); TF_Operation* arg = TF_FinishOperation(arg_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
llvm::dyn_cast<TF::ConstOp>(fold_results[0].getDefiningOp()).getValue(), &tensor)); xla::Shape xla_shape; TF_RETURN_IF_ERROR(tensorflow::TensorShapeToXLAShape( tensor.dtype(), tensor.shape(), &xla_shape)); xla::PjRtClient::HostBufferSemantics host_buffer_semantics = xla::PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes; TF_ASSIGN_OR_RETURN( auto buffer,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
} Node* KnownShapeBase(DataType dtype, absl::Span<const int> shape, const GraphDefBuilder::Options& opts) { if (opts.HaveError()) return nullptr; NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", opts.op_registry()); TensorProto value; value.set_dtype(dtype); for (int dim : shape) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K 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/c/c_api_function_test.cc
#include "tensorflow/core/platform/strcat.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { // Specification for expected input/output and its type. // DataType value of DT_INVALID signifies that we don't want to // check the data type. typedef std::pair<string, DataType> IOSpec; const char* kFeedStackToString = "File \"feed.cc\", line 10, in alpha";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); int num_retvals = 1; TFE_Execute(op, out_value, &num_retvals, status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 10:03:59 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
ElementsAttr GetTensorValueAsElementsAttr( TF::VarHandleOp var_handle_op, const tensorflow::Tensor& resource_tensor, const tensorflow::DeviceMgr* mgr, OpBuilder builder) { if (resource_tensor.dtype() != tensorflow::DT_RESOURCE) { return GetTensorValueAsElementsAttr(resource_tensor, builder); } auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K 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/jit/compilability_check_util.cc
return false; } if (node.type_string() == "Const") { const AttrValue* attr = node.attrs().Find("dtype"); if (!op_filter_.allow_string_consts && attr != nullptr && attr->type() == DT_STRING) { *uncompilable_reason = "Const op with type DT_STRING is not supported by XLA."; return false; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K 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)