- Sort Score
- Result 10 results
- Languages All
Results 31 - 36 of 36 for Dtype (0.03 sec)
-
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/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/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/compiler/mlir/lite/utils/const_tensor_utils.cc
type, builder.getFloatAttr(element_ty, unique_index)); if (auto qtype = mlir::dyn_cast<QuantizedType>(element_ty)) { mlir::RankedTensorType new_type = tensorflow::GetTypeFromTFTensorShape( type.getShape(), qtype.getStorageType()); return DenseElementsAttr::get( new_type, builder.getIntegerAttr(qtype.getStorageType(), unique_index)); } llvm_unreachable("unhandled element type"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0)