- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 61 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/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/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/extract_outside_compilation_pass_test.cc
for (int i = 0; i < 2; ++i) { tensor_proto.add_int_val(1); } NodeDef const_def; TF_CHECK_OK(NodeDefBuilder("const", "Const") .Attr("dtype", DT_INT32) .Attr("value", tensor_proto) .Finalize(&const_def)); Status s; Node *const_node = g->AddNode(const_def, &s); TF_CHECK_OK(s); NodeDef fn_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
Location loc = graph_func.getLoc(); FunctionType ftype = graph_func.getFunctionType(); func::FuncOp func = rewriter.create<func::FuncOp>( graph_func.getLoc(), graph_func->getAttrOfType<StringAttr>(SymbolTable::getSymbolAttrName()) .getValue(), ftype); func->setAttrs(graph_func->getAttrs()); llvm::SmallVector<Type> arg_types; llvm::SmallVector<Type> res_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
assert(while_caller.body_function() == func); while_callers.push_back(while_caller); } } return while_callers; } bool IsResourceType(Type type) { return mlir::isa<TF::ResourceType>(getElementTypeOrSelf(type)); } bool OnlyOperatesOnCompositeDevices( TF::TPUExecuteAndUpdateVariablesOp& op, const TF::SideEffectAnalysis::Info& side_effect_analysis,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
LogicalResult ConstantFoldMul(mhlo::MulOp op, PatternRewriter &rewriter) { ShapedType type = mlir::dyn_cast<ShapedType>(op.getType()); Type etype = type.getElementType(); Attribute result = {}; if (mlir::isa<FloatType>(etype)) { result = BinaryFolder<mhlo::MulOp, FloatType, APFloat, std::multiplies<APFloat>>( &op); } else if (mlir::isa<IntegerType>(etype)) { result =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
attr.default_value().value_case() == AttrValue::kType) { type = DataType_Name(attr.default_value().type()); } else if (attr.has_allowed_values() && attr.allowed_values().value_case() == AttrValue::kList && !attr.allowed_values().list().type().empty()) { type = DataType_Name(attr.allowed_values().list().type(0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0)